Exemplo n.º 1
0
def delete_line(user):
    try:
        line = line_services.get_by_user_id(user.id)
    except NotFoundError:
        return
    else:
        line_services.delete(line)
Exemplo n.º 2
0
    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)