increase the polling period in functional tests (#1682)

* increase the polling period

* update timeout to 30s

* format
This commit is contained in:
Cheick Keita
2022-03-01 18:10:02 -08:00
committed by GitHub
parent d260689233
commit 50bac07d19

View File

@ -34,8 +34,7 @@ from onefuzz.backend import ContainerWrapper, wait
from onefuzz.cli import execute_api
from onefuzztypes.enums import OS, ContainerType, TaskState, VmState
from onefuzztypes.models import Job, Pool, Repro, Scaleset, Task
from onefuzztypes.primitives import (Container, Directory, File, PoolName,
Region)
from onefuzztypes.primitives import Container, Directory, File, PoolName, Region
from pydantic import BaseModel, Field
LINUX_POOL = "linux-test"
@ -260,13 +259,16 @@ def retry(
class TestOnefuzz:
def __init__(self, onefuzz: Onefuzz, logger: logging.Logger, test_id: UUID) -> None:
def __init__(
self, onefuzz: Onefuzz, logger: logging.Logger, test_id: UUID, polling_period=30
) -> None:
self.of = onefuzz
self.logger = logger
self.test_id = test_id
self.project = f"test-{self.test_id}"
self.start_log_marker = f"integration-test-injection-error-start-{self.test_id}"
self.stop_log_marker = f"integration-test-injection-error-stop-{self.test_id}"
self.polling_period = polling_period
def setup(
self,
@ -582,7 +584,7 @@ class TestOnefuzz:
return (not bool(jobs), msg, self.success)
if poll:
return wait(check_jobs_impl)
return wait(check_jobs_impl, frequency=self.polling_period)
else:
_, msg, result = check_jobs_impl()
self.logger.info(msg)
@ -724,7 +726,7 @@ class TestOnefuzz:
msg = "waiting on %d repros" % len(repros)
return (not bool(repros), msg, self.success)
return wait(check_repro_impl)
return wait(check_repro_impl, frequency=self.polling_period)
def get_jobs(self) -> List[Job]:
jobs = self.of.jobs.list(job_state=None)
@ -837,7 +839,7 @@ class TestOnefuzz:
# order.
self.inject_log(self.stop_log_marker)
wait(self.check_log_end_marker, frequency=5.0)
wait(self.check_log_end_marker, frequency=self.polling_period)
self.logger.info("application insights log flushed")
logs = self.of.debug.logs.keyword("error", limit=100000, timespan="PT3H")