Add replication key to metadata

MeltanoLabs · Aug 28, 2023 · 94fa72b · 94fa72b
1 parent fc0bd07
commit 94fa72b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -327,6 +327,8 @@ def test_incremental():
         Column("name", String),
         Column("updated_at", DateTime),
     )
+    replication_key = "updated_at"
+
     with engine.connect() as conn:
         if table.exists(conn):
             table.drop(conn)
@@ -380,6 +382,7 @@ def test_incremental():
                 metadata["metadata"]["selected"] = True
                 if metadata["breadcrumb"] == []:
                     metadata["metadata"]["replication-method"] = "INCREMENTAL"
+                    metadata["metadata"]["replication-key"] = replication_key
 
     test_runner = PostgresTestRunner(
         tap_class=TapPostgres,
@@ -392,7 +395,7 @@ def test_incremental():
         "value": {
             "bookmarks": {
                 altered_table_name: {
-                    "replication_key": "updated_at",
+                    "replication_key": replication_key,
                     "replication_key_value": "2022-10-01T00:00:00+00:00",
                 },
             },