mirror of https://github.com/Kkevsterrr/geneva
Added test for trace error cases
This commit is contained in:
parent
b36cb7f438
commit
33566f9cd2
|
@ -0,0 +1,36 @@
|
|||
import logging
|
||||
import sys
|
||||
|
||||
# Include the root of the project
|
||||
sys.path.append("..")
|
||||
|
||||
|
||||
import actions.strategy
|
||||
import actions.packet
|
||||
import actions.utils
|
||||
import actions.trace
|
||||
import actions.layer
|
||||
|
||||
from scapy.all import IP, TCP, UDP, DNS, DNSQR, sr1
|
||||
|
||||
|
||||
logger = logging.getLogger("test")
|
||||
|
||||
|
||||
def test_trace_error_cases():
|
||||
"""
|
||||
Tests that trace handles edge cases.
|
||||
"""
|
||||
# No IP header means the packet should just be returned
|
||||
packet = actions.packet.Packet(TCP(sport=2222, dport=3333, seq=100, ack=100, flags="S"))
|
||||
trace = actions.trace.TraceAction(None)
|
||||
p1, p2 = trace.run(packet, logger)
|
||||
assert p2 is None
|
||||
assert p1 == packet
|
||||
|
||||
# Mark the trace as having run already - it should not run again
|
||||
trace.ran = True
|
||||
packet = actions.packet.Packet(IP(src="127.0.0.1", dst="127.0.0.1")/TCP(sport=2222, dport=3333, seq=100, ack=100, flags="S"))
|
||||
p1, p2 = trace.run(packet, logger)
|
||||
assert p1 is None
|
||||
assert p2 is None
|
Loading…
Reference in New Issue