From a415cb12a0ed3bbdb0cc51ac37ee6c84eae06ab6 Mon Sep 17 00:00:00 2001
From: airflow <airflow@example.com>
Date: Wed, 24 Sep 2025 07:03:09 +0400
Subject: [PATCH] =?UTF-8?q?=D0=9E=D0=B1=D0=BD=D0=BE=D0=B2=D0=B8=D1=82?=
 =?UTF-8?q?=D1=8C=20test=5Fclickhouse=5Fdag.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 test_clickhouse_dag.py | 31 +++++++++++++++----------------
 1 file changed, 15 insertions(+), 16 deletions(-)

diff --git a/test_clickhouse_dag.py b/test_clickhouse_dag.py
index 538247e..7e5ba8a 100644
--- a/test_clickhouse_dag.py
+++ b/test_clickhouse_dag.py
@@ -1,23 +1,22 @@
 from airflow import DAG
-from airflow.providers.clickhouse.operators.clickhouse import ClickHouseOperator
-from airflow.utils.dates import days_ago
+from airflow.operators.python import PythonOperator
+from airflow_clickhouse_plugin.hooks.clickhouse import ClickHouseHook
+from datetime import datetime
 
-default_args = {
-    "owner": "airflow",
-}
+def test_clickhouse():
+    hook = ClickHouseHook(clickhouse_conn_id="my_clickhouse_provider")
+    # Простейший запрос
+    result = hook.get_records("SELECT 1")
+    print("ClickHouse result:", result)
 
 with DAG(
-    dag_id="test_clickhouse_select",
-    default_args=default_args,
-    description="Simple SELECT 1 in ClickHouse",
-    start_date=days_ago(1),
-    schedule_interval=None,      # запуск только вручную
+    dag_id="test_clickhouse_dag",
+    start_date=datetime(2024, 1, 1),
+    schedule=None,
     catchup=False,
-    tags=["clickhouse"],
+    tags=["example"],
 ) as dag:
-
-    select_1 = ClickHouseOperator(
-        task_id="select_one",
-        clickhouse_conn_id="my_clickhouse_provider",  # ваш Connection ID
-        sql="SELECT 1",
+    test_task = PythonOperator(
+        task_id="check_clickhouse",
+        python_callable=test_clickhouse
     )