mirror of
https://github.com/valitydev/redash.git
synced 2024-11-07 09:28:51 +00:00
37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
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) |