def test_gets_path_to_ptr_file_by_name_and_opens_it(self):
path_from_appdata_or_input('something')
self._mock_get_or_create_udd.assert_called_once()
self._mock_os.path.join.assert_called_once_with(
'user_data_dir', 'something')
self._mock_open.assert_called_once_with(
self._mock_os.path.join.return_value, 'r')
def test_prompts_path_to_csv_file_if_path_in_pointer_file_is_wrong(
self, mock_print):
# First call checks for ptr file, second call checks for csv file
self._mock_os.path.isfile.side_effect = [True, False]
path_from_appdata_or_input('something')
mock_print.assert_called_once_with(
"Stored path 'path_from_ptr_file' does not exist")
self._mock_read_path.assert_called_with(
"Enter valid path to 'something'")
def test_prompts_path_to_csv_file_if_pointer_file_nonexistent_returns_it(
self):
self._mock_os.path.isfile.return_value = False
result = path_from_appdata_or_input('something')
self._mock_read_path.assert_called_once_with(
"Enter valid path to 'something'")
assert result == self._mock_read_path.return_value
def test_saves_correct_path_to_pointer_file(self):
self._mock_os.path.isfile.return_value = False
path_from_appdata_or_input('something')
self._mock_open.return_value.write.assert_called_once_with(
'new_path_to_csv')
def test_creates_pointer_file_if_did_not_exist(self):
path_to_ptr_file = self._mock_os.path.join.return_value
self._mock_os.path.isfile.return_value = False
path_from_appdata_or_input('something')
self._mock_open.assert_called_once_with(path_to_ptr_file, 'w')
def test_if_user_doesnt_want_to_use_path_from_ptr_prompts_another(self):
self._mock_read_yes_no.return_value = False
path_from_appdata_or_input('something')
self._mock_read_path.assert_called_once_with(
"Enter valid path to 'something'")
def test_asks_if_you_want_to_use_path_from_ptr(self):
path_from_appdata_or_input('something')
self._mock_read_yes_no.assert_called_once_with(
"Do you want to use file 'path_from_ptr_file'?")
def test_read_path_to_csv_file_from_ptr_file_and_return_it(self):
self._mock_read_yes_no.return_value = True
result = path_from_appdata_or_input('something')
assert result == 'path_from_ptr_file'