def testFlowDeserializationOptionalMissing():
hopDict = {
"dscp": 0,
"dstIp": "2.1.1.1",
"dstPort": 0,
"ecn": 0,
"fragmentOffset": 0,
"icmpCode": 255,
"icmpVar": 255,
"ingressNode": "ingress",
"ipProtocol": "IP",
"packetLength": 0,
"srcIp": "5.5.1.1",
"srcPort": 0,
"state": "NEW",
"tag": "BASE",
"tcpFlagsAck": 0,
"tcpFlagsCwr": 0,
"tcpFlagsEce": 0,
"tcpFlagsFin": 0,
"tcpFlagsPsh": 0,
"tcpFlagsRst": 0,
"tcpFlagsSyn": 0,
"tcpFlagsUrg": 0
}
# check deserialization
flow = Flow(hopDict)
assert flow.srcIp == "5.5.1.1"
# should convert to string without problems
str(flow)
def _get_display_value(schema, json_object):
# type (str, Any) -> Any
if json_object is None:
return None
if _is_list_or_set_schema(schema):
if not isinstance(json_object, list):
raise ValueError("Got non-list value for list/set schema", schema,
":", json_object)
output_list = [
str(_get_display_value(_get_base_schema(schema), element))
for element in json_object
]
if _get_base_schema(schema) == "FlowTrace":
return "\n".join(output_list)
else:
return output_list
if schema == "AclTrace":
return AclTrace(json_object)
if schema == "Environment":
return Environment(json_object)
if schema == "FileLines":
return FileLines(json_object)
if schema == "Flow":
return Flow(json_object)
if schema == "FlowTrace":
return FlowTrace(json_object)
if schema == "Integer":
return int(json_object)
if schema == "Interface":
return Interface(json_object)
if schema == "Ip":
return str(json_object)
if schema == "Issue":
return Issue(json_object)
if schema == "Node":
return json_object["name"]
if schema == "Prefix":
return str(json_object)
if schema == "SelfDescribing":
return _get_display_value(json_object["schema"],
json_object.get("value"))
if schema == "String":
return str(json_object)
return json_object
def testFlowDeserialization():
hopDict = {
"dscp": 0,
"dstIp": "2.1.1.1",
"dstPort": 0,
"ecn": 0,
"fragmentOffset": 0,
"icmpCode": 255,
"icmpVar": 255,
"ingressInterface": "intface",
"ingressNode": "ingress",
"ingressVrf": "vrfAbc",
"ipProtocol": "IP",
"packetLength": 0,
"srcIp": "5.5.1.1",
"srcPort": 0,
"state": "NEW",
"tag": "BASE",
"tcpFlagsAck": 0,
"tcpFlagsCwr": 0,
"tcpFlagsEce": 0,
"tcpFlagsFin": 0,
"tcpFlagsPsh": 0,
"tcpFlagsRst": 0,
"tcpFlagsSyn": 0,
"tcpFlagsUrg": 0
}
# check deserialization
flow = Flow(hopDict)
assert flow.srcIp == "5.5.1.1"
assert flow.ingressInterface == "intface"
assert flow.ingressVrf == "vrfAbc"
# check the string representation has the essential elements (without forcing a strict format)
flowStr = str(flow)
assert "5.5.1.1:0" in flowStr
assert "intface" in flowStr
assert "vrfAbc" in flowStr
def __init__(self, dictionary):
self.edge = Edge(dictionary["edge"])
self.routes = list(dictionary.get("routes", []))
transformed_flow = dictionary.get("transformedFlow")
self.transformedFlow = Flow(
transformed_flow) if transformed_flow else None
def test_header_constraints_of():
hc = HeaderConstraints.of(
Flow(
ipProtocol='ICMP',
icmpCode=7,
srcIp="1.1.1.1",
dstIp="2.2.2.2",
dscp=0,
dstPort=0,
srcPort=0,
ecn=0,
fragmentOffset=0,
icmpVar=0,
ingressInterface='',
ingressNode='',
ingressVrf='',
packetLength=0,
state='new',
tag='tag',
tcpFlagsAck=0,
tcpFlagsCwr=0,
tcpFlagsEce=0,
tcpFlagsFin=0,
tcpFlagsPsh=0,
tcpFlagsRst=0,
tcpFlagsSyn=0,
tcpFlagsUrg=0,
))
assert hc.srcIps == "1.1.1.1"
assert hc.dstIps == "2.2.2.2"
assert hc.ipProtocols == ['ICMP']
assert hc.icmpCodes == '7'
assert hc.srcPorts is None
assert hc.dstPorts is None
assert hc.tcpFlags is None
hc = HeaderConstraints.of(
Flow(
ipProtocol='TCP',
srcPort=1000,
dstPort=2000,
tcpFlagsAck=True,
srcIp="1.1.1.1",
dstIp="2.2.2.2",
dscp=0,
ecn=0,
fragmentOffset=0,
icmpCode=0,
icmpVar=0,
ingressInterface='',
ingressNode='',
ingressVrf='',
packetLength=0,
state='new',
tag='tag',
tcpFlagsCwr=0,
tcpFlagsEce=0,
tcpFlagsFin=0,
tcpFlagsPsh=0,
tcpFlagsRst=0,
tcpFlagsSyn=0,
tcpFlagsUrg=0,
))
assert hc.srcIps == "1.1.1.1"
assert hc.dstIps == "2.2.2.2"
assert hc.ipProtocols == ['TCP']
assert hc.icmpCodes is None
assert hc.icmpTypes is None
assert hc.srcPorts == '1000'
assert hc.dstPorts == '2000'
assert hc.tcpFlags == [
MatchTcpFlags(
TcpFlags(ack=True),
useAck=True,
useCwr=True,
useEce=True,
useFin=True,
usePsh=True,
useRst=True,
useSyn=True,
useUrg=True,
)
]