redash/tests/handlers/test_favorites.py

from tests import BaseTestCase
from redash import models
from redash.models import db

class TestQueryFavoriteResource(BaseTestCase):
    def test_favorite(self):
        query = self.factory.create_query()

        rv = self.make_request('post', '/api/queries/{}/favorite'.format(query.id))
        self.assertEqual(rv.status_code, 200)

        rv = self.make_request('get', '/api/queries/{}'.format(query.id))
        self.assertEqual(rv.json['is_favorite'], True)
    
    def test_duplicate_favorite(self):
        query = self.factory.create_query()

        rv = self.make_request('post', '/api/queries/{}/favorite'.format(query.id))
        self.assertEqual(rv.status_code, 200)

        rv = self.make_request('post', '/api/queries/{}/favorite'.format(query.id))
        self.assertEqual(rv.status_code, 200)


    def test_unfavorite(self):
        query = self.factory.create_query()
        rv = self.make_request('post', '/api/queries/{}/favorite'.format(query.id))
        rv = self.make_request('delete', '/api/queries/{}/favorite'.format(query.id))
        self.assertEqual(rv.status_code, 200)

        rv = self.make_request('get', '/api/queries/{}'.format(query.id))
        self.assertEqual(rv.json['is_favorite'], False)

class TestQueryFavoriteListResource(BaseTestCase):
    def test_get_favorites(self):
        rv = self.make_request('get', '/api/queries/favorites')
        self.assertEqual(rv.status_code, 200)