def test_call_must_return_filepath_changed(self):
my_handler = 'path.to.my.handler'
with mock.patch('onmydesk.models.app_settings.ONMYDESK_FILE_HANDLER', my_handler):
my_handler_mocked = mock.MagicMock(return_value='/tmp/filepath-changed.tsv')
with mock.patch('onmydesk.models.my_import', return_value=my_handler_mocked) as my_import_mocked:
self.assertEqual(
output_file_handler('/tmp/filepath.tsv'),
'/tmp/filepath-changed.tsv')
my_import_mocked.assert_called_once_with(my_handler)
my_handler_mocked.assert_called_once_with('/tmp/filepath.tsv')
def test_call_must_return_filepath_changed(self):
my_handler = 'path.to.my.handler'
with mock.patch('onmydesk.models.app_settings.ONMYDESK_FILE_HANDLER',
my_handler):
my_handler_mocked = mock.MagicMock(
return_value='/tmp/filepath-changed.tsv')
with mock.patch(
'onmydesk.models.my_import',
return_value=my_handler_mocked) as my_import_mocked:
self.assertEqual(output_file_handler('/tmp/filepath.tsv'),
'/tmp/filepath-changed.tsv')
my_import_mocked.assert_called_once_with(my_handler)
my_handler_mocked.assert_called_once_with('/tmp/filepath.tsv')
def test_call_must_return_same_filepath_if_a_file_handler_not_exists(self):
with mock.patch('onmydesk.models.ONMYDESK_FILE_HANDLER', None):
self.assertEqual(output_file_handler('/tmp/filepath.tsv'), '/tmp/filepath.tsv')
def test_call_must_return_same_filepath_if_a_file_handler_not_exists(self):
with mock.patch('onmydesk.models.ONMYDESK_FILE_HANDLER', None):
self.assertEqual(output_file_handler('/tmp/filepath.tsv'),
'/tmp/filepath.tsv')