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])
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)