def test_replace_writes_expected_content(self, mock_open):
store = AlertCursorStore(PROFILE_NAME)
store.replace("checkpointname", 123)
user_path = path.join(path.expanduser("~"), ".code42cli")
path.join(user_path, ALERT_CHECKPOINT_FOLDER_NAME, PROFILE_NAME,
"checkpointname")
mock_open.return_value.write.assert_called_once_with("123")
def test_get_reads_expected_file(self, mock_open):
store = AlertCursorStore(PROFILE_NAME)
store.get(CURSOR_NAME)
user_path = path.join(path.expanduser("~"), ".code42cli")
expected_path = path.join(user_path, ALERT_CHECKPOINT_FOLDER_NAME,
PROFILE_NAME, CURSOR_NAME)
mock_open.assert_called_once_with(expected_path)
def test_replace_writes_to_expected_file(self, mock_open):
store = AlertCursorStore(PROFILE_NAME)
store.replace("checkpointname", 123)
user_path = path.join(path.expanduser("~"), ".code42cli")
expected_path = path.join(user_path, ALERT_CHECKPOINT_FOLDER_NAME,
PROFILE_NAME, "checkpointname")
mock_open.assert_called_once_with(expected_path, "w")
def test_clean_calls_remove_on_each_checkpoint(self, mock_open,
mock_remove, mock_listdir,
mock_isfile):
mock_listdir.return_value = ["fileone", "filetwo", "filethree"]
store = AlertCursorStore(PROFILE_NAME)
store.clean()
assert mock_remove.call_count == 3
def test_delete_when_checkpoint_does_not_exist_raises_cli_error(
self, mock_open, mock_remove
):
store = AlertCursorStore(PROFILE_NAME)
mock_remove.side_effect = FileNotFoundError
with pytest.raises(Code42CLIError):
store.delete("deleteme")
def test_delete_calls_remove_on_expected_file(self, mock_open,
mock_remove):
store = AlertCursorStore(PROFILE_NAME)
store.delete("deleteme")
user_path = path.join(path.expanduser("~"), ".code42cli")
expected_path = path.join(user_path, ALERT_CHECKPOINT_FOLDER_NAME,
PROFILE_NAME, "deleteme")
mock_remove.assert_called_once_with(expected_path)
def test_get_all_cursors_returns_all_checkpoints(self, mock_open,
mock_listdir,
mock_isfile):
mock_listdir.return_value = ["fileone", "filetwo", "filethree"]
store = AlertCursorStore(PROFILE_NAME)
cursors = store.get_all_cursors()
assert len(cursors) == 3
assert cursors[0].name == "fileone"
assert cursors[1].name == "filetwo"
assert cursors[2].name == "filethree"
def test_get_when_profile_does_not_exist_returns_none(self, mocker):
store = AlertCursorStore(PROFILE_NAME)
checkpoint = store.get(CURSOR_NAME)
mock_open = mocker.patch(f"{_NAMESPACE}.open")
mock_open.side_effect = FileNotFoundError
assert checkpoint is None
def test_get_returns_expected_timestamp(self, mock_open):
store = AlertCursorStore(PROFILE_NAME)
checkpoint = store.get(CURSOR_NAME)
assert checkpoint == 123456789
def _get_alert_cursor_store(profile_name):
return AlertCursorStore(profile_name)