Island: Move ransomware_report.py to ransomware/ransomware_report.py

This commit is contained in:
Mike Salvatore 2021-07-12 12:00:14 -04:00
parent 9e7e58658c
commit e4cd06d8c3
3 changed files with 5 additions and 3 deletions

View File

@ -2,10 +2,12 @@ import flask_restful
from flask import jsonify from flask import jsonify
from monkey_island.cc.resources.auth.auth import jwt_required from monkey_island.cc.resources.auth.auth import jwt_required
from monkey_island.cc.services.ransomware_report import get_propagation_stats from monkey_island.cc.services.ransomware import ransomware_report
class RansomwareReport(flask_restful.Resource): class RansomwareReport(flask_restful.Resource):
@jwt_required @jwt_required
def get(self): def get(self):
return jsonify({"report": None, "propagation_stats": get_propagation_stats()}) return jsonify(
{"report": None, "propagation_stats": ransomware_report.get_propagation_stats()}
)

View File

@ -1,6 +1,6 @@
import pytest import pytest
import monkey_island.cc.services.ransomware_report as ransomware_report from monkey_island.cc.services.ransomware import ransomware_report
from monkey_island.cc.services.reporting.report import ReportService from monkey_island.cc.services.reporting.report import ReportService
TEST_SCANNED_RESULTS = [{}, {}, {}, {}] TEST_SCANNED_RESULTS = [{}, {}, {}, {}]