Elgato_dark/bunkerweb
1
0
Fork 0
mirror of https://github.com/bunkerity/bunkerweb synced 2026-05-24 09:28:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor import statements in main.py of core backup tests

Browse source
This commit is contained in:
Théophile Diot 2024-04-04 21:53:56 +01:00
parent d7e6399d81
commit fa8a49f9f9
No known key found for this signature in database
GPG key ID: 248FEA4BAE400D06
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/core/backup/main.py
View file

@ -3,10 +3,7 @@ from os import getenv
from subprocess import PIPE, Popen
from time import sleep
from traceback import format_exc
from typing import List, Optional, Tuple, Union
from docker import DockerClient
from docker.models.containers import Container
from typing import List, Tuple, Union
try:
use_backup = getenv("USE_BACKUP", "no") == "yes"
@ -16,8 +13,11 @@ try:
if backup_rotation < 1:
backup_rotation = 1
scheduler_instance: Optional[Container] = None
scheduler_instance = None
if getenv("TEST_TYPE", "docker") == "docker":
from docker import DockerClient
from docker.models.containers import Container
docker_client = DockerClient(base_url=getenv("DOCKER_HOST", "unix:///var/run/docker.sock"))
scheduler_instances = docker_client.containers.list(filters={"label": "bunkerweb.SCHEDULER"})
@ -27,6 +27,7 @@ try:
exit(1)
scheduler_instance = scheduler_instances[0] # type: ignore
assert isinstance(scheduler_instance, Container), "Scheduler instance is not a container"
def exec_command(command: Union[str, List[str]]) -> Tuple[int, str, str]:
if scheduler_instance: