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