def test_wrapper_processing(self):
"""
Tests that the wrapper can process a valid mib file without errors.
"""
result = mib2pyopc('conpot/tests/data/VOGON-POEM-MIB.mib')
self.assertTrue('mibBuilder.exportSymbols("VOGON-POEM-MIB"' in result,
'mib2pyopc did not generate the expected output. Output: {0}'.format(result))
def test_wrapper_output(self):
"""
Tests that the wrapper generates output that can be consumed by the command responder.
"""
tmpdir = None
try:
tmpdir = tempfile.mkdtemp()
result = mib2pyopc('conpot/tests/data/VOGON-POEM-MIB.mib')
with open(os.path.join(tmpdir, 'VOGON-POEM-MIB' + '.py'), 'w') as output_file:
output_file.write(result)
cmd_responder = command_responder.CommandResponder('', 0, [tmpdir])
cmd_responder.opcEngine.msgAndPduDsp.mibInstrumController.mibBuilder.loadModules('VOGON-POEM-MIB')
result = cmd_responder._get_mibSymbol('VOGON-POEM-MIB', 'poemNumber')
self.assertIsNotNone(result, 'The expected MIB (VOGON-POEM-MIB) could not be loaded.')
finally:
shutil.rmtree(tmpdir)
