def test_generate_engine_success(self, mock_engine):
mockEngine = MagicMock()
mock_engine.return_value = mockEngine
testManager = SessionManager()
testManager.generateEngine()
testManager.engine.execute.assert_called_once()
self.assertEqual(testManager.engine, mockEngine)
def test_initialize_db(self, mock_base):
testManager = SessionManager()
mock_engine = MagicMock()
mock_engine.dialect.has_table.return_value = None
testManager.engine = mock_engine
testManager.initializeDatabase()
mock_base.metadata.create_all.assert_called_once_with(mock_engine)
def test_create_session(self, mock_generate, mock_maker):
testManager = SessionManager()
mock_session = MagicMock()
mock_session.return_value = 'newSession'
mock_maker.return_value = mock_session
newSession = testManager.createSession()
self.assertEqual(newSession, 'newSession')
self.assertEqual(newSession, testManager.session)
def test_init_local_vars(self, mock_log):
testManager = SessionManager(user='******',
pswd='pswd',
host='testing',
port='1',
db='test')
self.assertEqual(testManager.port, '1')
self.assertEqual(testManager.engine, None)
def test_init_local_vars(self, mock_log, mock_decrypt):
def returnVal(value):
return value
mock_decrypt.side_effect = returnVal
testManager = SessionManager(
user='******',
pswd='pswd',
host='testing',
port='1',
db='test'
)
self.assertEqual(testManager.port, '1')
self.assertEqual(testManager.engine, None)
def test_close_connection(self, mock_commit):
mock_session = MagicMock()
mock_engine = MagicMock()
testManager = SessionManager()
testManager.session = mock_session
testManager.engine = mock_engine
testManager.closeConnection()
mock_commit.assert_called_once()
mock_session.close.assert_called_once()
mock_engine.dispose.assert_called_once()
def test_rollback_session(self):
mock_session = MagicMock()
testManager = SessionManager()
testManager.session = mock_session
testManager.rollbackChanges()
mock_session.rollback.assert_called_once()
def test_commit_session(self):
mock_session = MagicMock()
testManager = SessionManager()
testManager.session = mock_session
testManager.commitChanges()
mock_session.commit.assert_called_once()
def test_start_session(self):
mock_session = MagicMock()
testManager = SessionManager()
testManager.session = mock_session
testManager.startSession()
mock_session.begin_nested.assert_called_once()
def test_generate_engine_error(self, mock_engine):
testManager = SessionManager()
with self.assertRaises(Exception):
testManager.generateEngine()
def test_init_env_vars(self, mock_log):
testManager = SessionManager()
self.assertEqual(testManager.host, None)
self.assertEqual(testManager.port, None)
def test_env_decryptor_boto_error(self, mock_boto):
mock_boto.client().decrypt.side_effect = ClientError
outEnv = SessionManager.decryptEnvVar('testing')
self.assertEqual(outEnv, b64encode('testing'.encode('utf-8')).decode('utf-8'))
def test_env_decryptor_non_encoded(self, mock_boto):
mock_boto.client().decrypt.return_value = {'Plaintext': 'testing'}
outEnv = SessionManager.decryptEnvVar('testing')
self.assertEqual(outEnv, 'testing')
def test_generate_engine_success(self, mock_engine):
testManager = SessionManager()
testManager.generateEngine()
self.assertEqual(testManager.engine, 'newEngine')
