From 45a9a4d0c59968747ce71c3ac1faab179dbf1543 Mon Sep 17 00:00:00 2001
From: Matthias Mair <code@mjmair.com>
Date: Fri, 20 May 2022 17:55:11 +0200
Subject: [PATCH] add in missing class references

---
 InvenTree/part/test_views.py                     | 5 +++--
 InvenTree/plugin/base/integration/test_mixins.py | 3 ++-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/InvenTree/part/test_views.py b/InvenTree/part/test_views.py
index f20fc3159f..60fba1fd97 100644
--- a/InvenTree/part/test_views.py
+++ b/InvenTree/part/test_views.py
@@ -1,12 +1,13 @@
 """ Unit tests for Part Views (see views.py) """
 
-from django.test import TestCase
 from django.urls import reverse
 
+from InvenTree.helpers import InvenTreeTestCase
+
 from .models import Part
 
 
-class PartViewTestCase(TestCase):
+class PartViewTestCase(InvenTreeTestCase):
 
     fixtures = [
         'category',
diff --git a/InvenTree/plugin/base/integration/test_mixins.py b/InvenTree/plugin/base/integration/test_mixins.py
index 20a99e71bd..f60e6a2802 100644
--- a/InvenTree/plugin/base/integration/test_mixins.py
+++ b/InvenTree/plugin/base/integration/test_mixins.py
@@ -7,6 +7,7 @@ from django.urls import include, re_path, reverse
 
 from error_report.models import Error
 
+from InvenTree.helpers import InvenTreeTestCase
 from plugin import InvenTreePlugin
 from plugin.helpers import MixinNotImplementedError
 from plugin.mixins import (APICallMixin, AppMixin, NavigationMixin,
@@ -250,7 +251,7 @@ class APICallMixinTest(BaseMixinDefinition, TestCase):
             self.mixin_wrong2.has_api_call()
 
 
-class PanelMixinTests(TestCase):
+class PanelMixinTests(InvenTreeTestCase):
     """Test that the PanelMixin plugin operates correctly"""
 
     fixtures = [