class CassandraDriverTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.check_uuid = str(uuid4())
cls.trigger_uuid = str(uuid4())
@patch('alamo_alerter.backend.cassandra.Cluster')
@patch('alamo_alerter.backend.cassandra.PlainTextAuthProvider')
def setUp(self, auth_mock, cluster_mock):
session_mock = MagicMock()
cluster_mock.connect = MagicMock(return_value=session_mock)
self.trigger = dict(
id=self.trigger_uuid,
uuid=str(uuid4()),
severity='WARNING',
enabled=True,
result=dict(status=0, message='')
)
self.driver = CassandraDriver(
contact_points=[''],
username='******',
password='******',
keyspace='test'
)
def test_prepared_statements(self):
expected_calls = [call(SELECT_QUERY), call(INSERT_QUERY),
call(INSERT_SERVICE_QUERY)]
self.driver._session.prepare.assert_has_calls(
expected_calls, any_order=True
)
def test_get_result_method(self):
self.driver.get_result(self.check_uuid, self.trigger_uuid, 10)
self.driver._session.execute.assert_called_once_with(
self.driver._save_query_stmt,
(UUID(self.check_uuid), UUID(self.trigger_uuid), 10)
)
def test_save_result_method(self):
_now = datetime.now()
self.driver.save_result(
self.check_uuid, self.trigger, _now, _now, _now, False, '999'
)
self.assertTrue(self.driver._session.execute.called)
def setUp(self, auth_mock, cluster_mock):
session_mock = MagicMock()
cluster_mock.connect = MagicMock(return_value=session_mock)
self.trigger = dict(
id=self.trigger_uuid,
uuid=str(uuid4()),
severity='WARNING',
enabled=True,
result=dict(status=0, message='')
)
self.driver = CassandraDriver(
contact_points=[''],
username='******',
password='******',
keyspace='test'
)