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)
