def test_it_uses_the_ordered_policy_if_smart_ordering_is_true(self, Policy):
self.defaults["smart_ordering"] = True
actions_list = MagicMock()
with patch.object(TestableCommand, "collect_all_actions", return_value=actions_list):
sut = TestableCommand(**self.defaults)
sut.execute()
Policy.assert_called_with(actions_list)
Policy.return_value.execute.assert_called_once_with()
def test_it_uses_the_basic_policy_if_smart_ordering_is_false(self, Policy):
self.defaults['smart_ordering'] = False
actions_list = MagicMock()
with patch.object(TestableCommand,
'collect_all_actions',
return_value=actions_list):
sut = TestableCommand(**self.defaults)
sut.execute()
Policy.assert_called_with(actions_list)
Policy.return_value.execute.assert_called_once_with()
def test_it_wraps_the_basic_policy_in_a_transaction_policy_if_configured(
self, BasicSyncPolicy, TransactionSyncPolicy
):
self.defaults["smart_ordering"] = False
self.defaults["as_transaction"] = True
actions_list = MagicMock()
with patch.object(TestableCommand, "collect_all_actions", return_value=actions_list):
sut = TestableCommand(**self.defaults)
sut.execute()
TransactionSyncPolicy.assert_called_with(BasicSyncPolicy.return_value)
TransactionSyncPolicy.return_value.execute.assert_called_once_with()
def test_command_raises_error_if_not_CSV_file(self, SupportedFileChecker):
SupportedFileChecker.is_valid.return_value = False
files_mock = MagicMock()
sut = TestableCommand(
**{
'files': [files_mock],
'file_name_regex': DEFAULT_FILE_REGEX,
'create_external_system': MagicMock(),
'smart_ordering': MagicMock(),
'as_transaction': MagicMock()
})
with self.assertRaises(CommandError):
sut.execute()
SupportedFileChecker.is_valid.assert_called_with(files_mock)
def test_it_wraps_the_basic_policy_in_a_transaction_policy_if_configured(
self, BasicSyncPolicy, TransactionSyncPolicy):
self.defaults['smart_ordering'] = False
self.defaults['as_transaction'] = True
actions_list = MagicMock()
with patch.object(TestableCommand,
'collect_all_actions',
return_value=actions_list):
sut = TestableCommand(**self.defaults)
sut.execute()
TransactionSyncPolicy.assert_called_with(
BasicSyncPolicy.return_value)
TransactionSyncPolicy.return_value \
.execute.assert_called_once_with()
def test_command_raises_error_if_not_CSV_file(self, SupportedFileChecker):
SupportedFileChecker.is_valid.return_value = False
files_mock = MagicMock()
sut = TestableCommand(
**{
"files": [files_mock],
"file_name_regex": DEFAULT_FILE_REGEX,
"create_external_system": MagicMock(),
"smart_ordering": MagicMock(),
"as_transaction": MagicMock(),
}
)
with self.assertRaises(CommandError):
sut.execute()
SupportedFileChecker.is_valid.assert_called_with(files_mock)