diff --git a/kepi/trilby_api/tests/__init__.py b/kepi/trilby_api/tests/__init__.py index ecdbf5c..c600cd9 100644 --- a/kepi/trilby_api/tests/__init__.py +++ b/kepi/trilby_api/tests/__init__.py @@ -4,7 +4,8 @@ # Copyright (c) 2018-2021 Marnanel Thurman. # Licensed under the GNU Public License v2. -from django.test import TestCase, Client +from django.test import Client +from kepi.kepi.testing import KepiTestCase from rest_framework.test import force_authenticate, APIClient from kepi.trilby_api.models import * from django.conf import settings @@ -65,7 +66,7 @@ STATUS_EXPECTED = { 'pinned': False, } -class TrilbyTestCase(TestCase): +class TrilbyTestCase(KepiTestCase): def setUp(self):