def test_udp(self): ct = get_port_range_type_from_string('udp') self.assertEqual(ct, PortRangeType.UDP)
def test_none_or_empty(self): ct = get_port_range_type_from_string(None) self.assertIsNone(ct) ct = get_port_range_type_from_string('') self.assertIsNone(ct)
def test_tcp(self): ct = get_port_range_type_from_string('tcp') self.assertEqual(ct, PortRangeType.TCP)
def test_invalid(self): with self.assertRaises(InvalidArgument): get_port_range_type_from_string('foo')