Skip to content

Commit 82f63c7

Browse files
committed
update dialect tests
1 parent 41679fa commit 82f63c7

File tree

1 file changed

+58
-3
lines changed

1 file changed

+58
-3
lines changed

test/test_dialect.py

+58-3
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,64 @@
11
from sqlalchemy_teradata.dialect import TeradataDialect
22
from sqlalchemy.testing import fixtures
3+
from sqlalchemy import testing, select
4+
from sqlalchemy import Table, Column, Integer, String, MetaData, ForeignKey
5+
from sqlalchemy_teradata.base import CreateView,DropView
36

47
class TeradataDialectTest(fixtures.TestBase):
8+
59
def setup(self):
6-
self.dialect = TeradataDialect()
10+
self.conn = testing.db.connect()
11+
self.engine = self.conn.engine
12+
self.dialect = self.conn.dialect
13+
self.metadata = MetaData()
14+
15+
self.user_name = self.engine.execute('sel user').scalar()
16+
self.tbl_name = self.user_name + '_test_table'
17+
self.view_name = self.user_name + '_test_view'
18+
19+
# Setup test table (user should have necessary rights to create table)
20+
self.test_table = Table(self.tbl_name, self.metadata,
21+
Column('id', Integer, primary_key=True))
22+
# Setup a test view
23+
self.test_view = CreateView(self.view_name, select([self.test_table.c.id.label('view_id')]))
24+
25+
# Create tables
26+
self.metadata.create_all(self.engine)
27+
28+
#Create views
29+
self.conn.execute(self.test_view)
30+
31+
def tearDown(self):
32+
# drop view(s)
33+
self.conn.execute(DropView(self.view_name))
34+
35+
# drop table(s)
36+
self.metadata.drop_all(self.engine)
37+
self.conn.close()
38+
39+
def test_has_table(self):
40+
assert self.dialect.has_table(self.conn, self.test_table.name)
41+
42+
def test_get_table_names(self):
43+
tbls = self.dialect.get_table_names(self.conn)
44+
assert self.dialect.normalize_name(self.test_table.name) in tbls
45+
46+
def test_get_schema_names(self):
47+
schemas = self.dialect.get_schema_names(self.conn)
48+
assert self.dialect.normalize_name(self.user_name) in schemas
49+
50+
def test_get_view_names(self):
51+
views = self.dialect.get_view_names(self.conn)
52+
assert self.dialect.normalize_name(self.test_view.name) in views
53+
54+
def test_get_pk_constraint(self):
55+
assert False
56+
57+
def test_get_unique_constraint(self):
58+
assert False
59+
60+
def test_get_foreign_keys(self):
61+
assert False
762

8-
def test_Attrs(self):
9-
assert hasattr(self.dialect, 'name')
63+
def test_get_indexes(self):
64+
assert False

0 commit comments

Comments
 (0)