Fix custom fields tests

allegro · Nov 7, 2024 · b5efa9f · b5efa9f
1 parent 8113670
commit b5efa9f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/ralph/lib/custom_fields/tests/admin.py b/src/ralph/lib/custom_fields/tests/admin.py
@@ -1,6 +1,7 @@
 from django.contrib import admin
 
-from ..admin import CustomFieldValueAdminMixin
+from ..admin import CustomFieldAdmin, CustomFieldValueAdminMixin
+from ..models import CustomField
 from .models import ModelA, ModelB, SomeModel
 
 site = admin.AdminSite(name="cf_admin")
@@ -24,3 +25,4 @@ class ModelBAdmin(CustomFieldValueAdminMixin, admin.ModelAdmin):
 site.register(SomeModel, SomeModelAdmin)
 site.register(ModelA, ModelAAdmin)
 site.register(ModelB, ModelBAdmin)
+site.register(CustomField, CustomFieldAdmin)