chore: unit tests

snapADDY · Jul 20, 2023 · aed4a6e · aed4a6e
1 parent f20268a
commit aed4a6e
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/test_query.py b/tests/test_query.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from databank import QueryCollection
+from databank.query import QueryCollection, is_valid_query_header
 
 
 @pytest.fixture
@@ -18,3 +18,11 @@ def test_query_collection(queries: QueryCollection):
 
     with pytest.raises(KeyError):
         queries["invalid"]
+
+
+def test_is_valid_query_header():
+    assert is_valid_query_header("/* @name foo */")
+    assert not is_valid_query_header("/* name: foo bar */")
+    assert not is_valid_query_header("/** @name foo **/")
+    assert not is_valid_query_header("/* @name foo bar */")
+    assert not is_valid_query_header("-- @name: foo")