|
1 | | -import unittest |
2 | 1 | import tableauserverclient as TSC |
3 | 2 |
|
| 3 | +import pytest |
4 | 4 |
|
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