main: fix tests

carte-data · Feb 26, 2021 · da27ad0 · da27ad0
1 parent 18a9f62
commit da27ad0
Show file tree

Hide file tree

Showing 8 changed files with 229 additions and 34 deletions.
diff --git a/carte_cli/utils/config_parser.py b/carte_cli/utils/config_parser.py
@@ -1,7 +1,7 @@
 import io
+import importlib
 from ruamel.yaml import YAML
 from carte_cli.extractor.glue_extractor import GlueExtractor
-from databuilder.extractor.postgres_metadata_extractor import PostgresMetadataExtractor
 
 yaml = YAML()
 
@@ -13,6 +13,9 @@ def create_glue_connection(conn_dict):
 
 
 def create_postgres_connection(conn_dict):
+    PostgresMetadataExtractor = importlib.import_module(
+        "databuilder.extractor.postgres_metadata_extractor"
+    ).PostgresMetadataExtractor
     extractor = PostgresMetadataExtractor()
     extractor_scope = extractor.get_scope()
     config = conn_dict.get(CONFIG_KEY, {})

diff --git a/carte_cli/utils/frontmatter.py b/carte_cli/utils/frontmatter.py
@@ -16,7 +16,7 @@ def parse(filename):
     if len(data) >= 2 and data[1] != "":
         metadata = yaml.load(data[1])
 
-    if len(data) >= 2:
+    if len(data) >= 3:
         content = data[2]
 
     return metadata, content

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -25,6 +25,9 @@ SQLAlchemy = { version = "^1.3.23", optional = true }
 [tool.poetry.extras]
 postgres = ["psycopg2", "SQLAlchemy"]
 
+[tool.poetry.dev-dependencies]
+pytest = "^6.2.2"
+
 [build-system]
 requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"