vdk-test-utils: adopt ingester changes (#3119)

Adopt ingester changes in test utils. Ref: 03dde45 Signed-off-by: Dako Dakov <[email protected]>
vmware · Feb 15, 2024 · 395beed · 395beed
1 parent 28ed822
commit 395beed
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/projects/vdk-plugins/vdk-test-utils/src/vdk/plugin/test_utils/ingest_util_plugins.py b/projects/vdk-plugins/vdk-test-utils/src/vdk/plugin/test_utils/ingest_util_plugins.py
@@ -26,6 +26,7 @@ def pre_ingest_process(
         target: Optional[str] = None,
         collection_id: Optional[str] = None,
         metadata: IngestionMetadata = None,
+        method: Optional[str] = None,
     ) -> Tuple[List[Dict], Optional[IngestionMetadata]]:
         processed_payload = [{k: str(v) for (k, v) in i.items()} for i in payload]
         return processed_payload, metadata
@@ -56,6 +57,7 @@ def pre_ingest_process(
         destination_table: Optional[str] = None,
         target: Optional[str] = None,
         collection_id: Optional[str] = None,
+        method: Optional[str] = None,
     ) -> Tuple[List[Dict], Optional[IngestionMetadata]]:
         processed_payloads = []
         start_time = datetime.datetime(year=2022, month=1, day=27, hour=16)
@@ -91,6 +93,7 @@ def post_ingest_process(
         destination_table: Optional[str] = None,
         target: Optional[str] = None,
         collection_id: Optional[str] = None,
+        method: Optional[str] = None,
     ) -> Optional[IngestionMetadata]:
         start_time = None
         if metadata: