This is an automated email from the ASF dual-hosted git repository.
kaxil pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new cd947e3680a Remove provider tests that restate `template_fields` and
other class constants (#74359)
cd947e3680a is described below
commit cd947e3680af93ad42ed9aca745f6f1e0b2f3b68
Author: Kaxil Naik <[email protected]>
AuthorDate: Tue Oct 6 18:10:49 2026 +0100
Remove provider tests that restate `template_fields` and other class
constants (#74359)
These tests assert that a class attribute equals a copy of its own value,
so they fail on every intentional edit and pass on everything else. The
check-template-fields-valid prek hook already verifies that template
fields are set in __init__. Also drop two permanently skipped placeholders.
Single-field template guards and conn_type/default_conn_name/conn_name_attr
pins are kept: they pin decisions nothing else checks.
---
.../tests/unit/amazon/aws/operators/test_ecs.py | 26 --------
.../amazon/aws/operators/test_neptune_analytics.py | 6 --
.../unit/amazon/aws/transfers/test_mongo_to_s3.py | 10 ---
.../livy/tests/unit/apache/livy/hooks/test_livy.py | 5 --
.../unit/clickhousedb/hooks/test_clickhouse.py | 3 -
.../tests/unit/common/ai/hooks/test_pydantic_ai.py | 9 ---
.../ai/operators/test_llamaindex_retrieval.py | 13 ----
.../cloud/operators/test_cloud_logging_sink.py | 42 -------------
.../unit/google/cloud/operators/test_cloud_run.py | 73 ----------------------
.../azure/operators/test_analysis_services.py | 8 ---
.../unit/microsoft/azure/operators/test_compute.py | 9 ---
.../azure/sensors/test_analysis_services.py | 8 ---
.../unit/microsoft/azure/sensors/test_compute.py | 11 ----
.../tests/unit/postgres/hooks/test_postgres.py | 4 --
.../tests/unit/salesforce/operators/test_bulk.py | 17 -----
.../tests/unit/telegram/operators/test_telegram.py | 13 ----
.../tests/unit/teradata/operators/test_bteq.py | 4 --
17 files changed, 261 deletions(-)
diff --git a/providers/amazon/tests/unit/amazon/aws/operators/test_ecs.py
b/providers/amazon/tests/unit/amazon/aws/operators/test_ecs.py
index 1b0d710512b..dcb46ffa520 100644
--- a/providers/amazon/tests/unit/amazon/aws/operators/test_ecs.py
+++ b/providers/amazon/tests/unit/amazon/aws/operators/test_ecs.py
@@ -191,32 +191,6 @@ class TestEcsRunTaskOperator(EcsBaseTestCase):
assert fetcher.hook.region_name == "region"
- def test_template_fields_overrides(self):
- assert self.ecs.template_fields == (
- "task_definition",
- "cluster",
- "overrides",
- "launch_type",
- "capacity_provider_strategy",
- "volume_configurations",
- "group",
- "placement_constraints",
- "placement_strategy",
- "platform_version",
- "network_configuration",
- "tags",
- "awslogs_group",
- "awslogs_region",
- "awslogs_stream_prefix",
- "awslogs_fetch_interval",
- "container_name",
- "propagate_tags",
- "reattach",
- "number_logs_exception",
- "wait_for_completion",
- "deferrable",
- )
-
@pytest.mark.parametrize(
(
"launch_type",
diff --git
a/providers/amazon/tests/unit/amazon/aws/operators/test_neptune_analytics.py
b/providers/amazon/tests/unit/amazon/aws/operators/test_neptune_analytics.py
index ba927496608..72828e653c6 100644
--- a/providers/amazon/tests/unit/amazon/aws/operators/test_neptune_analytics.py
+++ b/providers/amazon/tests/unit/amazon/aws/operators/test_neptune_analytics.py
@@ -65,9 +65,6 @@ class TestNeptuneCreateGraphOperator:
assert "vector_search_config" in fields
assert "provisioned_memory" in fields
- def test_template_fields_renderers(self):
- assert NeptuneCreateGraphOperator.template_fields_renderers ==
{"vector_search_config": "json"}
-
def test_operator_extra_links(self):
assert len(NeptuneCreateGraphOperator.operator_extra_links) == 1
@@ -1176,9 +1173,6 @@ class TestNeptuneStartImportTaskOperator:
assert "source" in fields
assert "import_options" in fields
- def test_template_fields_renderers(self):
- assert NeptuneStartImportTaskOperator.template_fields_renderers ==
{"import_options": "json"}
-
def test_operator_extra_links(self):
assert len(NeptuneStartImportTaskOperator.operator_extra_links) == 1
assert
isinstance(NeptuneStartImportTaskOperator.operator_extra_links[0],
NeptuneImportTaskLink)
diff --git
a/providers/amazon/tests/unit/amazon/aws/transfers/test_mongo_to_s3.py
b/providers/amazon/tests/unit/amazon/aws/transfers/test_mongo_to_s3.py
index d286dd8b8cc..18cb99511b2 100644
--- a/providers/amazon/tests/unit/amazon/aws/transfers/test_mongo_to_s3.py
+++ b/providers/amazon/tests/unit/amazon/aws/transfers/test_mongo_to_s3.py
@@ -75,16 +75,6 @@ class TestMongoToS3Operator:
assert self.mock_operator.s3_key == S3_KEY
assert self.mock_operator.compression == COMPRESSION
- def test_template_field_overrides(self):
- assert self.mock_operator.template_fields == (
- "s3_bucket",
- "s3_key",
- "mongo_query",
- "mongo_collection",
- "mongo_conn_id",
- "aws_conn_id",
- )
-
@pytest.mark.db_test
def test_render_template(self, session, clean_dags_dagruns_and_dagbundles,
testing_dag_bundle):
if AIRFLOW_V_3_0_PLUS:
diff --git a/providers/apache/livy/tests/unit/apache/livy/hooks/test_livy.py
b/providers/apache/livy/tests/unit/apache/livy/hooks/test_livy.py
index 695f5b06e56..9a72edef770 100644
--- a/providers/apache/livy/tests/unit/apache/livy/hooks/test_livy.py
+++ b/providers/apache/livy/tests/unit/apache/livy/hooks/test_livy.py
@@ -110,11 +110,6 @@ class TestLivyDbHook:
hook.get_conn()
assert hook.base_url == expected
- @pytest.mark.skip("Inherited HttpHook does not handle missing hostname")
- def test_missing_host(self):
- with pytest.raises(AirflowException):
- LivyHook(livy_conn_id="missing_host").get_conn()
-
def test_build_body_minimal_request(self):
assert LivyHook.build_post_batch_body(file="appname") == {"file":
"appname"}
diff --git
a/providers/clickhousedb/tests/unit/clickhousedb/hooks/test_clickhouse.py
b/providers/clickhousedb/tests/unit/clickhousedb/hooks/test_clickhouse.py
index 00f1911e425..accf3a400c9 100644
--- a/providers/clickhousedb/tests/unit/clickhousedb/hooks/test_clickhouse.py
+++ b/providers/clickhousedb/tests/unit/clickhousedb/hooks/test_clickhouse.py
@@ -368,9 +368,6 @@ class TestClickHouseHookClassAttributes:
def test_conn_type(self):
assert ClickHouseHook.conn_type == "clickhouse"
- def test_hook_name(self):
- assert ClickHouseHook.hook_name == "ClickHouse"
-
def test_supports_autocommit(self):
assert ClickHouseHook.supports_autocommit is True
diff --git a/providers/common/ai/tests/unit/common/ai/hooks/test_pydantic_ai.py
b/providers/common/ai/tests/unit/common/ai/hooks/test_pydantic_ai.py
index 5f856c1d5c0..f0c69bee7c7 100644
--- a/providers/common/ai/tests/unit/common/ai/hooks/test_pydantic_ai.py
+++ b/providers/common/ai/tests/unit/common/ai/hooks/test_pydantic_ai.py
@@ -1453,9 +1453,6 @@ class TestPydanticAIAzureHook:
def test_conn_type(self):
assert PydanticAIAzureHook.conn_type == "pydanticai_azure"
- def test_hook_name(self):
- assert "Azure" in PydanticAIAzureHook.hook_name
-
def test_ui_metadata(self):
expected_placeholders = {
"host": "https://<resource>.openai.azure.com/openai/v1",
@@ -1586,9 +1583,6 @@ class TestPydanticAIBedrockHook:
def test_conn_type(self):
assert PydanticAIBedrockHook.conn_type == "pydanticai_bedrock"
- def test_hook_name(self):
- assert "Bedrock" in PydanticAIBedrockHook.hook_name
-
def test_ui_hides_host_and_password(self):
behaviour = PydanticAIBedrockHook.get_ui_field_behaviour()
assert "host" in behaviour["hidden_fields"]
@@ -1716,9 +1710,6 @@ class TestPydanticAIVertexHook:
def test_conn_type(self):
assert PydanticAIVertexHook.conn_type == "pydanticai_vertex"
- def test_hook_name(self):
- assert "Vertex" in PydanticAIVertexHook.hook_name
-
def test_ui_hides_host_and_password(self):
behaviour = PydanticAIVertexHook.get_ui_field_behaviour()
assert "host" in behaviour["hidden_fields"]
diff --git
a/providers/common/ai/tests/unit/common/ai/operators/test_llamaindex_retrieval.py
b/providers/common/ai/tests/unit/common/ai/operators/test_llamaindex_retrieval.py
index 6c1378c5c6f..320211e3eed 100644
---
a/providers/common/ai/tests/unit/common/ai/operators/test_llamaindex_retrieval.py
+++
b/providers/common/ai/tests/unit/common/ai/operators/test_llamaindex_retrieval.py
@@ -58,19 +58,6 @@ def _byo_embedding():
return MagicMock(name="MyBaseEmbedding", spec=["get_text_embedding",
"_get_query_embedding"])
-class TestRetrievalOperatorInit:
- def test_template_fields(self):
- assert set(LlamaIndexRetrievalOperator.template_fields) == {
- "query",
- "index_persist_dir",
- "persist_conn_id",
- "embed_model",
- "llm_conn_id",
- "embed_conn_id",
- "embedding_kwargs",
- }
-
-
class TestRetrievalOperatorOutput:
@patch("airflow.providers.common.ai.hooks.llamaindex.LlamaIndexHook.get_embedding_model")
def test_chunk_shape(self, mock_get_embed, _li, tmp_path):
diff --git
a/providers/google/tests/unit/google/cloud/operators/test_cloud_logging_sink.py
b/providers/google/tests/unit/google/cloud/operators/test_cloud_logging_sink.py
index 6b4e3cd2b38..9ef69e8e36d 100644
---
a/providers/google/tests/unit/google/cloud/operators/test_cloud_logging_sink.py
+++
b/providers/google/tests/unit/google/cloud/operators/test_cloud_logging_sink.py
@@ -130,19 +130,7 @@ update_test_ids = ["update_storage_sink",
"update_pubsub_sink"]
sink = LogSink(name=SINK_NAME,
destination="pubsub.googleapis.com/projects/my-project/topics/my-topic")
-def _assert_common_template_fields(template_fields):
- assert "project_id" in template_fields
- assert "gcp_conn_id" in template_fields
- assert "impersonation_chain" in template_fields
-
-
class TestCloudLoggingCreateSinkOperator:
- def test_template_fields(self):
- operator = CloudLoggingCreateSinkOperator(task_id=TASK_ID,
project_id=PROJECT_ID, sink_config=sink)
- assert "sink_config" in operator.template_fields
- assert "unique_writer_identity" in operator.template_fields
- _assert_common_template_fields(operator.template_fields)
-
def test_missing_required_params(self):
with pytest.raises(AirflowException) as excinfo:
CloudLoggingCreateSinkOperator(
@@ -307,15 +295,6 @@ class TestCloudLoggingCreateSinkOperator:
class TestCloudLoggingDeleteSinkOperator:
- def test_template_fields(self):
- operator = CloudLoggingDeleteSinkOperator(
- task_id=TASK_ID,
- sink_name=SINK_NAME,
- project_id=PROJECT_ID,
- )
- assert "sink_name" in operator.template_fields
- _assert_common_template_fields(operator.template_fields)
-
def test_missing_required_params(self):
with pytest.raises(AirflowException) as excinfo:
CloudLoggingDeleteSinkOperator(
@@ -414,13 +393,6 @@ class TestCloudLoggingDeleteSinkOperator:
class TestCloudLoggingListSinksOperator:
- def test_template_fields(self):
- operator = CloudLoggingListSinksOperator(
- task_id=TASK_ID,
- project_id=PROJECT_ID,
- )
- assert "project_id" in operator.template_fields
-
def test_missing_required_params(self):
with pytest.raises(AirflowException) as excinfo:
CloudLoggingListSinksOperator(
@@ -512,20 +484,6 @@ class TestCloudLoggingListSinksOperator:
class TestCloudLoggingUpdateSinksOperator:
- @pytest.mark.parametrize(("sink_config", "update_mask"),
update_test_cases, ids=update_test_ids)
- def test_template_fields(self, sink_config, update_mask):
- operator = CloudLoggingUpdateSinkOperator(
- task_id=TASK_ID,
- sink_name=SINK_NAME,
- sink_config=sink_config,
- update_mask=update_mask,
- project_id=PROJECT_ID,
- )
- assert "sink_config" in operator.template_fields
- assert "update_mask" in operator.template_fields
- assert "sink_name" in operator.template_fields
- _assert_common_template_fields(operator.template_fields)
-
def test_missing_required_params(self):
with pytest.raises(AirflowException) as excinfo:
CloudLoggingDeleteSinkOperator(
diff --git
a/providers/google/tests/unit/google/cloud/operators/test_cloud_run.py
b/providers/google/tests/unit/google/cloud/operators/test_cloud_run.py
index 48e3a6358d6..720b90c4142 100644
--- a/providers/google/tests/unit/google/cloud/operators/test_cloud_run.py
+++ b/providers/google/tests/unit/google/cloud/operators/test_cloud_run.py
@@ -66,22 +66,7 @@ SERVICE = Service()
SERVICE.name = SERVICE_NAME
-def _assert_common_template_fields(template_fields):
- assert "project_id" in template_fields
- assert "region" in template_fields
- assert "gcp_conn_id" in template_fields
- assert "impersonation_chain" in template_fields
-
-
class TestCloudRunCreateJobOperator:
- def test_template_fields(self):
- operator = CloudRunCreateJobOperator(
- task_id=TASK_ID, project_id=PROJECT_ID, region=REGION,
job_name=JOB_NAME, job=JOB
- )
-
- _assert_common_template_fields(operator.template_fields)
- assert "job_name" in operator.template_fields
-
@mock.patch(CLOUD_RUN_HOOK_PATH)
def test_create(self, hook_mock):
hook_mock.return_value.create_job.return_value = JOB
@@ -102,18 +87,6 @@ class TestCloudRunCreateJobOperator:
class TestCloudRunExecuteJobOperator:
- def test_template_fields(self):
- operator = CloudRunExecuteJobOperator(
- task_id=TASK_ID, project_id=PROJECT_ID, region=REGION,
job_name=JOB_NAME, overrides=OVERRIDES
- )
-
- _assert_common_template_fields(operator.template_fields)
- assert "job_name" in operator.template_fields
- assert "overrides" in operator.template_fields
- assert "polling_period_seconds" in operator.template_fields
- assert "timeout_seconds" in operator.template_fields
- assert "transport" in operator.template_fields
-
@mock.patch(CLOUD_RUN_HOOK_PATH)
def test_execute_with_transport(self, hook_mock):
"""Test that transport parameter is passed to CloudRunHook."""
@@ -778,14 +751,6 @@ class TestCloudRunExecuteJobOperator:
class TestCloudRunDeleteJobOperator:
- def test_template_fields(self):
- operator = CloudRunDeleteJobOperator(
- task_id=TASK_ID, project_id=PROJECT_ID, region=REGION,
job_name=JOB_NAME
- )
-
- _assert_common_template_fields(operator.template_fields)
- assert "job_name" in operator.template_fields
-
@mock.patch(CLOUD_RUN_HOOK_PATH)
def test_execute(self, hook_mock):
hook_mock.return_value.delete_job.return_value = JOB
@@ -807,14 +772,6 @@ class TestCloudRunDeleteJobOperator:
class TestCloudRunUpdateJobOperator:
- def test_template_fields(self):
- operator = CloudRunUpdateJobOperator(
- task_id=TASK_ID, project_id=PROJECT_ID, region=REGION,
job_name=JOB_NAME, job=JOB
- )
-
- _assert_common_template_fields(operator.template_fields)
- assert "job_name" in operator.template_fields
-
@mock.patch(CLOUD_RUN_HOOK_PATH)
def test_execute(self, hook_mock):
hook_mock.return_value.update_job.return_value = JOB
@@ -837,13 +794,6 @@ class TestCloudRunUpdateJobOperator:
class TestCloudRunListJobsOperator:
- def test_template_fields(self):
- operator = CloudRunListJobsOperator(
- task_id=TASK_ID, project_id=PROJECT_ID, region=REGION, limit=2,
show_deleted=False
- )
-
- _assert_common_template_fields(operator.template_fields)
-
@mock.patch(CLOUD_RUN_HOOK_PATH)
def test_execute(self, hook_mock):
limit = 2
@@ -870,18 +820,6 @@ class TestCloudRunListJobsOperator:
class TestCloudRunCreateServiceOperator:
- def test_template_fields(self):
- operator = CloudRunCreateServiceOperator(
- task_id=TASK_ID,
- project_id=PROJECT_ID,
- region=REGION,
- service=SERVICE,
- service_name=SERVICE_NAME,
- )
-
- _assert_common_template_fields(operator.template_fields)
- assert "service_name" in operator.template_fields
-
@mock.patch(CLOUD_RUN_SERVICE_HOOK_PATH)
def test_execute(self, hook_mock):
hook_mock.return_value.create_service.return_value = SERVICE
@@ -961,17 +899,6 @@ class TestCloudRunCreateServiceOperator:
class TestCloudRunDeleteServiceOperator:
- def test_template_fields(self):
- operator = CloudRunDeleteServiceOperator(
- task_id=TASK_ID,
- project_id=PROJECT_ID,
- region=REGION,
- service_name=SERVICE_NAME,
- )
-
- _assert_common_template_fields(operator.template_fields)
- assert "service_name" in operator.template_fields
-
@mock.patch(CLOUD_RUN_SERVICE_HOOK_PATH)
def test_execute(self, hook_mock):
hook_mock.return_value.delete_service.return_value = SERVICE
diff --git
a/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_analysis_services.py
b/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_analysis_services.py
index 7bf216c50a3..219c4c52e02 100644
---
a/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_analysis_services.py
+++
b/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_analysis_services.py
@@ -99,14 +99,6 @@ class TestAzureAnalysisServicesRefreshOperator:
request_timeout=request_timeout,
)
- def test_defines_template_fields(self):
- assert AzureAnalysisServicesRefreshOperator.template_fields == (
- "azure_analysis_services_conn_id",
- "server_name",
- "database",
- "refresh_type",
- )
-
def test_execute_defers_without_refresh_id(self):
with pytest.raises(TaskDeferred) as deferred:
build_operator().execute(context=build_context())
diff --git
a/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_compute.py
b/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_compute.py
index 879dd219da7..965af485c11 100644
---
a/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_compute.py
+++
b/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_compute.py
@@ -43,15 +43,6 @@ class TestAzureVirtualMachineStartOperator:
assert op.wait_for_completion is True
assert op.azure_conn_id == CONN_ID
- def test_template_fields(self):
- op = AzureVirtualMachineStartOperator(
- task_id="start_vm",
- resource_group_name=RESOURCE_GROUP,
- vm_name=VM_NAME,
- )
- assert "resource_group_name" in op.template_fields
- assert "vm_name" in op.template_fields
-
@patch("airflow.providers.microsoft.azure.operators.compute.AzureComputeHook")
def test_execute_start_instance(self, mock_hook_cls):
op = AzureVirtualMachineStartOperator(
diff --git
a/providers/microsoft/azure/tests/unit/microsoft/azure/sensors/test_analysis_services.py
b/providers/microsoft/azure/tests/unit/microsoft/azure/sensors/test_analysis_services.py
index a86854aab70..e5687f167b2 100644
---
a/providers/microsoft/azure/tests/unit/microsoft/azure/sensors/test_analysis_services.py
+++
b/providers/microsoft/azure/tests/unit/microsoft/azure/sensors/test_analysis_services.py
@@ -62,14 +62,6 @@ class TestAzureAnalysisServicesSensor:
with pytest.raises(ValueError, match="request_timeout must be greater
than zero"):
build_sensor(request_timeout=0)
- def test_defines_template_fields(self):
- assert AzureAnalysisServicesSensor.template_fields == (
- "azure_analysis_services_conn_id",
- "server_name",
- "database",
- "refresh_id",
- )
-
def test_execute_defers_with_provided_refresh_id(self):
with pytest.raises(TaskDeferred) as deferred:
build_sensor().execute(context={})
diff --git
a/providers/microsoft/azure/tests/unit/microsoft/azure/sensors/test_compute.py
b/providers/microsoft/azure/tests/unit/microsoft/azure/sensors/test_compute.py
index 5d3c93b0305..705da5aa114 100644
---
a/providers/microsoft/azure/tests/unit/microsoft/azure/sensors/test_compute.py
+++
b/providers/microsoft/azure/tests/unit/microsoft/azure/sensors/test_compute.py
@@ -66,17 +66,6 @@ class TestAzureVirtualMachineStateSensor:
)
assert sensor.target_state == "{{ params.state }}"
- def test_template_fields(self):
- sensor = AzureVirtualMachineStateSensor(
- task_id="sense_vm",
- resource_group_name=RESOURCE_GROUP,
- vm_name=VM_NAME,
- target_state="running",
- )
- assert "resource_group_name" in sensor.template_fields
- assert "vm_name" in sensor.template_fields
- assert "target_state" in sensor.template_fields
-
@pytest.mark.parametrize(
("return_value", "expected"),
[
diff --git a/providers/postgres/tests/unit/postgres/hooks/test_postgres.py
b/providers/postgres/tests/unit/postgres/hooks/test_postgres.py
index 55fcb03d106..6056fcd4f16 100644
--- a/providers/postgres/tests/unit/postgres/hooks/test_postgres.py
+++ b/providers/postgres/tests/unit/postgres/hooks/test_postgres.py
@@ -1334,10 +1334,6 @@ class
TestPostgresHookPPG3(_BasePostgresHookRuntimeTests):
)
self.conn.commit.assert_called_once()
- @pytest.mark.skip(reason="Notice handling is callback-based in psycopg3
and cannot be tested this way.")
- def test_get_all_db_log_messages(self, mocker):
- pass
-
@pytest.mark.usefixtures("reset_logging_config")
def test_log_db_messages_by_db_proc(self, mocker):
proc_name = "raise_notice"
diff --git a/providers/salesforce/tests/unit/salesforce/operators/test_bulk.py
b/providers/salesforce/tests/unit/salesforce/operators/test_bulk.py
index 309680f660c..b480ff66746 100644
--- a/providers/salesforce/tests/unit/salesforce/operators/test_bulk.py
+++ b/providers/salesforce/tests/unit/salesforce/operators/test_bulk.py
@@ -29,23 +29,6 @@ class TestSalesforceBulkOperator:
Test class for SalesforceBulkOperator
"""
- def test_template_fields(self):
- """
- Test that template_fields are correctly defined and renderable.
- """
- operator = SalesforceBulkOperator(
- task_id="test_template_fields",
- operation="insert",
- object_name="Account",
- payload=[],
- )
- assert operator.template_fields == (
- "object_name",
- "payload",
- "external_id_field",
- "salesforce_conn_id",
- )
-
@pytest.mark.db_test
def test_template_rendering(self, create_task_instance_of_operator):
"""
diff --git a/providers/telegram/tests/unit/telegram/operators/test_telegram.py
b/providers/telegram/tests/unit/telegram/operators/test_telegram.py
index 783877e8957..2f36c1e07a5 100644
--- a/providers/telegram/tests/unit/telegram/operators/test_telegram.py
+++ b/providers/telegram/tests/unit/telegram/operators/test_telegram.py
@@ -138,16 +138,6 @@ class TestTelegramOperator:
{"custom_arg": "value", "text": "some non empty text - higher
precedence"},
)
- def test_should_return_template_fields(self):
- hook = TelegramOperator(
- telegram_conn_id="telegram_default",
- chat_id="-420913222",
- task_id="telegram",
- text="some non empty text - higher precedence",
- telegram_kwargs={"custom_arg": "value", "text": "some text, that
will be ignored"},
- )
- assert hook.template_fields == ("text", "chat_id", "telegram_conn_id")
-
@mock.patch("airflow.providers.telegram.operators.telegram.TelegramHook")
def test_should_return_templatized_text_field(self, mock_hook):
operator = TelegramOperator(
@@ -188,9 +178,6 @@ class TestTelegramOperator:
class TestTelegramFileOperator:
- def test_should_return_template_fields(self):
- assert TelegramFileOperator.template_fields == ("chat_id",
"telegram_conn_id")
-
def test_should_return_templatized_chat_id_field(self):
operator = TelegramFileOperator(
telegram_conn_id="telegram_default",
diff --git a/providers/teradata/tests/unit/teradata/operators/test_bteq.py
b/providers/teradata/tests/unit/teradata/operators/test_bteq.py
index e258a0ac8b1..5935a0402fa 100644
--- a/providers/teradata/tests/unit/teradata/operators/test_bteq.py
+++ b/providers/teradata/tests/unit/teradata/operators/test_bteq.py
@@ -142,10 +142,6 @@ class TestBteqOperator:
# When/Then (no exception should be raised)
operator.on_kill()
- def test_template_fields(self):
- # Verify template fields are defined correctly
- assert BteqOperator.template_fields == ("sql", "teradata_conn_id",
"ssh_conn_id")
-
def test_execute_raises_if_no_sql_or_file(self):
op = BteqOperator(task_id="fail_case", teradata_conn_id="td_conn")
with pytest.raises(
