def test_003_sql_conversion(self):
from tests.files.schema_1 import testData
db = PgDatabase()
for t in testData:
_tbl = table_from_string(t[0])
self.assertNotEqual(_tbl, None)
self.assertEqual(len(_tbl.fields), t[1])
self.assertEqual(db.create_string(_tbl), t[2])
def test_002_connect(self):
kwargs = {'dbname': 'slurpy', 'user': '******', 'password': '******' }
db = PgDatabase()
self.assertEqual(db.connect(**kwargs), True)
self.assertEqual(db.close(), True)
kwargs = {'dbname': 'slurpy', 'user': '******', 'password': '******' }
self.assertEqual(db.connect(**kwargs), False)
self.assertEqual(db.close(), True)
def test_005_catalog(self):
c = Catalog(os.path.join(testDir, 'files', 'test.lrcat'))
self.assertEqual(c.version, '0300025')
self.assertEqual(c.get_schema_from_catalog(), True)
kwargs = {'dbname': 'slurpy', 'user': '******', 'password': '******' }
db = PgDatabase()
self.assertEqual(db.connect(**kwargs), True)
self.assertEqual(c.create_database(db, True), True)
def test_004_creation(self):
from tests.files.schema_1 import testData
kwargs = {'dbname': 'slurpy', 'user': '******', 'password': '******' }
db = PgDatabase()
self.assertEqual(db.connect(**kwargs), True)
self.assertEqual(db.dropall(), True)
for t in testData:
_tbl = table_from_string(t[0])
# Drop the table before we start
self.assertEqual(db.drop(_tbl), True)
self.assertEqual(db.create_string(_tbl), t[2])
# Simple create, no drop before creation
self.assertEqual(db.create(_tbl), t[3])
if not t[3]:
continue
# Create, dropping table before creation, should succeed
self.assertEqual(db.create(_tbl, True), True)
# Simple create, no drop before creation, should fail
self.assertEqual(db.create(_tbl), False)
# Drop the table
self.assertEqual(db.drop(_tbl), True)
self.assertEqual(db.close(), True)
def test_001_conversions(self):
db = PgDatabase()
stmt = 'select ?,? from blah'
args = ['abc', 'def']
self.assertEqual(db._convert_query_stmt(stmt), 'select %s,%s from blah')
def __init__(self):
self._db = PgDatabase()
self.config = {}
self.session = None
