def setUp(self):
config = ConnectionConfiguration.objects.create(name='my_name', connection_type='django.db.backends.sqlite3',
info='"name": "test-data/test_site.sqlite3"')
# get connection
manager = ConnectionManager(config.info_to_json())
self.connection = manager.get(config.name)
super(UserTableSelectionFormTests, self).setUp()
def setUp(self):
config = Configuration('test-data/config/sqlite3_config.json')
self.sqlite3 = ConnectionManager(config.get_connection_info()).get('my_site_db')
config = Configuration('test-data/config/mysql_config.json')
self.mysql = ConnectionManager(config.get_connection_info()).get('my_other_db')
config = Configuration('test-data/config/pg_config.json')
self.pg = ConnectionManager(config.get_connection_info()).get('my_pg_db')
super(ConnectionTests, self).setUp()
def __init__(self, config_file='', from_str='', token=None):
if config_file:
from_str = open(config_file).read()
self.config = Configuration(from_str=from_str)
self.cm = ConnectionManager(self.config.get_connection_info())
self.pm = PropertyManager(self.cm, self.config, token=token)
def setUp(self):
config = Configuration('test-data/config/sqlite3_config.json')
self.sqlite3 = ConnectionManager(
config.get_connection_info()).get('my_site_db')
config = Configuration('test-data/config/mysql_config.json')
self.mysql = ConnectionManager(
config.get_connection_info()).get('my_other_db')
config = Configuration('test-data/config/pg_config.json')
self.pg = ConnectionManager(
config.get_connection_info()).get('my_pg_db')
super(ConnectionTests, self).setUp()
def get_connection(self):
manager = ConnectionManager(self.info_to_json())
return manager.get(self.name)
def test_exception_connection_not_found(self):
config = Configuration('test-data/config/sqlite3_config.json')
cm = ConnectionManager(config.get_connection_info())
with self.assertRaises(ConnectionNotFound):
cm.get('wrong_db')
class ConnectionTests(TestCase):
def setUp(self):
config = Configuration('test-data/config/sqlite3_config.json')
self.sqlite3 = ConnectionManager(
config.get_connection_info()).get('my_site_db')
config = Configuration('test-data/config/mysql_config.json')
self.mysql = ConnectionManager(
config.get_connection_info()).get('my_other_db')
config = Configuration('test-data/config/pg_config.json')
self.pg = ConnectionManager(
config.get_connection_info()).get('my_pg_db')
super(ConnectionTests, self).setUp()
def test_tables(self):
# sqlite also reports the system special table
self.assertEqual(len(self.sqlite3.tables()), 3)
self.assertEqual(len(self.mysql.tables()), 3)
self.assertEqual(len(self.pg.tables()), 3)
def test_table_primary_key(self):
self.assertEqual(self.sqlite3.primary_key_of('users'),
'users.id@my_site_db')
self.assertEqual(self.mysql.primary_key_of('users'),
'users.id@my_other_db')
self.assertEqual(self.pg.primary_key_of('users'), 'users.id@my_pg_db')
def test_table_properties(self):
# sqlite falsely reports the id, too
self.assertEqual(len(self.sqlite3.get_data_properties('Users')[0]), 6)
self.assertEqual(len(self.mysql.get_data_properties('users')[0]), 5)
self.assertEqual(len(self.pg.get_data_properties('users')[0]), 5)
# make sure first element is column name
self.assertTrue(
type(self.sqlite3.get_data_properties('Users')[0][0][0]) in
[unicode, str])
self.assertTrue(
type(self.mysql.get_data_properties('users')[0][0][0]) in
[unicode, str])
self.assertTrue(
type(self.pg.get_data_properties('users')[0][0][0]) in
[unicode, str])
# also check the properties from other tables pointing to Users as well
self.assertEqual(
len(
self.sqlite3.get_data_properties('Users',
from_related=True)[0]), 9)
self.assertEqual(
len(self.mysql.get_data_properties('users', from_related=True)[0]),
6)
self.assertEqual(
len(self.pg.get_data_properties('users', from_related=True)[0]), 6)
# also check other tables pointing to Users as well
self.assertEqual(
len(
self.sqlite3.get_data_properties('Running',
from_related=True)[0]), 9)
self.assertEqual(
len(
self.mysql.get_data_properties('running',
from_related=True)[0]), 6)
self.assertEqual(
len(self.pg.get_data_properties('running', from_related=True)[0]),
6)
def test_related_tables(self):
self.assertIn('Running', self.sqlite3.get_related_tables('Users')[0])
self.assertEqual(
[[u'Running', u'Users.id@my_site_db', u'Running.user@my_site_db']],
self.sqlite3.get_related_tables('Users')[1])
self.assertEqual([[
u'running', u'users.id@my_other_db', u'running.user@my_other_db'
]],
self.mysql.get_related_tables('users')[1])
self.assertEqual(
[['running', u'users.id@my_pg_db', u'running.user_id@my_pg_db']],
self.pg.get_related_tables('users')[1])
def test_exception_connection_not_found(self):
config = Configuration('test-data/config/sqlite3_config.json')
cm = ConnectionManager(config.get_connection_info())
with self.assertRaises(ConnectionNotFound):
cm.get('wrong_db')
class ConnectionTests(TestCase):
def setUp(self):
config = Configuration('test-data/config/sqlite3_config.json')
self.sqlite3 = ConnectionManager(config.get_connection_info()).get('my_site_db')
config = Configuration('test-data/config/mysql_config.json')
self.mysql = ConnectionManager(config.get_connection_info()).get('my_other_db')
config = Configuration('test-data/config/pg_config.json')
self.pg = ConnectionManager(config.get_connection_info()).get('my_pg_db')
super(ConnectionTests, self).setUp()
def test_tables(self):
# sqlite also reports the system special table
self.assertEqual(len(self.sqlite3.tables()), 3)
self.assertEqual(len(self.mysql.tables()), 3)
self.assertEqual(len(self.pg.tables()), 3)
def test_table_primary_key(self):
self.assertEqual(self.sqlite3.primary_key_of('users'), 'users.id@my_site_db')
self.assertEqual(self.mysql.primary_key_of('users'), 'users.id@my_other_db')
self.assertEqual(self.pg.primary_key_of('users'), 'users.id@my_pg_db')
def test_table_properties(self):
# sqlite falsely reports the id, too
self.assertEqual(len(self.sqlite3.get_data_properties('Users')[0]), 6)
self.assertEqual(len(self.mysql.get_data_properties('users')[0]), 5)
self.assertEqual(len(self.pg.get_data_properties('users')[0]), 5)
# make sure first element is column name
self.assertTrue(type(self.sqlite3.get_data_properties('Users')[0][0][0]) in [unicode, str])
self.assertTrue(type(self.mysql.get_data_properties('users')[0][0][0]) in [unicode, str])
self.assertTrue(type(self.pg.get_data_properties('users')[0][0][0]) in [unicode, str])
# also check the properties from other tables pointing to Users as well
self.assertEqual(len(self.sqlite3.get_data_properties('Users', from_related=True)[0]), 9)
self.assertEqual(len(self.mysql.get_data_properties('users', from_related=True)[0]), 6)
self.assertEqual(len(self.pg.get_data_properties('users', from_related=True)[0]), 6)
# also check other tables pointing to Users as well
self.assertEqual(len(self.sqlite3.get_data_properties('Running', from_related=True)[0]), 9)
self.assertEqual(len(self.mysql.get_data_properties('running', from_related=True)[0]), 6)
self.assertEqual(len(self.pg.get_data_properties('running', from_related=True)[0]), 6)
def test_related_tables(self):
self.assertIn('Running', self.sqlite3.get_related_tables('Users')[0])
self.assertEqual([[u'Running', u'Users.id@my_site_db', u'Running.user@my_site_db']],
self.sqlite3.get_related_tables('Users')[1])
self.assertEqual([[u'running', u'users.id@my_other_db', u'running.user@my_other_db']],
self.mysql.get_related_tables('users')[1])
self.assertEqual([['running', u'users.id@my_pg_db', u'running.user_id@my_pg_db']],
self.pg.get_related_tables('users')[1])
def setUp(self):
config = Configuration('test-data/config/ctc_config.json')
ctc_manager = ConnectionManager(config.get_connection_info())
self.pm = PropertyManager(ctc_manager, config)
def get_connection(self):
manager = ConnectionManager(self.info_to_json())
return manager.get(self.name)
