def test_init(self): new_netflow = _New_Netflow_v9(self.sample, self.count) eq_(new_netflow.data, self.sample) eq_(new_netflow.count, self.count) eq_(new_netflow._raw_data, '') eq_(new_netflow.length, list()) eq_(new_netflow.flowset_id, list()) #lsit of flowset id eq_(new_netflow.template_header_list, list()) #contains flowset_id and length eq_(new_netflow.flow_data_list, list()) #contains raw data eq_(new_netflow.template_template_list, list()) #contains template_id and field count eq_(new_netflow.data_header_list, list()) #contains template_id and length of the data eq_(new_netflow.template_data_dict, dict()) #contains templateflow data as template_id:data eq_(new_netflow.data_data_dict, list()) #contains dataflow data as template_id:data eq_(new_netflow.unparsed_raw_data, list()) eq_(new_netflow.now, int(time.time())) assert_not_equal(new_netflow.data, sample_v5) assert_not_equal(new_netflow.count, 1) assert_not_equal(new_netflow.unparsed_raw_data, sample_v9) assert_not_equal(new_netflow.now, int(time.time()) - 1)
def test_get_flowset_id_length_data(self): new_netflow = _New_Netflow_v9(self.sample, self.count) eq_(new_netflow.get_flowset_id_length_data(self.sample, self.count), None, "test get flowset id") eq_(new_netflow.length, [[80, 56]]) eq_(new_netflow.flowset_id, [0, 300], "test get flowset id") eq_(new_netflow.template_header_list, [(0, 80), (300, 56)]) eq_(new_netflow.flow_data_list, [ '\x00\x00\x00P\x01,\x00\x12\x00\x08\x00\x04\x00\x0c\x00\x04\x00\x0f\x00\x04\x00\n\x00\x04\x00\x0e\x00\x04\x00\x02\x00\x04\x00\x01\x00\x04\x00\x07\x00\x02\x00\x0b\x00\x02\x00\x06\x00\x01\x00\x04\x00\x01\x00\x05\x00\x01\x00\x11\x00\x02\x00\x10\x00\x02\x00\t\x00\x01\x00\r\x00\x01\x00\x15\x00\x04\x00\x16\x00\x04', '\x01,\x008\n\x00\x00\x02\n\x00\x00\x03\x00\x00\x00\x00\x00\x00\x00\x03\x00\x00\x00\x05\x00\x00\x00\x01\x00\x00\x00@\x10\x92\x00P\x00\x11\x01\x00\x03\x00\x02 \x1f\x01\xbc\x90G\x01\xbb\xa5\xe7\x00\x00\x00' ]) assert_not_equal(new_netflow.length, list()) assert_not_equal(new_netflow.flowset_id, list())
def test_clean_unparsed_raw_data(self): new_netflow = _New_Netflow_v9(self.sample, self.count) eq_(new_netflow.get_flowset_id_length_data(self.sample, self.count), None, "test get flowset id") eq_( new_netflow.get_raw_template_dict_raw_data_dict( sid, netflow_out, device_name, col_type, ip, collected_at), None, "test update template data list") eq_(new_netflow.update_template_data_list_dict(), None, "test update template data list") eq_(new_netflow.unparsed_raw_data, list()) assert_not_equal(new_netflow.flowset_id, list(self.sample)) assert_not_equal(new_netflow.flowset_id, dict())
def test_get_raw_template_dict_raw_data_dict(self): new_netflow = _New_Netflow_v9(self.sample, self.count) eq_(new_netflow.get_flowset_id_length_data(self.sample, self.count), None, "test get flowset id") eq_( new_netflow.get_raw_template_dict_raw_data_dict( sid, netflow_out, device_name, col_type, ip, collected_at), None, "get_raw_data") eq_(new_netflow.template_template_list, [(300, 18)]) expected_data_header_list = [( 300, 56, '\n\x00\x00\x02\n\x00\x00\x03\x00\x00\x00\x00\x00\x00\x00\x03\x00\x00\x00\x05\x00\x00\x00\x01\x00\x00\x00@\x10\x92\x00P\x00\x11\x01\x00\x03\x00\x02 \x1f\x01\xbc\x90G\x01\xbb\xa5\xe7\x00\x00\x00' )] eq_(new_netflow.data_header_list, expected_data_header_list) assert_not_equal(new_netflow.template_template_list, dict()) assert_not_equal(new_netflow.template_template_list, list()) assert_not_equal(new_netflow.template_template_list, list((300, 0)))