redash/tests/models/test_api_keys.py
2016-03-10 10:09:26 +02:00

18 lines
634 B
Python

from tests import BaseTestCase
from redash.models import ApiKey
class TestApiKeyGetByObject(BaseTestCase):
def test_returns_none_if_not_exists(self):
dashboard = self.factory.create_dashboard()
self.assertIsNone(ApiKey.get_by_object(dashboard))
def test_returns_only_active_key(self):
dashboard = self.factory.create_dashboard()
api_key = self.factory.create_api_key(object=dashboard, active=False)
self.assertIsNone(ApiKey.get_by_object(dashboard))
api_key = self.factory.create_api_key(object=dashboard)
self.assertEqual(api_key, ApiKey.get_by_object(dashboard))