From 99e426d0e69efab5a114549f14cd7e110de1facc Mon Sep 17 00:00:00 2001 From: Andy Babic Date: Wed, 19 Jun 2019 08:54:41 +0100 Subject: [PATCH] Fix base class reference --- wagtailmenus/api/v1/tests/test_base_views.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/wagtailmenus/api/v1/tests/test_base_views.py b/wagtailmenus/api/v1/tests/test_base_views.py index bbc0253af..8a2b2d76a 100644 --- a/wagtailmenus/api/v1/tests/test_base_views.py +++ b/wagtailmenus/api/v1/tests/test_base_views.py @@ -1,12 +1,12 @@ from django.test import override_settings, TestCase -from wagtailmenus.api.v1.views import MenuGeneratorView +from wagtailmenus.api.v1.views import BaseMenuGeneratorView from wagtailmenus.api.v1.serializers import FlatMenuSerializer, MainMenuSerializer class TestGetSerializerClass(TestCase): - class CustomView(MenuGeneratorView): + class CustomView(BaseMenuGeneratorView): default_serializer_class = MainMenuSerializer serializer_class_setting_name = 'MAIN_MENU_SERIALIZER'