def test_given_no_main_user_then_no_associations_fixed(self,
line_extension_associate,
find_main_user_line,
fix_associations):
line_extension = Mock(LineExtension)
created_line_extension = line_extension_associate.return_value = Mock(LineExtension, line_id=1, extension_id=2)
find_main_user_line.return_value = None
result = ule_service.associate_line_extension(line_extension)
assert_that(result, equal_to(created_line_extension))
find_main_user_line.assert_called_once_with(created_line_extension.line_id)
self.assertNotCalled(fix_associations)
def test_given_main_user_associated_then_associations_get_fixed(self,
line_extension_associate,
find_main_user_line,
fix_associations):
line_extension = Mock(LineExtension)
created_line_extension = line_extension_associate.return_value = Mock(LineExtension, line_id=1, extension_id=2)
main_user_line = find_main_user_line.return_value = Mock(UserLine, user_id=3, line_id=1)
result = ule_service.associate_line_extension(line_extension)
assert_that(result, equal_to(created_line_extension))
find_main_user_line.assert_called_once_with(created_line_extension.line_id)
fix_associations.assert_called_once_with(main_user_line)
def test_given_no_main_user_then_no_associations_fixed(
self, line_extension_associate, find_main_user_line,
fix_associations):
line_extension = Mock(LineExtension)
created_line_extension = line_extension_associate.return_value = Mock(
LineExtension, line_id=1, extension_id=2)
find_main_user_line.return_value = None
result = ule_service.associate_line_extension(line_extension)
assert_that(result, equal_to(created_line_extension))
find_main_user_line.assert_called_once_with(
created_line_extension.line_id)
self.assertNotCalled(fix_associations)
def test_given_main_user_associated_then_associations_get_fixed(
self, line_extension_associate, find_main_user_line,
fix_associations):
line_extension = Mock(LineExtension)
created_line_extension = line_extension_associate.return_value = Mock(
LineExtension, line_id=1, extension_id=2)
main_user_line = find_main_user_line.return_value = Mock(UserLine,
user_id=3,
line_id=1)
result = ule_service.associate_line_extension(line_extension)
assert_that(result, equal_to(created_line_extension))
find_main_user_line.assert_called_once_with(
created_line_extension.line_id)
fix_associations.assert_called_once_with(main_user_line)
def _create_internal_association(line_extension):
ule_services.associate_line_extension(line_extension)
def _create_internal_association(line_extension):
ule_services.associate_line_extension(line_extension)