diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 19d84650118e0..213494c1390c6 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -593,6 +593,7 @@ repos: ^providers/apache/beam/.*\.py$| ^providers/apache/flink/.*\.py$| ^providers/apache/iceberg/.*\.py$| + ^providers/apache/kafka/.*\.py$| ^providers/cloudant/.*\.py$| ^providers/cohere/.*\.py$| ^providers/common/compat/.*\.py$| diff --git a/providers/apache/kafka/tests/unit/apache/kafka/hooks/test_client.py b/providers/apache/kafka/tests/unit/apache/kafka/hooks/test_client.py index 568032b495798..d56bc17b20707 100644 --- a/providers/apache/kafka/tests/unit/apache/kafka/hooks/test_client.py +++ b/providers/apache/kafka/tests/unit/apache/kafka/hooks/test_client.py @@ -27,8 +27,6 @@ from airflow.models import Connection from airflow.providers.apache.kafka.hooks.client import KafkaAdminClientHook -pytestmark = pytest.mark.db_test - log = logging.getLogger(__name__) diff --git a/providers/apache/kafka/tests/unit/apache/kafka/hooks/test_consume.py b/providers/apache/kafka/tests/unit/apache/kafka/hooks/test_consume.py index cb0b9d0846c50..0bad37aeac32c 100644 --- a/providers/apache/kafka/tests/unit/apache/kafka/hooks/test_consume.py +++ b/providers/apache/kafka/tests/unit/apache/kafka/hooks/test_consume.py @@ -27,8 +27,6 @@ # Import Hook from airflow.providers.apache.kafka.hooks.consume import KafkaConsumerHook -pytestmark = pytest.mark.db_test - class TestConsumerHook: """ diff --git a/providers/apache/kafka/tests/unit/apache/kafka/hooks/test_produce.py b/providers/apache/kafka/tests/unit/apache/kafka/hooks/test_produce.py index 6e5074bddb1ee..7b48026a47613 100644 --- a/providers/apache/kafka/tests/unit/apache/kafka/hooks/test_produce.py +++ b/providers/apache/kafka/tests/unit/apache/kafka/hooks/test_produce.py @@ -26,9 +26,6 @@ from airflow.models import Connection from airflow.providers.apache.kafka.hooks.produce import KafkaProducerHook -pytestmark = pytest.mark.db_test - - log = logging.getLogger(__name__) diff --git a/providers/apache/kafka/tests/unit/apache/kafka/operators/test_consume.py b/providers/apache/kafka/tests/unit/apache/kafka/operators/test_consume.py index 35221ac38065a..facd0be92c976 100644 --- a/providers/apache/kafka/tests/unit/apache/kafka/operators/test_consume.py +++ b/providers/apache/kafka/tests/unit/apache/kafka/operators/test_consume.py @@ -28,9 +28,6 @@ # Import Operator from airflow.providers.apache.kafka.operators.consume import ConsumeFromTopicOperator -pytestmark = pytest.mark.db_test - - log = logging.getLogger(__name__) diff --git a/providers/apache/kafka/tests/unit/apache/kafka/operators/test_produce.py b/providers/apache/kafka/tests/unit/apache/kafka/operators/test_produce.py index 94017ea962abb..92f408408aa58 100644 --- a/providers/apache/kafka/tests/unit/apache/kafka/operators/test_produce.py +++ b/providers/apache/kafka/tests/unit/apache/kafka/operators/test_produce.py @@ -25,9 +25,6 @@ from airflow.models import Connection from airflow.providers.apache.kafka.operators.produce import ProduceToTopicOperator -pytestmark = pytest.mark.db_test - - log = logging.getLogger(__name__) diff --git a/providers/apache/kafka/tests/unit/apache/kafka/sensors/test_kafka.py b/providers/apache/kafka/tests/unit/apache/kafka/sensors/test_kafka.py index a4cc68d512e57..817b8bdb2fe4d 100644 --- a/providers/apache/kafka/tests/unit/apache/kafka/sensors/test_kafka.py +++ b/providers/apache/kafka/tests/unit/apache/kafka/sensors/test_kafka.py @@ -26,9 +26,6 @@ from airflow.models import Connection from airflow.providers.apache.kafka.sensors.kafka import AwaitMessageSensor, AwaitMessageTriggerFunctionSensor -pytestmark = pytest.mark.db_test - - log = logging.getLogger(__name__) diff --git a/providers/apache/kafka/tests/unit/apache/kafka/triggers/test_await_message.py b/providers/apache/kafka/tests/unit/apache/kafka/triggers/test_await_message.py index 194c772a80213..cd9380e530893 100644 --- a/providers/apache/kafka/tests/unit/apache/kafka/triggers/test_await_message.py +++ b/providers/apache/kafka/tests/unit/apache/kafka/triggers/test_await_message.py @@ -27,8 +27,6 @@ from tests_common.test_utils.version_compat import AIRFLOW_V_3_0_PLUS, get_base_airflow_version_tuple -pytestmark = pytest.mark.db_test - def apply_function_false(message): return False diff --git a/providers/apache/kafka/tests/unit/apache/kafka/triggers/test_msg_queue.py b/providers/apache/kafka/tests/unit/apache/kafka/triggers/test_msg_queue.py index 7e960ca1599e3..8b3ae12344630 100644 --- a/providers/apache/kafka/tests/unit/apache/kafka/triggers/test_msg_queue.py +++ b/providers/apache/kafka/tests/unit/apache/kafka/triggers/test_msg_queue.py @@ -30,8 +30,6 @@ from airflow.providers.apache.kafka.triggers.msg_queue import KafkaMessageQueueTrigger from airflow.providers.common.messaging.triggers.msg_queue import MessageQueueTrigger -pytestmark = pytest.mark.db_test - def apply_function_false(message): return False