forked from osmocom/wireshark
96 lines
3.5 KiB
Python
96 lines
3.5 KiB
Python
#
|
|
# Wireshark tests
|
|
# By Gerald Combs <gerald@wireshark.org>
|
|
#
|
|
# Copyright (c) 2018 Dario Lombardo <lomato@gmail.com>
|
|
#
|
|
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
#
|
|
'''outputformats tests'''
|
|
|
|
import json
|
|
import os.path
|
|
import subprocesstest
|
|
import fixtures
|
|
from matchers import *
|
|
|
|
|
|
@fixtures.fixture
|
|
def check_outputformat(cmd_tshark, request, dirs, capture_file):
|
|
def check_outputformat_real(format_option, pcap_file='dhcp.pcap',
|
|
extra_args=[], expected=None, multiline=False):
|
|
''' Check a capture file against a sample, in json format. '''
|
|
self = request.instance
|
|
tshark_proc = self.assertRun([cmd_tshark, '-r', capture_file(pcap_file),
|
|
'-T', format_option] + extra_args)
|
|
|
|
# If a filename is given, load the expected values from those.
|
|
if isinstance(expected, str):
|
|
testdata = open(os.path.join(dirs.baseline_dir, expected)).read()
|
|
if multiline:
|
|
expected = [json.loads(line) for line in testdata.splitlines()]
|
|
else:
|
|
expected = json.loads(testdata)
|
|
actual = tshark_proc.stdout_str
|
|
if multiline:
|
|
actual = actual.splitlines()
|
|
self.assertEqual(len(expected), len(actual))
|
|
for expectedObj, actualStr in zip(expected, actual):
|
|
self.assertEqual(expectedObj, json.loads(actualStr))
|
|
else:
|
|
actual = json.loads(actual)
|
|
self.assertEqual(expected, actual)
|
|
|
|
return check_outputformat_real
|
|
|
|
|
|
@fixtures.mark_usefixtures('base_env')
|
|
@fixtures.uses_fixtures
|
|
class case_outputformats(subprocesstest.SubprocessTestCase):
|
|
maxDiff = 1000000
|
|
|
|
def test_outputformat_json(self, check_outputformat):
|
|
'''Decode some captures into json'''
|
|
check_outputformat("json", expected="dhcp.json")
|
|
|
|
def test_outputformat_jsonraw(self, check_outputformat):
|
|
'''Decode some captures into jsonraw'''
|
|
check_outputformat("jsonraw", expected="dhcp.jsonraw")
|
|
|
|
def test_outputformat_ek(self, check_outputformat):
|
|
'''Decode some captures into ek'''
|
|
check_outputformat("ek", expected="dhcp.ek", multiline=True)
|
|
|
|
def test_outputformat_ek_raw(self, check_outputformat):
|
|
'''Decode some captures into ek, with raw data'''
|
|
check_outputformat("ek", expected="dhcp-raw.ek", multiline=True, extra_args=['-x'])
|
|
|
|
def test_outputformat_json_select_field(self, check_outputformat):
|
|
'''Checks that the -e option works with -Tjson.'''
|
|
check_outputformat("json", extra_args=['-eframe.number', '-c1'], expected=[
|
|
{
|
|
"_index": "packets-2004-12-05",
|
|
"_type": "doc",
|
|
"_score": None,
|
|
"_source": {
|
|
"layers": {
|
|
"frame.number": [
|
|
"1"
|
|
]
|
|
}
|
|
}
|
|
}
|
|
])
|
|
|
|
def test_outputformat_ek_select_field(self, check_outputformat):
|
|
'''Checks that the -e option works with -Tek.'''
|
|
check_outputformat("ek", extra_args=['-eframe.number', '-c1'], expected=[
|
|
{"index": {"_index": "packets-2004-12-05", "_type": "doc"}},
|
|
{"timestamp": "1102274184317", "layers": {"frame_number": ["1"]}}
|
|
], multiline=True)
|
|
|
|
def test_outputformat_ek_filter_field(self, check_outputformat):
|
|
''' Check that the option -j works with -Tek.'''
|
|
check_outputformat("ek", extra_args=['-j', 'dhcp'], expected="dhcp-filter.ek",
|
|
multiline=True)
|