def test_get_connection_failure(self):
"""
Tests `get_connection`.
Failure.
"""
try:
mysql_utility.get_connection(MYSQL_HOST, MYSQL_USER, 'lorem',
MYSQL_DATABASE)
except Exception as e:
self.assertIsInstance(e, MySQLdb.OperationalError)
def test_get_connection_success(self):
"""
Tests `get_connection`.
Success.
"""
actual = None
try:
actual = mysql_utility.get_connection(MYSQL_HOST, MYSQL_USER,
MYSQL_PASSWORD,
MYSQL_DATABASE)
self.assertIsInstance(actual, MySQLdb.connection)
except Exception as e:
raise e
finally:
mysql_utility.close_connection_and_cursor(actual, None)
def test_get_show_tables(self):
"""
Tests `get_show_tables`.
"""
connection = mysql_utility.get_connection(MYSQL_HOST, MYSQL_USER,
MYSQL_PASSWORD,
MYSQL_DATABASE)
cursor = mysql_utility.get_cursor(connection)
actual = mysql_utility.get_show_tables(cursor, MYSQL_DATABASE)
self.assertIsInstance(actual, tuple)
self.assertEqual({
'TABLE_NAME': 'users',
'COLUMN_NAME': 'id'
}, actual[0])
self.assertEqual({
'TABLE_NAME': 'users',
'COLUMN_NAME': 'mapped_id'
}, actual[1])
def test_get_show_columns(self):
"""
Tests `get_show_columns`.
"""
connection = actual = None
try:
connection = mysql_utility.get_connection(MYSQL_HOST, MYSQL_USER,
MYSQL_PASSWORD,
MYSQL_DATABASE)
cursor = mysql_utility.get_cursor(connection)
table = 'users'
actual = mysql_utility.get_show_columns(cursor, table)
self.assertIsInstance(actual, tuple)
self.assertEqual('id', actual[0]['Field'])
self.assertEqual('mapped_id', actual[1]['Field'])
except Exception as e:
raise e
finally:
mysql_utility.close_connection_and_cursor(connection, cursor)
def test_get_max_id(self):
"""
Tests `get_max_id`.
"""
connection = actual = None
try:
connection = mysql_utility.get_connection(MYSQL_HOST, MYSQL_USER,
MYSQL_PASSWORD,
MYSQL_DATABASE)
cursor = mysql_utility.get_cursor(connection)
expected = 2
actual = mysql_utility.get_max_id(cursor, MYSQL_DATABASE, 'users',
'id')
self.assertEqual(expected, actual)
expected = 10
actual = mysql_utility.get_max_id(cursor, MYSQL_DATABASE, 'users',
'mapped_id')
self.assertEqual(expected, actual)
except Exception as e:
raise e
finally:
mysql_utility.close_connection_and_cursor(connection, cursor)