From 92f3a7fe408eb6a6a9ee6dec130875a29f465322 Mon Sep 17 00:00:00 2001
From: Jens Scheffler <95105677+jscheffl@users.noreply.github.com>
Date: Tue, 28 Jan 2025 23:10:27 +0100
Subject: [PATCH] Fix Kafka provider tests after move (#46214)

---
 .../provider_tests/apache/kafka/operators/test_consume.py     | 2 +-
 .../provider_tests/apache/kafka/operators/test_produce.py     | 2 +-
 .../apache/kafka/triggers/test_await_message.py               | 4 ++--
 3 files changed, 4 insertions(+), 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
index 7e02a273894b30..f070923890727e 100644
--- 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
index ed41a18a1705ff..c6f8bb54b2e73a 100644
--- 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
index 0f2b3f6f6b0746..1684ec867356d3 100644
--- 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,