def delete_line(user): try: line = line_services.get_by_user_id(user.id) except NotFoundError: return else: line_services.delete(line)
def test_get_by_user_id(self, mock_get_by_user_id): user_id = 1 line = Mock() mock_get_by_user_id.return_value = line result = line_services.get_by_user_id(user_id) mock_get_by_user_id.assert_called_once_with(user_id) self.assertEquals(result, line)