From fb384aa1393cd488f2b0c150914db05bc86f76f8 Mon Sep 17 00:00:00 2001 From: Antoni Ivanov Date: Tue, 10 Oct 2023 17:58:14 +0300 Subject: [PATCH] vdk-test-utils: make IngestIntoMemoryPlugin method configurable (#2783) IngestIntoMemoryPlugin is used in tests to mock ingestion targets. It would be useful the method name to be configruable so it can be used as different methods --- .../vdk-test-utils/src/vdk/plugin/test_utils/util_plugins.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/projects/vdk-plugins/vdk-test-utils/src/vdk/plugin/test_utils/util_plugins.py b/projects/vdk-plugins/vdk-test-utils/src/vdk/plugin/test_utils/util_plugins.py index 3fd075c919..679ca2a33b 100644 --- a/projects/vdk-plugins/vdk-test-utils/src/vdk/plugin/test_utils/util_plugins.py +++ b/projects/vdk-plugins/vdk-test-utils/src/vdk/plugin/test_utils/util_plugins.py @@ -201,8 +201,9 @@ class Payload: target: Optional[str] collection_id: Optional[str] - def __init__(self): + def __init__(self, method_name="memory"): self.payloads: List[IngestIntoMemoryPlugin.Payload] = [] + self.method_name = method_name def ingest_payload( self, @@ -222,4 +223,4 @@ def ingest_payload( def initialize_job(self, context: JobContext) -> None: log.info("Initialize data job with IngestIntoMemory Plugin.") - context.ingester.add_ingester_factory_method("memory", lambda: self) + context.ingester.add_ingester_factory_method(self.method_name, lambda: self)