Skip to content

Commit 23a71f5

Browse files
committed
chore: pytestify test_connection_
Embrace pytest's testing methodology in test_connection_
1 parent 9b84601 commit 23a71f5

File tree

1 file changed

+35
-31
lines changed

1 file changed

+35
-31
lines changed

test/test_connection_.py

Lines changed: 35 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -1,34 +1,38 @@
1-
import unittest
21
import tableauserverclient as TSC
32

3+
import pytest
44

5-
class DatasourceModelTests(unittest.TestCase):
6-
def test_require_boolean_query_tag_fails(self):
7-
conn = TSC.ConnectionItem()
8-
conn._connection_type = "postgres"
9-
with self.assertRaises(ValueError):
10-
conn.query_tagging = "no"
11-
12-
def test_set_query_tag_normal_conn(self):
13-
conn = TSC.ConnectionItem()
14-
conn._connection_type = "postgres"
15-
conn.query_tagging = True
16-
self.assertEqual(conn.query_tagging, True)
17-
18-
def test_ignore_query_tag_for_hyper(self):
19-
conn = TSC.ConnectionItem()
20-
conn._connection_type = "hyper"
21-
conn.query_tagging = True
22-
self.assertEqual(conn.query_tagging, None)
23-
24-
def test_ignore_query_tag_for_teradata(self):
25-
conn = TSC.ConnectionItem()
26-
conn._connection_type = "teradata"
27-
conn.query_tagging = True
28-
self.assertEqual(conn.query_tagging, None)
29-
30-
def test_ignore_query_tag_for_snowflake(self):
31-
conn = TSC.ConnectionItem()
32-
conn._connection_type = "snowflake"
33-
conn.query_tagging = True
34-
self.assertEqual(conn.query_tagging, None)
5+
6+
def test_require_boolean_query_tag_fails():
7+
conn = TSC.ConnectionItem()
8+
conn._connection_type = "postgres"
9+
with pytest.raises(ValueError):
10+
conn.query_tagging = "no"
11+
12+
13+
def test_set_query_tag_normal_conn():
14+
conn = TSC.ConnectionItem()
15+
conn._connection_type = "postgres"
16+
conn.query_tagging = True
17+
assert conn.query_tagging
18+
19+
20+
def test_ignore_query_tag_for_hyper():
21+
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"
37+
conn.query_tagging = True
38+
assert conn.query_tagging is None

0 commit comments

Comments
 (0)