Add schema name extract in the visitor

Signed-off-by: Gabriel de Maeztu <[email protected]>
merqurio · Aug 25, 2024 · d2ae0bd · d2ae0bd
1 parent 77a94c1
commit d2ae0bd
Show file tree

Hide file tree

Showing 2 changed files with 125 additions and 85 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -184,20 +184,20 @@ mod tests {
 
         let matches = cmd.try_get_matches_from(vec![
             "oxigration",
-            "generate",
+            "migrate",
             "-d",
-            "test_schemas/",
+            "tests/schemas/baseline/",
             "-c",
             "postgresql://test@localhost/test",
         ]);
 
         assert!(matches.is_ok());
         let matches = matches.unwrap();
-        assert_eq!(matches.subcommand_name(), Some("generate"));
+        assert_eq!(matches.subcommand_name(), Some("migrate"));
         if let Some(sub_matches) = matches.subcommand_matches("generate") {
             assert_eq!(
                 sub_matches.get_one::<String>("dir").unwrap(),
-                "test_schemas/"
+                "tests/schemas/baseline/"
             );
             assert_eq!(
                 sub_matches.get_one::<String>("connection").unwrap(),
@@ -212,20 +212,20 @@ mod tests {
 
         let matches = cmd.try_get_matches_from(vec![
             "oxigration",
-            "migrate",
+            "generate",
             "-d",
-            "test_schemas/",
+            "tests/schemas/generated/",
             "-c",
             "postgresql://test@localhost/test",
         ]);
 
         assert!(matches.is_ok());
         let matches = matches.unwrap();
-        assert_eq!(matches.subcommand_name(), Some("migrate"));
+        assert_eq!(matches.subcommand_name(), Some("generate"));
         if let Some(sub_matches) = matches.subcommand_matches("migrate") {
             assert_eq!(
                 sub_matches.get_one::<String>("dir").unwrap(),
-                "test_schemas/"
+                "tests/schemas/generated/"
             );
             assert_eq!(
                 sub_matches.get_one::<String>("connection").unwrap(),