def test_GetNamespaces_MalformedID_ThrowsException(self, mock_stix_scanner):
patcher = mock.patch('adapters.certuk_mod.publisher.publisher_edge_object.PublisherEdgeObject.__bases__', (mock.MagicMock,))
with patcher:
patcher.is_local = True
mock_stix_scanner.side_effect = PublisherEdgeObjectTests.stix_scanner_setup(
PublisherEdgeObjectTests.malformed_id)
edge_object_under_test = PublisherEdgeObject(None)
edge_object_under_test.id_ = None
edge_object_under_test.filters = None
self.assertRaises(Exception, edge_object_under_test.ns_dict)
def test_GetNamespaces_NoAlias_ReturnsNothing(self, mock_stix_scanner):
patcher = mock.patch('adapters.certuk_mod.publisher.publisher_edge_object.PublisherEdgeObject.__bases__', (mock.MagicMock,))
with patcher:
patcher.is_local = True
mock_stix_scanner.side_effect = PublisherEdgeObjectTests.stix_scanner_setup(
PublisherEdgeObjectTests.empty_alias_id)
edge_object_under_test = PublisherEdgeObject(None)
edge_object_under_test.id_ = None
edge_object_under_test.filters = None
namespaces = edge_object_under_test.ns_dict()
self.assertEqual(namespaces, {})
def test_GetNamespaces_NamespacesOK_ReturnsCorrectValues(self, mock_stix_scanner):
patcher = mock.patch('adapters.certuk_mod.publisher.publisher_edge_object.PublisherEdgeObject.__bases__', (mock.MagicMock,))
with patcher:
patcher.is_local = True
mock_stix_scanner.side_effect = PublisherEdgeObjectTests.stix_scanner_setup(PublisherEdgeObjectTests.ok_ids)
edge_object_under_test = PublisherEdgeObject(None)
edge_object_under_test.id_ = None
edge_object_under_test.filters = None
namespaces = edge_object_under_test.ns_dict()
self.assertEqual(namespaces, {
'http://www.namespace-1.com': 'Alias_1',
'http://www.namespace-2.com': 'Alias_2'
})
def visualiser_item_get(request, id_):
try:
root_edge_object = PublisherEdgeObject.load(id_)
if root_edge_object.ty == 'pkg':
package = root_edge_object.obj
validation_info = {}
else:
package = PackageGenerator.build_package(root_edge_object)
validation_info = PackageValidationInfo.validate(
package).validation_dict
return JsonResponse(
{
"root_id": id_,
"package": package.to_dict(),
"validation_info": validation_info,
"type_info": [{
"id_": id_,
"ty": root_edge_object.ty
}]
},
status=200)
except EdgeError as e:
if e.message == id_ + " not found":
return JsonResponse({'error': e.message}, status=404)
else:
return JsonResponse({'error': e.message}, status=500)
except Exception as e:
return JsonResponse({'error': e.message}, status=500)
def test_PublisherEdgeObject_RootHasGrandChildrenFromDifferentNamespace_NamespacePresent(self):
edge_object = PublisherEdgeObject.load('PurpleSecureSystems:indicator-02db75d2-77e2-4774-a219-293318051515')
namespaces = edge_object.ns_dict()
self.assertEqual(namespaces, {
'http://www.purplesecure.com': 'PurpleSecureSystems',
'http://www.fireeye.com': 'fireeye'
})
def visualiser_get(request, id_):
try:
root_edge_object = PublisherEdgeObject.load(id_)
graph = create_graph([(0, None, root_edge_object, REL_TYPE_EDGE)], [],
[], [], [], [], request)
return JsonResponse(graph, status=200)
except Exception as e:
return JsonResponse({'error': e.message}, status=500)
def generate_partial_review_data(request, id, revision):
root_edge_object = PublisherEdgeObject.load(id,
filters=request.user.filters(),
revision=revision,
include_revision_index=True)
package = PackageGenerator.build_package(root_edge_object)
validation_info = PackageValidationInfo.validate(package)
def user_loader(idref):
return EdgeObject.load(idref, request.user.filters())
edges = EdgeGenerator.gather_edges(root_edge_object.edges,
load_by_id=user_loader)
req_user = _get_request_username(request)
if root_edge_object.created_by_username != req_user:
validation_info.validation_dict.update({
id: {
"created_by": {
"status":
ValidationStatus.WARN,
"message":
"This object was created by %s not %s" %
(root_edge_object.created_by_username, req_user)
}
}
})
if any(item['is_external'] for item in edges):
validation_info.validation_dict.update({
id: {
"external_references": {
"status":
ValidationStatus.ERROR,
"message":
"This object contains External References, clone "
"object and remove missing references before"
" publishing"
}
}
})
# add root object to edges for javascript to construct object
edges.append({
'ty': root_edge_object.ty,
'id_': root_edge_object.id_,
'is_external': False
})
return {
'root_edge_object': root_edge_object,
'package': package,
"trust_groups": root_edge_object.tg,
"validation_info": validation_info,
"edges": edges,
"revision": root_edge_object.version
}
def ajax_load_object(request, id_):
root_edge_object = PublisherEdgeObject.load(id_)
package = PackageGenerator.build_package(root_edge_object)
return JsonResponse(
{
"root_id": id_,
"package": package.to_dict(),
"type_info": [{
"id_": id_,
"ty": root_edge_object.ty
}]
},
status=200)
def visualiser_get_extended(request):
json_data = json.loads(request.body)
root_id = json_data['id']
bl_ids = json_data['id_bls']
match_ids = json_data['id_matches']
hide_edge_ids = json_data['hide_edge_ids']
show_edge_ids = json_data['show_edge_ids']
hidden_ids = json_data['hidden_ids']
try:
root_edge_object = PublisherEdgeObject.load(root_id)
graph = create_graph([(0, None, root_edge_object, REL_TYPE_EDGE)],
bl_ids, match_ids, hide_edge_ids, show_edge_ids,
hidden_ids, request)
return JsonResponse(graph, status=200)
except Exception as e:
return JsonResponse({'error': e.message}, status=500)
def test_STIXScanner_AnyEdgeObject_CalledWithCorrectArguments(self, mock_stix_scanner):
patcher = mock.patch('adapters.certuk_mod.publisher.publisher_edge_object.PublisherEdgeObject.__bases__', (mock.MagicMock,))
with patcher:
patcher.is_local = True
mock_stix_scanner.side_effect = PublisherEdgeObjectTests.stix_scanner_setup(
PublisherEdgeObjectTests.ok_ids)
mock_root_id = PublisherEdgeObjectTests.ok_ids.keys()[0]
mock_filter = 'Dummy filter'
edge_object_under_test = PublisherEdgeObject(None)
edge_object_under_test.id_ = mock_root_id
edge_object_under_test.filters = mock_filter
edge_object_under_test.ns_dict()
mock_stix_scanner.assert_called_with({'_id': mock_root_id}, mock_filter)
def ajax_publish(request, data):
success = True
error_message = ""
root_id = None
try:
root_id = data['root_id']
edge_object = PublisherEdgeObject.load_and_parse(root_id)
package = PackageGenerator.build_package(edge_object)
namespace_info = edge_object.ns_dict()
Publisher.push_package(package, namespace_info)
# Narrow down which exceptions we catch...?
except Exception, e:
message = ''
taxii_response = get_exception_stack_variable('tr')
if taxii_response:
message = '\nTAXII Staus Message:\n' + json.dumps(
taxii_response.to_dict())
log_error(e, 'Publisher', message)
success = False
error_message = e.message
def test_no_identity_key_on_list_ids(self):
returned_doc = PublisherEdgeObject.strip_uuids_from_identities(
self.doc_no_identity_key)
self.assertDictEqual(returned_doc, self.doc_no_identity_key)
def test_remove_uuid_name_fully_populated(self):
returned_doc = PublisherEdgeObject.strip_uuids_from_identities(
self.doc_fully_populated)
self.assertDictEqual(returned_doc, self.doc_fully_populated_returned)
def get_duplicates(request, id_):
root_edge_object = PublisherEdgeObject.load(id_,
filters=request.user.filters())
duplicates = DuplicateFinder.find_duplicates(root_edge_object)
return JsonResponse({"duplicates": duplicates})
def test_PublisherEdgeObject_CreatedByUserNamePresent(self):
edge_object = PublisherEdgeObject.load('PurpleSecureSystems:indicator-02db75d2-77e2-4774-a219-293318051515')
username = edge_object.created_by_username
self.assertEqual(username, 'admin')
