def test_exportToScapy(self):
# Test if exportToScapy function in ScapyExporter is working correctly.
# Test if created file runs without error
# Verify the content of the exported file is as expected, including size info.
# check for both iterable and non-iterable symbols
m1 = RawMessage("\x09\x70\x95\xcc\xef")
m2 = RawMessage("\x0a\x70\x03\x8f\x23\x5f")
m3 = RawMessage("\x09\x70\x31\xa0")
m4 = RawMessage("\x0a\xd6\xb5\x5b")
messages = [m1, m2, m3, m4]
symbols = Symbol(messages=messages)
with self.assertRaises(TypeError):
iter(symbols)
self.assertEqual(
ScapyExporter().exportToScapy(symbols, 'test1.py', 'ProtocolName'),
None)
self.assertEqual(os.system("python test1.py"), 0)
import test1
self.assertEqual(test1.ProtocolName_Symbol().fields_desc[0].name,
'Field')
self.assertEqual(test1.ProtocolName_Symbol().fields_desc[0].default,
None)
self.assertEqual(test1.ProtocolName_Symbol().fields_desc[0].cls,
(0, None)) # size
with self.assertRaises(AttributeError):
test1.ProtocolName_Symbol1().fields_desc[0].name
os.remove('test1.py')
symbl = Format.clusterByAlignment(messages,
minEquivalence=50,
internalSlick=True)
self.assertEqual(type(symbl), list)
self.assertEqual(
ScapyExporter().exportToScapy(symbl, 'test2.py', 'ProtocolName'),
None)
self.assertEqual(os.system("python test2.py"), 0)
import test2
self.assertEqual(test2.ProtocolName_Symbol0().fields_desc[0].name,
'Field00')
self.assertEqual(test2.ProtocolName_Symbol0().fields_desc[0].cls,
(0, 32)) # size
self.assertEqual(test2.ProtocolName_Symbol1().fields_desc[1].default,
'p')
self.assertEqual(test2.ProtocolName_Symbol1().Field01, 'p')
self.assertEqual(len(test2.ProtocolName_Symbol1().fields_desc), 3)
os.remove('test2.py')
