tests/TestSSH.py

   1 #!/usr/bin/env python2.5
   2
   3 import unittest
   4 import sys
   5 import threading
   6 import random
   7 import time
   8
   9 import PcapPlayback
  10
  11 class SSHTestCase(PcapPlayback.PcapPlayback):
  12   def testSSH(self):
  13     pcap1="pcaps/illum-ssh.pcap"
  14     ip_in_pcap1="192.168.1.107"
  15     pcap2="pcaps/tapdance-ssh.pcap"
  16     ip_in_pcap2="192.168.1.135"
  17     self.pcap_playback(pcap1, ip_in_pcap1, pcap2, ip_in_pcap2, 63.0)
  18     time.sleep(5)
  19     self.server.judgement_day()
  20     time.sleep(2)
  21     flow_pairs, okay, leftovers, forged, dropped = self.server.print_global_flow_table()
  22     self.assertEqual(okay, 49)
  23     self.assertEqual(dropped, 4)
  24     self.assertEqual(forged, 0)
  25
  26 if __name__ == "__main__":
  27   PcapPlayback.main()