Support setting amount to be generated

1 files changed, 10 insertions, 8 deletions

diff --git a/test/test_parser.ml b/test/test_parser.ml
index 941ac56..5a56e6a 100644
--- a/test/test_parser.ml
+++ b/test/test_parser.ml
@@ -13,26 +13,28 @@ let test_parse s () =
 
 
 let rows_suite =
-  [ "Can parse uuidv4", `Quick, test_parse "user: id _uuidv4"
+  [ "Can parse uuidv4", `Quick, test_parse "user (1): id _uuidv4"
   ; ( "Can parse multiple tables"
     , `Quick
-    , test_parse "user: id _uuidv4\npurchases: id _uuidv4" )
-  ; "Resolves conflicting names", `Quick, test_parse "purchases: name _uuidv4"
-  ; "Supports names", `Quick, test_parse "purchases: name _name"
-  ; "Supports ints with min/max", `Quick, test_parse "purchases: name _int<1,20>"
+    , test_parse "user (1): id _uuidv4\npurchases (1): id _uuidv4" )
+  ; "Supports names", `Quick, test_parse "purchases (1): name _name"
+  ; "Supports ints with min/max", `Quick, test_parse "purchases (1): name _int(1,20)"
+  ; ( "Supports setting amount of fixture to generate"
+    , `Quick
+    , test_parse "purchases (5): name _int(1,20)" )
   ]
 
 
 let relations_suite =
   [ ( "Can reference other tables"
     , `Quick
-    , test_parse "user: id _uuidv4\npurchases: id _uuidv4, userid user.id" )
+    , test_parse "user (2): id _uuidv4\npurchases (3): id _uuidv4, userid user.id" )
   ; ( "Can use ; as a separator"
     , `Quick
-    , test_parse "user: id _uuidv4; purchases: id _uuidv4" )
+    , test_parse "user (2): id _uuidv4; purchases (5): id _uuidv4" )
   ; ( "Supports multiple newlines"
     , `Quick
-    , test_parse "user: id _uuidv4\n\n\n\n\npurchases: id _uuidv4" )
+    , test_parse "user (2): id _uuidv4\n\n\n\n\npurchases (5): id _uuidv4" )
   ]