p41275903
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Minor fixes and improvements

This commit is contained in:
VakarisZ 2020-05-11 17:58:58 +03:00
parent 7663615a4f
commit 9b350b8bf5
5 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
envs/monkey_zoo/blackbox/tests/performance/telemetry_performance_test_workflow.py
View File

@ -13,9 +13,8 @@ class TelemetryPerformanceTestWorkflow(BasicTest):
def run(self):
try:
# TelemetryPerformanceTest(island_client=self.island_client).test_telemetry_performance()
TelemetryPerformanceTest(island_client=self.island_client).test_telemetry_performance()
performance_test = EndpointPerformanceTest(self.name, self.performance_config, self.island_client)
assert performance_test.run()
finally:
pass
# self.island_client.reset_env()
self.island_client.reset_env()

2
monkey/monkey_island/cc/resources/reporting/report.py
View File

@ -6,7 +6,6 @@ from flask import jsonify
from monkey_island.cc.auth import jwt_required
from monkey_island.cc.services.reporting.report import ReportService
from monkey_island.cc.services.reporting.zero_trust_service import ZeroTrustService
from monkey_island.cc.testing.profiler_decorator import profile
ZERO_TRUST_REPORT_TYPE = "zero_trust"
SECURITY_REPORT_TYPE = "security"
@ -22,7 +21,6 @@ __author__ = ["itay.mizeretz", "shay.nehmad"]
class Report(flask_restful.Resource):
@jwt_required()
@profile()
def get(self, report_type=SECURITY_REPORT_TYPE, report_data=None):
if report_type == SECURITY_REPORT_TYPE:
return ReportService.get_report()

2
monkey/monkey_island/cc/resources/zero_trust/finding_event.py
View File

@ -3,12 +3,10 @@ import json
from monkey_island.cc.auth import jwt_required
from monkey_island.cc.services.reporting.zero_trust_service import ZeroTrustService
from monkey_island.cc.testing.profiler_decorator import profile
class ZeroTrustFindingEvent(flask_restful.Resource):
@jwt_required()
@profile()
def get(self, finding_id: str):
return {'events_json': json.dumps(ZeroTrustService.get_events_by_finding(finding_id), default=str)}

5
monkey/monkey_island/cc/services/reporting/zero_trust_service.py
View File

@ -109,8 +109,7 @@ class ZeroTrustService(object):
@staticmethod
def get_all_findings():
pipeline = [{'$match': {}},
{'$addFields': {'oldest_events': {'$slice': ['$events', EVENT_FETCH_CNT]},
pipeline = [{'$addFields': {'oldest_events': {'$slice': ['$events', EVENT_FETCH_CNT]},
'latest_events': {'$slice': ['$events', -1*EVENT_FETCH_CNT]},
'event_count': {'$size': '$events'}}},
{'$unset': ['events']}]
@ -130,7 +129,7 @@ class ZeroTrustService(object):
elif overlap_count <= 0:
return []
else:
return events[ -overlap_count :]
return events[-1 * overlap_count:]
@staticmethod
def __get_enriched_finding(finding):

4
monkey/monkey_island/cc/ui/src/components/report-components/zerotrust/EventsModal.js
View File

@ -1,4 +1,4 @@
import React, {Component} from 'react';
import React from 'react';
import {Modal} from 'react-bootstrap';
import EventsTimeline from './EventsTimeline';
import * as PropTypes from 'prop-types';
@ -6,7 +6,7 @@ import saveJsonToFile from '../../utils/SaveJsonToFile';
import EventsModalButtons from './EventsModalButtons';
import AuthComponent from '../../AuthComponent';
import Pluralize from 'pluralize';
import SkippedEventsTimeline from "./SkippedEventsTimeline";
import SkippedEventsTimeline from './SkippedEventsTimeline';
const FINDING_EVENTS_URL = '/api/zero-trust/finding-event/';