class TestEnum(unittest.TestCase): def setUp(self): self.registry = CollectorRegistry() self.enum = Enum('e', 'help', states=['a', 'b', 'c'], registry=self.registry) self.labels = Enum('el', 'help', ['l'], states=['a', 'b', 'c'], registry=self.registry) def test_enum(self): self.assertEqual(1, self.registry.get_sample_value('e', {'e': 'a'})) self.assertEqual(0, self.registry.get_sample_value('e', {'e': 'b'})) self.assertEqual(0, self.registry.get_sample_value('e', {'e': 'c'})) self.enum.state('b') self.assertEqual(0, self.registry.get_sample_value('e', {'e': 'a'})) self.assertEqual(1, self.registry.get_sample_value('e', {'e': 'b'})) self.assertEqual(0, self.registry.get_sample_value('e', {'e': 'c'})) self.assertRaises(ValueError, self.enum.state, 'd') self.assertRaises(ValueError, Enum, 'e', 'help', registry=None) def test_labels(self): self.labels.labels('a').state('c') self.assertEqual(0, self.registry.get_sample_value('el', {'l': 'a', 'el': 'a'})) self.assertEqual(0, self.registry.get_sample_value('el', {'l': 'a', 'el': 'b'})) self.assertEqual(1, self.registry.get_sample_value('el', {'l': 'a', 'el': 'c'})) self.assertRaises(ValueError, self.labels.state, 'a') def test_overlapping_labels(self): with pytest.raises(ValueError): Enum('e', 'help', registry=None, labelnames=['e'])
def setUp(self): self.registry = CollectorRegistry() self.enum = Enum('e', 'help', states=['a', 'b', 'c'], registry=self.registry) self.labels = Enum('el', 'help', ['l'], states=['a', 'b', 'c'], registry=self.registry)
def test_labels(self): self.labels.labels('a').state('c') self.assertEqual( 0, self.registry.get_sample_value('el', { 'l': 'a', 'el': 'a' })) self.assertEqual( 0, self.registry.get_sample_value('el', { 'l': 'a', 'el': 'b' })) self.assertEqual( 1, self.registry.get_sample_value('el', { 'l': 'a', 'el': 'c' })) e = Enum('e', 'help', registry=None, labelnames=['e']) self.assertRaises(ValueError, e.labels, '')
def test_overlapping_labels(self): with pytest.raises(ValueError): Enum('e', 'help', registry=None, labelnames=['e'])