def test_given_extension_associated_then_caller_id_and_exten_removed(self,
remove_caller_id,
find_extension,
remove_exten_and_context):
line_id = 1
extension = find_extension.return_value = Mock(Extension)
ule_service.fix_main_user_dissociation(line_id)
remove_caller_id.assert_called_once_with(line_id)
find_extension.assert_called_once_with(line_id)
remove_exten_and_context.assert_called_once_with(extension)
def test_given_no_extension_associated_then_caller_id_removed(self,
remove_caller_id,
find_extension,
remove_exten_and_context):
line_id = 1
find_extension.return_value = None
ule_service.fix_main_user_dissociation(line_id)
remove_caller_id.assert_called_once_with(line_id)
find_extension.assert_called_once_with(line_id)
self.assertNotCalled(remove_exten_and_context)
