Преглед на файлове

Fix CustomLinkButtonClassChoices references in tests

jeremystretch преди 4 години
родител
ревизия
8e69961744
променени са 1 файла, в които са добавени 2 реда и са изтрити 2 реда
  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',
             'name': 'Custom Link X',
             'content_type': site_ct.pk,
             'content_type': site_ct.pk,
             'weight': 100,
             'weight': 100,
-            'button_class': CustomLinkButtonClassChoices.CLASS_DEFAULT,
+            'button_class': CustomLinkButtonClassChoices.DEFAULT,
             'link_text': 'Link X',
             'link_text': 'Link X',
             'link_url': 'http://example.com/?x'
             'link_url': 'http://example.com/?x'
         }
         }
@@ -81,7 +81,7 @@ class CustomLinkTestCase(ViewTestCases.PrimaryObjectViewTestCase):
         )
         )
 
 
         cls.bulk_edit_data = {
         cls.bulk_edit_data = {
-            'button_class': CustomLinkButtonClassChoices.CLASS_CYAN,
+            'button_class': CustomLinkButtonClassChoices.CYAN,
             'weight': 200,
             'weight': 200,
         }
         }