monkey/envs/monkey_zoo/blackbox/analyzers/performance_analyzer.py

from datetime import timedelta

from envs.monkey_zoo.blackbox.analyzers.analyzer import Analyzer
from envs.monkey_zoo.blackbox.analyzers.analyzer_log import AnalyzerLog
from envs.monkey_zoo.blackbox.island_client.monkey_island_client import MonkeyIslandClient

MAX_ALLOWED_SINGLE_PAGE_TIME = timedelta(seconds=1)
MAX_ALLOWED_TOTAL_TIME = timedelta(seconds=3)


class PerformanceAnalyzer(Analyzer):

    def __init__(self, island_client: MonkeyIslandClient):
        self.island_client = island_client
        self.log = AnalyzerLog(self.__class__.__name__)

    def analyze_test_results(self) -> bool:
        self.log.clear()
        total_time = timedelta()

        self.island_client.clear_caches()
        timings = self.island_client.time_all_report_pages()

        single_page_time_less_then_max = True

        for page, elapsed in timings.items():
            self.log.add_entry(f"page {page} took {str(elapsed)}")
            total_time += elapsed
            if elapsed > MAX_ALLOWED_SINGLE_PAGE_TIME:
                single_page_time_less_then_max = False

        total_time_less_then_max = total_time < MAX_ALLOWED_TOTAL_TIME

        self.log.add_entry(f"total time is {str(total_time)}")

        return total_time_less_then_max and single_page_time_less_then_max