def get_virtual_service(self, index=None, address=None, port=None,
protocol=None):
if index is None:
validate_ip(address)
validate_port(port)
validate_protocol(protocol)
service_id = {"vs": address, "port": port, "prot": protocol}
else:
service_id = {"vs": index}
response = self._get("/showvs", service_id)
service = get_data(response)
# again line below will fail with ValidationError if empty response
virt_serv = self.build_virtual_service(service)
return virt_serv
def test_invalid(self):
with assert_raises(ValidationError):
validate_protocol('sds')
def test_udp(self):
assert_is_none(validate_protocol('udp'))
def test_TCP(self):
assert_is_none(validate_protocol('TCP'))