From e003a5f9ed3b9253898a42f5733db21c4c1d89fe Mon Sep 17 00:00:00 2001 From: Imri Paran Date: Fri, 24 Jan 2025 15:01:54 +0100 Subject: [PATCH] tests(e2e): data quality (#19513) * tests(e2e): data quality - use the data quality pipeline new schema with a list of service configurations * format --- .../cli_e2e/base/config_builders/builders.py | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/ingestion/tests/cli_e2e/base/config_builders/builders.py b/ingestion/tests/cli_e2e/base/config_builders/builders.py index 3368fcb6ed0..a1a213eba21 100644 --- a/ingestion/tests/cli_e2e/base/config_builders/builders.py +++ b/ingestion/tests/cli_e2e/base/config_builders/builders.py @@ -139,14 +139,20 @@ class DataQualityConfigBuilder(BaseBuilder): def build(self) -> dict: """build profiler config""" - del self.config["source"]["sourceConfig"]["config"] - self.config["source"]["sourceConfig"] = { - "config": { - "type": TestSuiteConfigType.TestSuite.value, - "entityFullyQualifiedName": self.entity_fqn, - }, + self.config["source"]["sourceConfig"]["config"] = { + "type": TestSuiteConfigType.TestSuite.value, + "entityFullyQualifiedName": self.entity_fqn, } + self.config["source"]["sourceConfig"]["config"]["serviceConnections"] = [ + { + "serviceName": self.config["source"]["serviceName"], + "serviceConnection": self.config["source"]["serviceConnection"], + } + ] + + del self.config["source"]["serviceConnection"] + self.config["processor"] = { "type": "orm-test-runner", "config": {"testCases": self.test_case_defintions},