wally/test_run_class.py

from typing import List, Callable, Any, Dict, Optional, Set
from concurrent.futures import ThreadPoolExecutor


from .timeseries import SensorDatastore
from .node_interfaces import NodeInfo, IRPCNode
from .start_vms import OSCreds, OSConnection
from .storage import Storage
from .config import Config
from .fuel_rest_api import Connection


class TestRun:
    """Test run information"""
    def __init__(self, config: Config, storage: Storage) -> None:
        # NodesInfo list
        self.nodes_info = []  # type: List[NodeInfo]

        # Nodes list
        self.nodes = []  # type: List[IRPCNode]

        self.build_meta = {}  # type: Dict[str,Any]
        self.clear_calls_stack = []  # type: List[Callable[['TestRun'], None]]

        # openstack credentials
        self.fuel_openstack_creds = None  # type: Optional[OSCreds]
        self.os_creds = None  # type: Optional[OSCreds]
        self.os_connection = None  # type: Optional[OSConnection]
        self.fuel_conn = None  # type: Optional[Connection]
        self.rpc_code = None  # type: bytes

        self.storage = storage
        self.config = config
        self.sensors_data = SensorDatastore()
        self.sensors_run_on = set()  # type: Set[str]

    def get_pool(self):
        return ThreadPoolExecutor(self.config.get('worker_pool_sz', 32))