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)