def setUp(self): self.command = Command()
class XML_Import_Tests(CustomSettingsTestCase): new_settings = dict( INSTALLED_APPS=( 'dingos', ) ) def setUp(self): self.command = Command() def common_import_delta(self, xml_file): """ Returns the resulting list of elements parsing a given XML file in IODEF format """ @deltaCalc def t_import(*args,**kwargs): return self.command.handle(*args,**kwargs) (delta,result) = t_import(xml_file, identifier_ns_uri=None) #pp.pprint(delta) return delta def test_botnet_example_import(self,show_result=SHOW_RESULTS): expected = [('DataTypeNameSpace', 2), ('Fact', 34), ('FactDataType', 1), ('FactTerm', 28), ('FactTerm2Type', 28), ('FactValue', 34), ('Identifier', 1), ('IdentifierNameSpace', 1), ('InfoObject', 1), ('InfoObject2Fact', 40), ('InfoObjectFamily', 1), ('InfoObjectType', 1), ('NodeID', 40), ('Revision', 1)] result = self.common_import_delta('tests/mocks/botnet_iodef.xml') if show_result: pp.pprint(result) else: self.assertEqual( expected, result ) def test_scan_example(self,show_result=SHOW_RESULTS): expected = [ ('DataTypeNameSpace', 2), ('Fact', 30), ('FactDataType', 1), ('FactTerm', 24), ('FactTerm2Type', 24), ('FactValue', 33), ('Identifier', 1), ('IdentifierNameSpace', 1), ('InfoObject', 1), ('InfoObject2Fact', 36), ('InfoObjectFamily', 1), ('InfoObjectType', 1), ('NodeID', 36), ('Revision', 1)] result = self.common_import_delta('tests/mocks/scan_iodef.xml') if show_result: pp.pprint(result) else: self.assertEqual( expected, result ) def test_worm_example(self,show_result=SHOW_RESULTS): expected = [ ('DataTypeNameSpace', 3), ('Fact', 32), ('FactDataType', 3), ('FactTerm', 28), ('FactTerm2Type', 28), ('FactValue', 31), ('Identifier', 1), ('IdentifierNameSpace', 1), ('InfoObject', 1), ('InfoObject2Fact', 32), ('InfoObjectFamily', 1), ('InfoObjectType', 1), ('NodeID', 32), ('Revision', 1)] result = self.common_import_delta('tests/mocks/worm_iodef.xml') if show_result: pp.pprint(result) else: self.assertEqual( expected, result )