redash/tests/models/test_api_keys.py
Arik Fraimovich 2dff8b9a00
Black support for the Python codebase (#4297)
* Apply black formatting

* Add auto formatting when committing to master

* Update CONTRIBUTING.md re. Black & Prettier
2019-12-11 13:54:29 +02:00

17 lines
633 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))