class Test(unittest.TestCase):
def setUp(self):
self._PL = PcapLoader( HtmlPcapDataFile )
def testHeaderOnlyEventCount(self):
''' HeaderOnly is default '''
eventcount = 0
for event in self._PL: #@UnusedVariable
if event:
eventcount += 1
self.assertEqual( HeaderOnlyEventCount, eventcount )
pass
def testInstallReader(self):
DR = DnsReader()
CI = ConnInfo( proto = 'UDP', dport=53)
self._PL.ClearReaderList()
self._PL.setReader( DR, CI )
for event in self._PL: #@UnusedVariable
#print event
pass
self.assertEqual( len(DR.GetDnsEntries()), DnsEntriesInPcap )
def testStreamReader(self):
SR = StreamReader()
CI = ConnInfo( proto = 'TCP', dip = '69.4.231.52', dport=80)
self._PL.ClearReaderList()
self._PL.setReader( SR, CI )
for event in self._PL: #@UnusedVariable
#print event
pass
self.assertEqual( len(SR.GetDialogue()), DialogueInConnection )
def testHttpReader(self):
HR = HttpReader()
CI = ConnInfo( proto = 'TCP', dip = '69.4.231.52', dport=80)
self._PL.ClearReaderList()
self._PL.setReader( HR, CI )
i = 0
for event in self._PL: #@UnusedVariable
#print event
self.assertEqual(event, HttpDialogue[i])
i += 1
pass
self.assertEqual( len(HR.GetDialogue()), HttpDialogueCount )
def setUp(self):
self._PL = PcapLoader( HtmlPcapDataFile )