Fix Kafka provider tests after move (apache#46214)

got686-yandex · Jan 30, 2025 · 92f3a7f · 92f3a7f
1 parent 72850b7
commit 92f3a7f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/providers/apache/kafka/tests/provider_tests/apache/kafka/operators/test_consume.py b/providers/apache/kafka/tests/provider_tests/apache/kafka/operators/test_consume.py
@@ -64,7 +64,7 @@ def test_operator(self):
         operator = ConsumeFromTopicOperator(
             kafka_config_id="kafka_d",
             topics=["test"],
-            apply_function="providers.tests.apache.kafka.operators.test_consume._no_op",
+            apply_function="provider_tests.apache.kafka.operators.test_consume._no_op",
             task_id="test",
             poll_timeout=0.0001,
         )

diff --git a/providers/apache/kafka/tests/provider_tests/apache/kafka/operators/test_produce.py b/providers/apache/kafka/tests/provider_tests/apache/kafka/operators/test_produce.py
@@ -69,7 +69,7 @@ def test_operator_string(self):
         operator = ProduceToTopicOperator(
             kafka_config_id="kafka_d",
             topic="test_1",
-            producer_function="providers.tests.apache.kafka.operators.test_produce._simple_producer",
+            producer_function="provider_tests.apache.kafka.operators.test_produce._simple_producer",
             producer_function_args=(b"test", b"test"),
             task_id="test",
             synchronous=False,

diff --git a/providers/apache/kafka/tests/provider_tests/apache/kafka/triggers/test_await_message.py b/providers/apache/kafka/tests/provider_tests/apache/kafka/triggers/test_await_message.py
@@ -100,7 +100,7 @@ async def test_trigger_run_good(self, mocker):
 
         trigger = AwaitMessageTrigger(
             kafka_config_id="kafka_d",
-            apply_function="providers.tests.apache.kafka.triggers.test_await_message.apply_function_true",
+            apply_function="provider_tests.apache.kafka.triggers.test_await_message.apply_function_true",
             topics=["noop"],
             poll_timeout=0.0001,
             poll_interval=5,
@@ -117,7 +117,7 @@ async def test_trigger_run_bad(self, mocker):
 
         trigger = AwaitMessageTrigger(
             kafka_config_id="kafka_d",
-            apply_function="providers.tests.apache.kafka.triggers.test_await_message.apply_function_false",
+            apply_function="provider_tests.apache.kafka.triggers.test_await_message.apply_function_false",
             topics=["noop"],
             poll_timeout=0.0001,
             poll_interval=5,