1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
# -*- coding: utf-8 -*-
"""
Basic SQLAlchemy tests.
"""
import unittest
import pytest
from .helpers import config, eq_
try:
import sqlalchemy as sa
except ImportError:
pytest.skip('SQLAlchemy is not available', allow_module_level=True)
from sqlalchemy.orm import sessionmaker, declarative_base
engine = sa.create_engine(
'mssql+pymssql://%s:%s@%s:%s/%s' % (
config.user,
config.password,
config.server,
config.port,
config.database
),
echo=False
)
meta = sa.MetaData()
Base = declarative_base(metadata=meta)
Session = sessionmaker(bind=engine)
#sess = Session()
class SAObj(Base):
__tablename__ = 'sa_test_objs'
id = sa.Column(sa.Integer, primary_key=True)
name = sa.Column(sa.String(50))
data = sa.Column(sa.PickleType)
#saotbl = SAObj.__table__
#
#saotbl.drop(engine, checkfirst=True)
#saotbl.create(engine)
@pytest.mark.mssql_server_required
class TestSA(unittest.TestCase):
def setUp(self):
self.sess = Session()
self.saotbl = SAObj.__table__
self.saotbl.drop(engine, checkfirst=True)
self.saotbl.create(engine)
def tearDown(self):
# issue rollback first, otherwise clearing the table might give us
# an error that the session is in a bad state
self.sess.rollback()
self.sess.execute(SAObj.__table__.delete())
self.sess.commit()
def test_basic_usage(self):
s = SAObj(name='foobar')
self.sess.add(s)
self.sess.commit()
assert s.id
assert self.sess.query(SAObj).count() == 1
def test_pickle_type(self):
s = SAObj(name='foobar', data=['one'])
self.sess.add(s)
self.sess.commit()
res = self.sess.execute(sa.select(self.saotbl.c.data))
row = res.fetchone()
eq_(row[0], ['one'])
|
