1 files changed, 66 insertions, 0 deletions
diff --git a/test/apps/test_migrate.py b/test/apps/test_migrate.py
new file mode 100644
index 0000000..957509a
--- /dev/null
+++ b/test/apps/test_migrate.py
@@ -0,0 +1,66 @@
+"""
+
+Part of the bsfs test suite.
+A copy of the license is provided with the project.
+Author: Matthias Baumgartner, 2022
+"""
+# imports
+import contextlib
+import io
+import os
+import sys
+import unittest
+import unittest.mock
+
+# bsie imports
+from bsfs.schema import Schema
+
+# objects to test
+from bsfs.apps.migrate import main
+
+
+## code ##
+
+class TestMigrate(unittest.TestCase):
+    def test_main(self):
+        config = os.path.join(os.path.dirname(__file__), 'config.json')
+        schema_1 = os.path.join(os.path.dirname(__file__), 'schema-1.nt')
+        schema_2 = os.path.join(os.path.dirname(__file__), 'schema-2.nt')
+
+        # provide no config
+        with contextlib.redirect_stderr(io.StringIO()):
+            self.assertRaises(SystemExit, main, [])
+
+        # read schema from file
+        with open(schema_1) as ifile:
+            target = Schema.from_string(ifile.read())
+        graph = main([config, schema_1])
+        self.assertTrue(target <= graph.schema)
+
+        # read schema from multiple files
+        with open(schema_1) as ifile:
+            target = Schema.from_string(ifile.read())
+        with open(schema_2) as ifile:
+            target = target + Schema.from_string(ifile.read())
+        graph = main([config, schema_1, schema_2])
+        self.assertTrue(target <= graph.schema)
+
+        # read schema from stdin
+        with open(schema_1, 'rt') as ifile:
+            target = Schema.from_string(ifile.read())
+        with open(schema_1, 'rt') as ifile:
+            with unittest.mock.patch('sys.stdin', ifile):
+                graph = main([config])
+        self.assertTrue(target <= graph.schema)
+
+        # remove predicates
+        # NOTE: cannot currently test this since there's nothing to remove in the loaded (empty) schema.
+
+
+## main ##
+
+if __name__ == '__main__':
+    unittest.main()
+
+## EOF ##
+