mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
33 lines
635 B
Python
33 lines
635 B
Python
from django.test import TestCase
|
|
|
|
from utilities.choices import ChoiceSet
|
|
|
|
|
|
class ExampleChoices(ChoiceSet):
|
|
|
|
CHOICE_A = 'a'
|
|
CHOICE_B = 'b'
|
|
CHOICE_C = 'c'
|
|
CHOICE_1 = 1
|
|
CHOICE_2 = 2
|
|
CHOICE_3 = 3
|
|
|
|
CHOICES = (
|
|
('Letters', (
|
|
(CHOICE_A, 'A'),
|
|
(CHOICE_B, 'B'),
|
|
(CHOICE_C, 'C'),
|
|
)),
|
|
('Digits', (
|
|
(CHOICE_1, 'One'),
|
|
(CHOICE_2, 'Two'),
|
|
(CHOICE_3, 'Three'),
|
|
)),
|
|
)
|
|
|
|
|
|
class ChoiceSetTestCase(TestCase):
|
|
|
|
def test_values(self):
|
|
self.assertListEqual(ExampleChoices.values(), ['a', 'b', 'c', 1, 2, 3])
|