Răsfoiți Sursa

Fix CustomLinkButtonClassChoices references in tests

jeremystretch 4 ani în urmă
părinte
comite
8e69961744
1 a modificat fișierele cu 2 adăugiri și 2 ștergeri
  1. 2 2
      netbox/extras/tests/test_views.py

+ 2 - 2
netbox/extras/tests/test_views.py

@@ -68,7 +68,7 @@ class CustomLinkTestCase(ViewTestCases.PrimaryObjectViewTestCase):
             'name': 'Custom Link X',
             'content_type': site_ct.pk,
             'weight': 100,
-            'button_class': CustomLinkButtonClassChoices.CLASS_DEFAULT,
+            'button_class': CustomLinkButtonClassChoices.DEFAULT,
             'link_text': 'Link X',
             'link_url': 'http://example.com/?x'
         }
@@ -81,7 +81,7 @@ class CustomLinkTestCase(ViewTestCases.PrimaryObjectViewTestCase):
         )
 
         cls.bulk_edit_data = {
-            'button_class': CustomLinkButtonClassChoices.CLASS_CYAN,
+            'button_class': CustomLinkButtonClassChoices.CYAN,
             'weight': 200,
         }