editoast: test persist_batch on big chunks

OpenRailAssociation · Nov 23, 2022 · ede0ae3 · ede0ae3
1 parent 537f127
commit ede0ae3
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/editoast/src/schema/buffer_stop.rs b/editoast/src/schema/buffer_stop.rs
@@ -102,4 +102,15 @@ mod test {
             assert!(BufferStop::persist_batch(&data, infra.id, conn).is_ok());
         });
     }
+
+    #[test]
+    fn test_persist_large() {
+        test_infra_transaction(|conn, infra| {
+            let data = (0..(2_usize.pow(16) * 2))
+                .map(|_| BufferStop::default())
+                .collect::<Vec<BufferStop>>();
+
+            assert!(BufferStop::persist_batch(&data, infra.id, conn).is_ok());
+        });
+    }
 }