def test_create__with_node(self): """Testing create instance with node.""" node = entities.Node('Entity', attrib={'Type': 'Test'}) entities.Node('Value', parent=node) entity = entities.Entity.from_node(node=node) self.assertIsInstance(entity, entities.Entity)
def test_create__without_entities(self): """Testing create instance with wrong node without entities.""" node = entities.Node('MaltegoMessage') entities.Node('MaltegoTransformRequestMessage', parent=node) with self.assertRaises(exceptions.MalformedMessageError): messages.TransformRequest.from_node(node=node)
def test_create__with_parent(self): """Testing create instance with parent.""" parent_name = 'Parent' parent = entities.Node(parent_name) name = 'Test' node = entities.Node(name, parent=parent) self.assertEqual(node.getparent(), parent)
def test_from_node__without_entities(self): """Testing create instance from node.""" node = entities.Node('MaltegoMessage') request = entities.Node('MaltegoTransformResponseMessage', parent=node) entities.Node('Entitie', parent=request) with self.assertRaises(exceptions.MalformedMessageError): messages.TransformResponse.from_node(node)
def test_create__node_weight(self): """Testing create instance with node weight.""" weight = '0' node = entities.Node('Entity', attrib={'Type': 'Test'}) entities.Node('Value', parent=node) entities.Node('Weight', value=weight, parent=node) entity = entities.Entity.from_node(node=node) self.assertEqual(entity.weight, weight)
def test_create__with_node_icon_url(self): """Testing create instance with node icon url.""" value = 'Test' node = entities.Node('Entity', attrib={'Type': 'Test'}) entities.Node('Value', parent=node) entities.Node('IconURL', value=value, parent=node) entity = entities.Entity.from_node(node=node) self.assertEqual(entity.icon_url, value)
def test_create__additional_fields(self): """Testing create instance with node additional fields.""" field_name = 'Test' node = entities.Node('Entity', attrib={'Type': 'Test'}) entities.Node('Value', parent=node) fields = entities.Node('AdditionalFields', parent=node) entities.Node('Field', 'V', attrib={'Name': field_name}, parent=fields) entity = entities.Entity.from_node(node=node) self.assertEqual(len(entity.fields), 1)
def test_create__with_node_labels(self): """Testing create instance with node labels.""" label_name = 'Test' node = entities.Node('Entity', 'Test', attrib={'Type': 'Test'}) entities.Node('Value', parent=node) labels = entities.Node('DisplayInformation', parent=node) entities.Node('Label', 'V', attrib={'Name': label_name}, parent=labels) entity = entities.Entity.from_node(node=node) self.assertEqual(len(entity.labels), 1)
def test_create(self): """Testing create instance.""" node = entities.Node('MaltegoMessage') request = entities.Node('MaltegoTransformRequestMessage', parent=node) entities_node = entities.Node('Entities', parent=request) entity = entities.Node( 'Entity', attrib={'Type': 'Test'}, parent=entities_node ) entities.Node('Value', value='Test', parent=entity) message = messages.TransformRequest.from_node(node=node) self.assertIsInstance(message, messages.TransformRequest)
def test_method_to_response(self): """Testing call not implemented method transform.""" node = entities.Node('MaltegoMessage') request = entities.Node('MaltegoTransformRequestMessage', parent=node) entities_node = entities.Node('Entities', parent=request) entity = entities.Node( 'Entity', attrib={'Type': 'Test'}, parent=entities_node ) entities.Node('Value', value='Test', parent=entity) message = messages.TransformRequest.from_node(node=node) transform = transforms.BaseTransform(message) with self.assertRaises(NotImplementedError): transform.to_response()
def test_create__with_value(self): """Testing create instance with value.""" name = 'Test' node = entities.Node(name, value=name) self.assertEqual( etree.tostring(node).decode('ascii'), '<{0}>{0}</{0}>'.format(name) )
def test_from_node(self): """Testing create instance from node.""" node = entities.Node('UIMessage', 'Test', attrib={'MessageType': 'ts'}) instance = entities.UIMessage.from_node(node) self.assertIsInstance(instance, entities.UIMessage) self.assertEqual(instance.value, 'Test') self.assertEqual(instance.message_type, 'ts')
def test_create(self): """Testing create instance.""" name = 'Test' node = entities.Node(name) self.assertEqual(node.tag, name) self.assertEqual( etree.tostring(node).decode('ascii'), '<{}/>'.format(name) )
def test_create__without_field_name(self): """Testing create instance with wrong node without field name.""" node = entities.Node('MaltegoMessage') request = entities.Node('MaltegoTransformRequestMessage', parent=node) entities_node = entities.Node('Entities', parent=request) entity = entities.Node( 'Entity', attrib={'Type': 'Test'}, parent=entities_node ) entities.Node('Value', value='Test', parent=entity) fields = entities.Node('TransformFields', parent=request) entities.Node('Field', parent=fields) with self.assertRaises(exceptions.MalformedMessageError): messages.TransformRequest.from_node(node=node)
def test_create__with_node_fields(self): """Testing create instance with node fields.""" node = entities.Node('MaltegoMessage') request = entities.Node('MaltegoTransformRequestMessage', parent=node) entities_node = entities.Node('Entities', parent=request) entity = entities.Node( 'Entity', attrib={'Type': 'Test'}, parent=entities_node ) entities.Node('Value', value='Test', parent=entity) fields = entities.Node('TransformFields', parent=request) entities.Node('Field', attrib={'Name': 'Test'}, parent=fields) message = messages.TransformRequest.from_node(node=node) self.assertIsInstance(message, messages.TransformRequest)
def test_from_node(self): """Testing create instance from node.""" node = entities.Node('MaltegoMessage') request = entities.Node('MaltegoTransformResponseMessage', parent=node) entities_node = entities.Node('Entities', parent=request) entity = entities.Node( 'Entity', 'Test', attrib={'Type': 'Test'}, parent=entities_node ) entities.Node('Value', value='Test', parent=entity) ui_messages_node = entities.Node('UIMessages', parent=request) entities.Node( 'UIMessage', 'Test', attrib={'MessageType': 'Test'}, parent=ui_messages_node ) instance = messages.TransformResponse.from_node(node) self.assertIsInstance(instance, messages.TransformResponse) self.assertEqual(instance.entities[0].value, 'Test') self.assertEqual(instance.ui_messages[0].value, 'Test')
def test_create_from_node(self): """Testing create instance with node.""" node = entities.Node('Label', 'Test', attrib={'Name': 'Test'}) label = entities.Label.from_node(node=node) self.assertIsInstance(label, entities.Label)
def test_create__with_node(self): """Testing create instance with node.""" node = entities.Node('Field', 'Test', attrib={'Name': 'Test'}) field = entities.Field.from_node(node=node) self.assertIsInstance(field, entities.Field)
def test_create__with_wrong_node(self): """Testing create instance with wrong node.""" node = entities.Node('Wrong') with self.assertRaises(exceptions.MalformedEntityError): entities.Field.from_node(node=node)
def test_create__wrong_node_type(self): """Testing create instance with wrong node without type.""" node = entities.Node('Entity') with self.assertRaises(exceptions.MalformedEntityError): entities.Entity.from_node(node=node)
def test_create__with_unicode_value(self): name = 'Test' value = u'\xf3' node = entities.Node(name, value=value) self.assertTrue(etree.tostring(node))
def test_create__wrong_node_value(self): """Testing create instance with wrong node without value.""" node = entities.Node('Entity', attrib={'Type': 'Test'}) with self.assertRaises(exceptions.MalformedEntityError): entities.Entity.from_node(node=node)