test: Use BaseTestCase class

jamesx00 · Jun 17, 2022 · 3ba7c95 · 3ba7c95
1 parent 49294e8
commit 3ba7c95
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/django_omise/tests/test_core_utils.py b/django_omise/tests/test_core_utils.py
@@ -1,11 +1,13 @@
-from django.test import TestCase
-
 from django.contrib.auth import get_user_model
 
 from django_omise.models.core import Customer, Card, Charge
 from django_omise.models.choices import Currency
 from django_omise.omise import omise
-from django_omise.utils.core_utils import get_model_from_omise_object
+from django_omise.utils.core_utils import (
+    get_model_from_omise_object,
+)
+
+from django_omise.tests.base import OmiseBaseTestCase
 
 from unittest import mock
 
@@ -19,7 +21,7 @@
 User = get_user_model()
 
 # Create your tests here.
-class CoreUtilTestCase(TestCase):
+class CoreUtilTestCase(OmiseBaseTestCase):
     def setUp(self):
         self.user = User.objects.create_user(
             username="test_user1", email="test_user1@email.com"