def test_not_expired(self):
frame = AutoVivification({'frame': {'time_epoch': {'raw': [123]}}})
flow = Flow(frame)
self.assertEqual(flow.not_expired(), True)
Flow.newest_overall_frame_time = 123
self.assertEqual(flow.not_expired(), True)
Flow.newest_overall_frame_time = 123 + Conf.FLOW_BUFFER_TIME
self.assertEqual(flow.not_expired(), False)
def test_get_flow_id(self):
Conf.FLOW_DEF = [['def1'], ['def2']]
frame = AutoVivification({
'def1': 1,
'def2': 2,
})
self.assertEqual(Flow.get_flow_id(frame), '[1, 2]')
frame = AutoVivification({
'def1': 1,
})
self.assertEqual(Flow.get_flow_id(frame), '[1, {}]')
frame = AutoVivification({
'def3': 3,
})
self.assertEqual(Flow.get_flow_id(frame), None)
def test_framecount(self):
Conf.FRAME_TIME = ['t']
Conf.FLOW_DEF = [['def1']]
f0 = AutoVivification({
't': 2,
'def1': 1,
'unimportant': 1,
})
f1 = AutoVivification({
't': 0,
'def1': 1,
'unimportant': 1,
})
f = Flow(f0)
f.add_frame(f1)
self.assertEqual(f.framecount, 2)
def test___hash__(self):
Conf.FRAME_TIME = ['t']
Conf.FLOW_DEF = [['def1'], ['def2']]
frame = AutoVivification({
't': 0,
'def1': 1,
'def2': 2,
})
self.assertEqual(hash(Flow(frame)), hash('[1, 2]'))
def test_id(self):
Conf.FRAME_TIME = ['t']
Conf.FLOW_DEF = [['def1']]
frame = AutoVivification({
't': 0,
'def1': 1,
})
self.assertEqual(
Flow(frame).id,
Flow.get_flow_id(frame),
)
frame2 = AutoVivification({
't': 0,
'def1': 2,
})
self.assertNotEqual(
Flow(frame2).id,
Flow.get_flow_id(frame),
)
def test_frames_returns_copy(self):
Conf.FRAME_TIME = ['t']
Conf.FLOW_DEF = [['def1']]
f = Flow(AutoVivification({
't': 0,
'def1': 1,
'unimportant': 1,
}))
f.add_frame(AutoVivification({
't': 0,
'def1': 1,
'unimportant': 1,
}))
frames_before_add = f.frames
f.frames['add'] = 'something'
self.assertEqual(
frames_before_add,
f.frames,
)
def test__eq__(self):
Conf.FRAME_TIME = ['t']
Conf.FLOW_DEF = [['def1']]
self.assertEqual(
Flow(AutoVivification({
't': 0,
'def1': 1,
})),
Flow(AutoVivification({
't': 0,
'def1': 1,
})),
)
self.assertNotEqual(
Flow(AutoVivification({
't': 0,
'def1': 1,
})),
Flow(AutoVivification({
't': 0,
'def1': 2,
})),
)
self.assertEqual(
Flow(AutoVivification({
't': 0,
'def1': 1,
'unimportant': 1,
})),
Flow(AutoVivification({
't': 0,
'def1': 1,
'unimportant': 2,
})),
)
def test_newest_frame_time(self):
Conf.FRAME_TIME = ['t']
Conf.FLOW_DEF = [['def1']]
f0 = AutoVivification({
't': 2,
'def1': 1,
'unimportant': 1,
})
f1 = AutoVivification({
't': 0,
'def1': 1,
'unimportant': 1,
})
f2 = AutoVivification({
't': 1,
'def1': 1,
'unimportant': 1,
})
f = Flow(f0)
f.add_frame(f1)
f.add_frame(f2)
self.assertEqual(f.newest_frame_time, f0['t'])