class TestPacketsPerSecond(unittest.TestCase):
def reinit(self):
self.__stringbuffer1 = StringIO.StringIO()
self.__stringbuffer2 = StringIO.StringIO()
self.__p1 = PacketsPerSecond(self.__stringbuffer1)
self.__p2 = PacketsPerSecond(self.__stringbuffer2)
def data_example(self):
na = 2
nq = 2
data = PacketsExample({'na': na, 'nq': nq})
for i in range(na):
data.add_packet({'flags': '8000'})
for i in range(nq):
data.add_packet({'flags': '0'})
return data
def setUp(self):
self.reinit()
def test_right_format(self):
self.reinit()
result = self.__p1.get_data()
self.assertEquals(type(result), dict)
self.assertTrue('pps' in result)
self.assertEqual(type(result['pps']), float)
self.assertGreaterEqual(result['pps'], 0)
def test_no_data(self):
self.reinit()
result = self.__p1.get_data()
self.assertEqual(result['pps'], 0)
def test_data_example(self):
before_init_time = time()
self.reinit()
after_init_time = time()
example = self.data_example()
for packet in example:
self.__p1(packet)
before_end_time = time()
result = self.__p1.get_data()
after_end_time = time()
na = example.expected_value('na')
nq = example.expected_value('nq')
self.assertLessEqual((na + nq) / (after_end_time - before_init_time),
result['pps'])
self.assertGreaterEqual(
(na + nq) / (before_end_time - after_init_time),
result['pps'])
def test_reset(self):
before_init_time = time()
self.reinit()
after_init_time = time()
example = self.data_example()
for i in range(2):
for packet in example:
self.__p1(packet)
before_end_time = time()
result = self.__p1.get_data()
after_end_time = time()
na = example.expected_value('na')
nq = example.expected_value('nq')
self.assertLessEqual(
(na + nq) / (after_end_time - before_init_time),
result['pps'])
self.assertGreaterEqual(
(na + nq) / (before_end_time - after_init_time),
result['pps'])
before_init_time = time()
self.__p1.reset()
after_init_time = time()
def test_file(self):
self.reinit()
self.assertEquals(self.__stringbuffer1, self.__p1.get_file())
def tearDown(self):
pass
def reinit(self):
self.__stringbuffer1 = StringIO.StringIO()
self.__stringbuffer2 = StringIO.StringIO()
self.__p1 = PacketsPerSecond(self.__stringbuffer1)
self.__p2 = PacketsPerSecond(self.__stringbuffer2)
