From c866e38454e304c94a3947becba8300e198dd9c8 Mon Sep 17 00:00:00 2001
From: Sema Checherinda <Sema.Checherinda@clickhouse.com>
Date: Wed, 4 Dec 2024 22:47:45 +0100
Subject: [PATCH] fix test

---
 tests/queries/0_stateless/03279_with_clickhouse_driver.py | 7 +++++--
 tests/queries/0_stateless/03279_with_clickhouse_driver.sh | 8 --------
 2 files changed, 5 insertions(+), 10 deletions(-)
 mode change 100644 => 100755 tests/queries/0_stateless/03279_with_clickhouse_driver.py
 delete mode 100755 tests/queries/0_stateless/03279_with_clickhouse_driver.sh

diff --git a/tests/queries/0_stateless/03279_with_clickhouse_driver.py b/tests/queries/0_stateless/03279_with_clickhouse_driver.py
old mode 100644
new mode 100755
index 15c97436854..43400e48f36
--- a/tests/queries/0_stateless/03279_with_clickhouse_driver.py
+++ b/tests/queries/0_stateless/03279_with_clickhouse_driver.py
@@ -1,4 +1,7 @@
-import sys
+#!/usr/bin/env python3
+# Tags: no-fasttest
+
+import os
 from clickhouse_driver import Client
 
 
@@ -11,5 +14,5 @@ def run(database):
 
 
 if __name__ == "__main__":
-    database = sys.argv[1]
+    database =  os.environ["CLICKHOUSE_DATABASE"]
     run(database)
diff --git a/tests/queries/0_stateless/03279_with_clickhouse_driver.sh b/tests/queries/0_stateless/03279_with_clickhouse_driver.sh
deleted file mode 100755
index 40493441d0d..00000000000
--- a/tests/queries/0_stateless/03279_with_clickhouse_driver.sh
+++ /dev/null
@@ -1,8 +0,0 @@
-#!/usr/bin/env bash
-# Tags: no-fasttest
-
-CURDIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
-# shellcheck source=../shell_config.sh
-. "$CURDIR"/../shell_config.sh
-
-python3 03279_with_clickhouse_driver.py $CLICKHOUSE_DATABASE
\ No newline at end of file