From d6aa736688e1d0376a63cd4a587c118abf1ff839 Mon Sep 17 00:00:00 2001 From: narrieta Date: Fri, 27 Oct 2023 09:18:48 -0700 Subject: [PATCH] update test --- tests_e2e/tests/agent_cgroups/agent_cpu_quota.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests_e2e/tests/agent_cgroups/agent_cpu_quota.py b/tests_e2e/tests/agent_cgroups/agent_cpu_quota.py index 79f95fc547..36995a057c 100644 --- a/tests_e2e/tests/agent_cgroups/agent_cpu_quota.py +++ b/tests_e2e/tests/agent_cgroups/agent_cpu_quota.py @@ -1,14 +1,15 @@ from typing import List, Dict, Any -from tests_e2e.tests.lib.agent_test import AgentTest +from tests_e2e.tests.lib.agent_test import AgentVmTest +from tests_e2e.tests.lib.agent_test_context import AgentVmTestContext from tests_e2e.tests.lib.logging import log -class AgentCPUQuota(AgentTest): +class AgentCPUQuota(AgentVmTest): """ The test verify that the agent detects when it is throttled for using too much CPU, that it detects processes that do belong to the agent's cgroup, and that resource metrics are generated. """ - def __init__(self, context): + def __init__(self, context: AgentVmTestContext): super().__init__(context) self._ssh_client = self._context.create_ssh_client()