ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-17 20:38:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Onefuzz.container.reset to the SDK (#198)

Browse Source
This commit is contained in:
Anshuman Goel
2020-10-23 14:31:12 -07:00
committed by GitHub
parent 6989cf9e34
commit 2fe56f94c1
2 changed files with 45 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
src/cli/onefuzz/api.py
View File

@ -274,6 +274,29 @@ class Containers(Endpoint):
self.logger.debug("list containers")
return self._req_model_list("GET", responses.ContainerInfoBase)
def reset(
self, container_types: Optional[List[enums.ContainerType]] = None
) -> None:
"""Reset containers unless specified.
[Caution]: It can lead to unexpected results.
The default conatiners are listed ContainerType.reset_defaults
"""
if not container_types:
container_types = enums.ContainerType.reset_defaults()
if not container_types:
raise Exception("Container type is None")
for container in self.list():
if (
container.metadata
and "container_type" in container.metadata
and enums.ContainerType(container.metadata["container_type"])
in container_types
):
self.logger.info("removing container: %s", container.name)
self.delete(container.name)
class Repro(Endpoint):
""" Interact with Reproduction VMs """
@ -1364,25 +1387,7 @@ class Onefuzz:
self.scalesets.shutdown(scaleset.scaleset_id, now=True)
if containers:
for container in self.containers.list():
if (
container.metadata
and "container_type" in container.metadata
and container.metadata["container_type"]
in [
"analysis",
"coverage",
"crashes",
"inputs",
"no_repro",
"readonly_inputs",
"reports",
"setup",
"unique_reports",
]
):
self.logger.info("removing container: %s", container.name)
self.containers.delete(container.name)
self.containers.reset()
def reset(
self,