-
Notifications
You must be signed in to change notification settings - Fork 2
/
PacketInserterTest.py
41 lines (40 loc) · 1.68 KB
/
PacketInserterTest.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
import unittest
from PacketInserter import *
from scapy.all import *
import states.InserterState as state
import tcpSynFlood.TCPPacketBuilder as pktBuilder
class PacketInserterTest(unittest.TestCase):
def setUp(self):
self.inserter = PacketInserter()
def test_well_construct_test(self):
ins = self.inserter
self.assertEqual([],ins.getPacketsToAppend())
self.assertEqual("",ins.getInputName())
self.assertEqual("", ins.getOutputName())
self.assertEqual("",ins.getInputDir())
self.assertEqual("",ins.getOutputDir())
self.assertAlmostEqual(0.006,ins.getResponseDt())
self.assertEqual(state.ReadOkState(ins),ins.getState())
self.assertAlmostEqual(0.001,ins.getTimestamp())
self.assertEqual(30,ins.getServerTolerance())
def test_setter_getter(self):
ins = self.inserter
dullPacket = Ether()/IP()
ins.withPackets([dullPacket])
self.assertEqual([dullPacket],ins.getPacketsToAppend())
ins.withPcapInput("lol.pcap")
self.assertEqual("lol.pcap",ins.getInputName())
ins.withInputDir("input/")
self.assertEqual("input/",ins.getInputDir())
ins.withPcapOutput("lol-modified.pcap")
self.assertEqual("lol-modified.pcap",ins.getOutputName())
ins.withResponseDt(0.008)
self.assertAlmostEqual(0.008,ins.getResponseDt())
ins.withServerIp("200.3.1.2")
self.assertEqual("200.3.1.2",ins.getServerIp())
ins.withTimestamp(0.01)
self.assertAlmostEqual(0.01,ins.getTimestamp())
ins.withServerTolerance(30)
self.assertEqual(30,ins.getServerTolerance())
if __name__ == "__main__":
unittest.main()