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

… runs.

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)
 

test/test_indexing.py

@@ -38,7 +38,7 @@ def test_find_files():
 
     # Test works with alternative suffix
     files = database.find_files("test/", "*.py")
-    assert len(files) == 8
+    assert len(files) == 9
 
     for f in files:
         assert Path(f).suffix == ".py"

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)