def test_upsert_object_succesfully_edited(self, extract_model_mock, edit_mock, add_mock, get_operation_mock, is_upsert_supported_mock): op_name = mock.MagicMock() params = mock.MagicMock() is_upsert_supported_mock.return_value = True error = FtdConfigurationError("Obj duplication error") error.obj = mock.MagicMock() add_mock.side_effect = error result = self._resource.upsert_object(op_name, params) assert result == edit_mock.return_value is_upsert_supported_mock.assert_called_once_with(op_name) extract_model_mock.assert_called_once_with(op_name) get_operation_mock.assert_called_once_with(extract_model_mock.return_value) add_mock.assert_called_once_with(get_operation_mock.return_value, params) edit_mock.assert_called_once_with(get_operation_mock.return_value, error.obj, params)
def test_upsert_object_neither_added_nor_edited(self, extract_model_mock, edit_mock, add_mock, get_operation_mock, is_upsert_supported_mock): op_name = mock.MagicMock() params = mock.MagicMock() is_upsert_supported_mock.return_value = True error = FtdConfigurationError("Obj duplication error") error.obj = mock.MagicMock() add_mock.side_effect = error edit_mock.side_effect = FtdConfigurationError("Some object edit error") self.assertRaises( FtdConfigurationError, self._resource.upsert_object, op_name, params ) is_upsert_supported_mock.assert_called_once_with(op_name) extract_model_mock.assert_called_once_with(op_name) get_operation_mock.assert_called_once_with(extract_model_mock.return_value) add_mock.assert_called_once_with(get_operation_mock.return_value, params) edit_mock.assert_called_once_with(get_operation_mock.return_value, error.obj, params)