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)