def test_wrapper_processing(self): """ Tests that the wrapper can process a valid mib file without errors. """ result = mib2pysnmp('conpot/tests/data/VOGON-POEM-MIB.mib') self.assertTrue('mibBuilder.exportSymbols("VOGON-POEM-MIB"' in result, 'mib2pysnmp 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. """ try: tmpdir = tempfile.mkdtemp() result = mib2pysnmp('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], None) cmd_responder.snmpEngine.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)