also add model to test

njit-jerse · Jul 30, 2024 · 5d0d644 · 5d0d644
1 parent eb1e4b1
commit 5d0d644
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/TestMain.py b/TestMain.py
@@ -61,7 +61,8 @@ def test_build_specimin_command(self):
         targets = [{
                     "method": "getMode(ColumnMetadata, Map<String, String>)",
                     "file": "IndexMode.java",
-                    "package": 'org.apache.cassandra.index.sasi.conf'
+                    "package": 'org.apache.cassandra.index.sasi.conf',
+                    "model": "cf"
                    }]
 
         target_dir = '/user/ISSUES/cf-6077'
@@ -76,7 +77,8 @@ def test_build_specimin_command(self):
         targets = [{
                     "method": "transform(String, byte[])",
                     "file": "Avro2Confluent.java",
-                    "package": 'com.fillmore_labs.kafka.sensors.serde.confluent.interop'
+                    "package": 'com.fillmore_labs.kafka.sensors.serde.confluent.interop',
+                    "model" : "cf"
                    }]
 
         target_dir = '/user/ISSUES/cf-6019'
@@ -109,7 +111,8 @@ def test_run_specimin(self):
         targets = [{
                     "method": "bar()",
                     "file": "Simple.java",
-                    "package": "com.example"
+                    "package": "com.example",
+                    "model" : "cf"
                    }]
         target_dir = 'resources/onefilesimple'
 
@@ -121,4 +124,4 @@ def test_run_specimin(self):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()