Add built-in test

meltano · Oct 21, 2024 · 4dab2f3 · 4dab2f3
1 parent 42d8fd0
commit 4dab2f3
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/singer_sdk/testing/suites.py b/singer_sdk/testing/suites.py
@@ -36,6 +36,7 @@
     TargetInvalidSchemaTest,
     TargetNoPrimaryKeys,
     TargetOptionalAttributes,
+    TargetPrimaryKeyUpdates,
     TargetRecordBeforeSchemaTest,
     TargetRecordMissingKeyProperty,
     TargetRecordMissingOptionalFields,
@@ -104,6 +105,7 @@ class TestSuite(t.Generic[T]):
         # TargetMultipleStateMessages,
         TargetNoPrimaryKeys,
         TargetOptionalAttributes,
+        TargetPrimaryKeyUpdates,
         TargetRecordBeforeSchemaTest,
         TargetRecordMissingKeyProperty,
         TargetRecordMissingOptionalFields,

diff --git a/singer_sdk/testing/target_test_streams/pk_updates.singer b/singer_sdk/testing/target_test_streams/pk_updates.singer
@@ -0,0 +1,4 @@
+{"type": "SCHEMA", "stream": "example_stream", "schema": {"properties": {"id": {"type": "integer"}, "name": {"type": "string"}, "email": {"type": "string"}}, "key_properties": ["id"]}}
+{"type": "RECORD", "stream": "example_stream", "record": {"id": 1, "name": "Alice", "email": "alice@example.com"}}
+{"type": "SCHEMA", "stream": "example_stream", "schema": {"properties": {"id": {"type": "integer"}, "name": {"type": "string"}, "email": {"type": "string"}}, "key_properties": ["email"]}}
+{"type": "RECORD", "stream": "example_stream", "record": {"id": 2, "name": "Bob", "email": "bob@example.com"}}
diff --git a/singer_sdk/testing/target_tests.py b/singer_sdk/testing/target_tests.py
@@ -135,6 +135,12 @@ class TargetSchemaUpdates(TargetFileTestTemplate):
     name = "schema_updates"
 
 
+class TargetPrimaryKeyUpdates(TargetFileTestTemplate):
+    """Test Target handles Primary Key updates."""
+
+    name = "pk_updates"
+
+
 class TargetSpecialCharsInAttributes(TargetFileTestTemplate):
     """Test Target handles special chars in attributes."""