Add a simple test for the new script. Currently only checking that it…

… runs.
COSIMA · Jun 30, 2022 · 5bd09f6 · 5bd09f6
1 parent e742902
commit 5bd09f6
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 2 deletions.
diff --git a/cosima_cookbook/database_update.py b/cosima_cookbook/database_update.py
@@ -4,7 +4,7 @@
 import cosima_cookbook as cc
 
 
-def main():
+def main(argv=None):
     parser = argparse.ArgumentParser(description="Update COSIMA cookbook database.")
     parser.add_argument(
         "dirs", type=pathlib.Path, nargs="+", help="Directories to index."
@@ -17,7 +17,7 @@ def main():
         default="cosima_master.db",
         help="Database to update.",
     )
-    args = parser.parse_args()
+    args = parser.parse_args(argv)
 
     print(cc)
 

diff --git a/test/data/update/experiment_a/test1.nc b/test/data/update/experiment_a/test1.nc
diff --git a/test/data/update/experiment_b/test2.nc b/test/data/update/experiment_b/test2.nc
diff --git a/test/test_update.py b/test/test_update.py
@@ -0,0 +1,13 @@
+import shlex
+from cosima_cookbook import database_update
+
+
+def test_database_update(tmpdir):
+
+    args = shlex.split(
+        "-db {db} test/data/update/experiment_a test/data/update/experiment_b".format(
+            db=tmpdir.join("test.db")
+        )
+    )
+
+    database_update.main(args)