Skip to content

Commit 52a548e

Browse files
committed
chore: parameterize test
1 parent 23a71f5 commit 52a548e

File tree

1 file changed

+3
-16
lines changed

1 file changed

+3
-16
lines changed

test/test_connection_.py

Lines changed: 3 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -17,22 +17,9 @@ def test_set_query_tag_normal_conn():
1717
assert conn.query_tagging
1818

1919

20-
def test_ignore_query_tag_for_hyper():
20+
@pytest.mark.parametrize("conn_type", ["hyper", "teradata", "snowflake"])
21+
def test_ignore_query_tag(conn_type):
2122
conn = TSC.ConnectionItem()
22-
conn._connection_type = "hyper"
23-
conn.query_tagging = True
24-
assert conn.query_tagging is None
25-
26-
27-
def test_ignore_query_tag_for_teradata():
28-
conn = TSC.ConnectionItem()
29-
conn._connection_type = "teradata"
30-
conn.query_tagging = True
31-
assert conn.query_tagging is None
32-
33-
34-
def test_ignore_query_tag_for_snowflake():
35-
conn = TSC.ConnectionItem()
36-
conn._connection_type = "snowflake"
23+
conn._connection_type = conn_type
3724
conn.query_tagging = True
3825
assert conn.query_tagging is None

0 commit comments

Comments
 (0)