from aci_tasks.logger import Logger
from cloud.aci_container import AciContainer
from cloud.aci_container_group import AciContainerGroup
from aci_tasks.cloud.aci_resource_group import AciResourceGroup
from aci_tasks.cloud.aci_key_vault import (AciKeyVaultManager,
AciKeyVaultSecretStore)
if __name__ == "__main__":
'''
Note: The tests do these things also. From root of project
just run "pytest"
'''
# Create Strong task container and exec command
strong_config = Config(TaskType.STRONG)
logger = Logger()
rg = AciResourceGroup(logger, strong_config)
rg.create()
strong_container_group = AciContainerGroup(logger, strong_config)
strong_container_group.create()
strong_container = AciContainer(logger, strong_config)
strong_container.execute_command('/strong_task/dog.py')
# Create week task container and start/stop to exec command
weak_config = Config(TaskType.WEAK)
weak_container_group = AciContainerGroup(logger, weak_config)
weak_container_group.create()
def resource_group():
return ResourceGroup(Logger(), Config(TaskType.WEAK))
def weak_container_group():
return AciContainerGroup(Logger(), Config(TaskType.WEAK))
def strong_container_group():
return AciContainerGroup(Logger(), Config(TaskType.STRONG))
def strong_container_instance():
return AciContainer(Logger(), Config(TaskType.STRONG))
def setup():
rg = AciResourceGroup(Logger(), Config(TaskType.STRONG))
rg.create()
yield
rg.delete()
def weak_container_instance():
return AciContainer(Logger(), Config(TaskType.WEAK))