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 )