author | Radek Brich <brich.radek@ifortuna.cz> |
Tue, 06 May 2014 18:37:41 +0200 | |
changeset 100 | d6088dba8fea |
parent 50 | f71d3abbb18f |
child 104 | d8ff52a0390f |
permissions | -rw-r--r-- |
49
08e4dfe1b0cb
Add test for MyManager (enable only when MySQLdb is available). Configure tests using pgtoolkit.conf (same as used by other executables).
Radek Brich <radek.brich@devl.cz>
parents:
48
diff
changeset
|
1 |
#!/usr/bin/env python3 |
22 | 2 |
|
48
b82c7c2fb5af
PgManager: Fix logging, log queries before executing, possible exceptions are logged after. Add tests for RowDict. Add tests.py - runs all tests.
Radek Brich <radek.brich@devl.cz>
parents:
37
diff
changeset
|
3 |
from tests.config import Config |
b82c7c2fb5af
PgManager: Fix logging, log queries before executing, possible exceptions are logged after. Add tests for RowDict. Add tests.py - runs all tests.
Radek Brich <radek.brich@devl.cz>
parents:
37
diff
changeset
|
4 |
from pgtoolkit import pgmanager |
b82c7c2fb5af
PgManager: Fix logging, log queries before executing, possible exceptions are logged after. Add tests for RowDict. Add tests.py - runs all tests.
Radek Brich <radek.brich@devl.cz>
parents:
37
diff
changeset
|
5 |
|
22 | 6 |
import unittest |
7 |
||
8 |
||
9 |
class TestSuite(unittest.TestCase): |
|
10 |
||
11 |
def setUp(self): |
|
49
08e4dfe1b0cb
Add test for MyManager (enable only when MySQLdb is available). Configure tests using pgtoolkit.conf (same as used by other executables).
Radek Brich <radek.brich@devl.cz>
parents:
48
diff
changeset
|
12 |
cfg = Config('pgtoolkit.conf') |
08e4dfe1b0cb
Add test for MyManager (enable only when MySQLdb is available). Configure tests using pgtoolkit.conf (same as used by other executables).
Radek Brich <radek.brich@devl.cz>
parents:
48
diff
changeset
|
13 |
test_db_conn_params = cfg['databases']['test'] |
22 | 14 |
self.m = pgmanager.get_instance() |
49
08e4dfe1b0cb
Add test for MyManager (enable only when MySQLdb is available). Configure tests using pgtoolkit.conf (same as used by other executables).
Radek Brich <radek.brich@devl.cz>
parents:
48
diff
changeset
|
15 |
self.m.create_conn(dsn=test_db_conn_params) |
08e4dfe1b0cb
Add test for MyManager (enable only when MySQLdb is available). Configure tests using pgtoolkit.conf (same as used by other executables).
Radek Brich <radek.brich@devl.cz>
parents:
48
diff
changeset
|
16 |
self.m.create_conn('autocommit', isolation_level='autocommit', dsn=test_db_conn_params) |
22 | 17 |
|
18 |
def tearDown(self): |
|
19 |
self.m.destroy_conn() |
|
20 |
self.m.destroy_conn('autocommit') |
|
21 |
||
22 |
def test_connection_get_put(self): |
|
23 |
conn = self.m.get_conn() |
|
24 |
||
25 |
curs = conn.cursor() |
|
26 |
curs.execute('SELECT %s AS t', [True]) |
|
27 |
d = curs.fetchone_dict() |
|
28 |
self.assertTrue(d.t) |
|
29 |
||
30 |
self.m.put_conn(conn) |
|
31 |
||
32 |
def test_connection_with(self): |
|
33 |
with self.m.cursor() as curs: |
|
34 |
curs.execute('SELECT %s AS t', [True]) |
|
35 |
d = curs.fetchone_dict() |
|
36 |
self.assertTrue(d.t) |
|
37 |
||
38 |
def test_connection_autocommit(self): |
|
39 |
conn = self.m.get_conn('autocommit') |
|
40 |
curs = conn.cursor() |
|
41 |
name = 'test' |
|
42 |
curs.execute('DELETE FROM test WHERE name = %s', [name]) |
|
43 |
curs.execute('INSERT INTO test (name) VALUES (%s)', [name]) |
|
44 |
conn.close() |
|
45 |
||
46 |
conn = self.m.get_conn('autocommit') |
|
47 |
curs = conn.cursor() |
|
48 |
curs.execute('SELECT * FROM test WHERE name = %s', [name]) |
|
49 |
self.assertTrue(curs.rowcount == 1) |
|
50 |
self.m.put_conn(conn) |
|
51 |
||
49
08e4dfe1b0cb
Add test for MyManager (enable only when MySQLdb is available). Configure tests using pgtoolkit.conf (same as used by other executables).
Radek Brich <radek.brich@devl.cz>
parents:
48
diff
changeset
|
52 |
|
22 | 53 |
if __name__ == '__main__': |
54 |
unittest.main() |
|
55 |