async def test_update_with_invalid_params(self, key, value_increment, exception_name, exception_message):
storage_client_mock = MagicMock(spec=StorageClientAsync)
s = statistics.Statistics(storage_client_mock)
with pytest.raises(exception_name) as excinfo:
await s.update(key, value_increment)
assert exception_message == str(excinfo.value)
async def test_register_exception(self):
storage_client_mock = MagicMock(spec=StorageClientAsync)
s = statistics.Statistics(storage_client_mock)
with patch.object(statistics._logger, 'exception') as logger_exception:
with patch.object(s._storage, 'insert_into_tbl', side_effect=Exception):
with pytest.raises(Exception):
await s.register('T3Stat', 'Test stat')
args, kwargs = logger_exception.call_args
assert args[0] == 'Unable to create new statistic %s, error %s'
assert args[1] == 'T3Stat'
async def test_update_exception(self):
storage_client_mock = MagicMock(spec=StorageClientAsync)
s = statistics.Statistics(storage_client_mock)
msg = 'Unable to update statistics value based on statistics_key %s and value_increment %d,' \
' error %s', 'BUFFERED', 5, ''
with patch.object(s._storage, 'update_tbl', side_effect=Exception()):
with pytest.raises(Exception):
with patch.object(statistics._logger, 'exception') as logger_exception:
await s.update('BUFFERED', 5)
logger_exception.assert_called_once_with(*msg)
async def test_add_update_exception(self):
stat_dict = {'FOGBENCH/TEMPERATURE': 1}
storage_client_mock = MagicMock(spec=StorageClientAsync)
s = statistics.Statistics(storage_client_mock)
msg = 'Unable to update statistics value based on statistics_key %s and value_increment' \
' %s, error %s', "FOGBENCH/TEMPERATURE", 1, ''
with patch.object(s._storage, 'update_tbl', side_effect=Exception()):
with pytest.raises(Exception):
with patch.object(statistics._logger, 'exception') as logger_exception:
await s.add_update(stat_dict)
logger_exception.assert_called_once_with(*msg)
async def test_add_update(self):
stat_dict = {'FOGBENCH/TEMPERATURE': 1}
storage_client_mock = MagicMock(spec=StorageClientAsync)
s = statistics.Statistics(storage_client_mock)
payload = '{"where": {"column": "key", "condition": "=", "value": "FOGBENCH/TEMPERATURE"}, ' \
'"expressions": [{"column": "value", "operator": "+", "value": 1}]}'
async def mock_coro():
return {"response": "updated", "rows_affected": 1}
with patch.object(s._storage, 'update_tbl', return_value=mock_coro()) as stat_update:
await s.add_update(stat_dict)
stat_update.assert_called_once_with('statistics', payload)
async def test_load_keys(self):
"""Test the load key"""
storage_client_mock = MagicMock(spec=StorageClientAsync)
s = statistics.Statistics(storage_client_mock)
s._registered_keys = []
async def mock_coro():
return {'rows': [{"previous_value": 0, "value": 1,
"key": "K1", "description": "desc1"}]}
with patch.object(s._storage, 'query_tbl_with_payload', return_value=mock_coro()) as patch_query_tbl:
await s._load_keys()
assert "K1" in s._registered_keys
patch_query_tbl.assert_called_once_with('statistics', '{"return": ["key"]}')
async def test_update(self):
storage_client_mock = MagicMock(spec=StorageClientAsync)
s = statistics.Statistics(storage_client_mock)
async def mock_coro():
return {"response": "updated", "rows_affected": 1}
payload = '{"where": {"column": "key", "condition": "=", "value": "READING"}, ' \
'"expressions": [{"column": "value", "operator": "+", "value": 5}]}'
expected_result = {"response": "updated", "rows_affected": 1}
with patch.object(s._storage, 'update_tbl', return_value=mock_coro()) as stat_update:
await s.update('READING', 5)
assert expected_result['response'] == "updated"
stat_update.assert_called_once_with('statistics', payload)
async def test_load_keys_exception(self):
"""Test the load key exception"""
storage_client_mock = MagicMock(spec=StorageClientAsync)
s = statistics.Statistics(storage_client_mock)
s._registered_keys = []
async def mock_coro():
return Exception
with patch.object(statistics._logger, 'exception') as logger_exception:
with patch.object(s._storage, 'query_tbl_with_payload', return_value=mock_coro()):
await s._load_keys()
args, kwargs = logger_exception.call_args
assert args[0] == 'Failed to retrieve statistics keys, %s'
async def test_register_twice(self):
""" Test that register results in a database insert only once for same key"""
storageMock = MagicMock(spec=StorageClientAsync)
stats = statistics.Statistics(storageMock)
stats._registered_keys = []
async def mock_coro():
return {"response": "updated", "rows_affected": 1}
with patch.object(stats, '_load_keys', return_value=mock_coro()):
with patch.object(stats._storage, 'insert_into_tbl', return_value=mock_coro()) as stat_insert:
await stats.register('T2Stat', 'Test stat')
await stats.register('T2Stat', 'Test stat')
assert stat_insert.called
assert stat_insert.call_count == 1
stats._storage.insert_into_tbl.reset_mock()
async def test_insert_when_key_error(self):
stat_dict = {'FOGBENCH/TEMPERATURE': 1}
storage_client_mock = MagicMock(spec=StorageClientAsync)
s = statistics.Statistics(storage_client_mock)
async def mock_coro():
return {"response": "not updated", "rows_affected": 0}
with patch.object(s._storage, 'update_tbl', return_value=mock_coro()) as stat_update:
with patch.object(statistics._logger, 'exception') as logger_exception:
with pytest.raises(KeyError):
await s.add_update(stat_dict)
args, kwargs = logger_exception.call_args
assert args[0] == 'Statistics key %s has not been registered'
assert args[1] == 'FOGBENCH/TEMPERATURE'
assert stat_update.call_count == 1
async def test_register(self):
""" Test that register results in a database insert """
storageMock = MagicMock(spec=StorageClientAsync)
stats = statistics.Statistics(storageMock)
stats._registered_keys = []
async def mock_coro():
return {"response": "updated", "rows_affected": 1}
with patch.object(stats, '_load_keys', return_value=mock_coro()):
with patch.object(stats._storage, 'insert_into_tbl', return_value=mock_coro()) as stat_update:
await stats.register('T1Stat', 'Test stat')
args, kwargs = stat_update.call_args
assert args[0] == 'statistics'
expected_storage_args = json.loads(args[1])
assert expected_storage_args['key'] == 'T1Stat'
assert expected_storage_args['value'] == 0
assert expected_storage_args['previous_value'] == 0
assert expected_storage_args['description'] == 'Test stat'
stats._storage.insert_into_tbl.reset_mock()
