2.0 is on the way
diff --git a/wally/test_run_class.py b/wally/test_run_class.py
index ecdfa4f..db41a46 100644
--- a/wally/test_run_class.py
+++ b/wally/test_run_class.py
@@ -1,4 +1,4 @@
-from typing import List, Callable, Any, Dict, Optional
+from typing import List, Callable, Any, Dict, Optional, Set
from concurrent.futures import ThreadPoolExecutor
@@ -7,11 +7,12 @@
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):
+ def __init__(self, config: Config, storage: Storage) -> None:
# NodesInfo list
self.nodes_info = [] # type: List[NodeInfo]
@@ -20,16 +21,18 @@
self.build_meta = {} # type: Dict[str,Any]
self.clear_calls_stack = [] # type: List[Callable[['TestRun'], None]]
- self.sensors_mon_q = 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))