mavsdk_tests: only env vars that are set

Otherwise this raises a KeyError.
This commit is contained in:
Julian Oes
2020-03-18 13:31:18 +01:00
committed by Nuno Marques
parent 5ea40c57c6
commit ade39481ad
+8 -4
View File
@@ -183,15 +183,19 @@ class GzserverRunner(Runner):
"GAZEBO_MODEL_PATH": "GAZEBO_MODEL_PATH":
workspace_dir + "/Tools/sitl_gazebo/models", workspace_dir + "/Tools/sitl_gazebo/models",
"PX4_SIM_SPEED_FACTOR": str(speed_factor), "PX4_SIM_SPEED_FACTOR": str(speed_factor),
"DISPLAY": os.environ['DISPLAY'], "DISPLAY": os.environ['DISPLAY']}
"PX4_HOME_LAT": os.environ['PX4_HOME_LAT'], self.add_to_env_if_set("PX4_HOME_LAT")
"PX4_HOME_LON": os.environ['PX4_HOME_LON'], self.add_to_env_if_set("PX4_HOME_LON")
"PX4_HOME_ALT": os.environ['PX4_HOME_ALT']} self.add_to_env_if_set("PX4_HOME_ALT")
self.cmd = "gzserver" self.cmd = "gzserver"
self.args = ["--verbose", self.args = ["--verbose",
workspace_dir + "/Tools/sitl_gazebo/worlds/" + workspace_dir + "/Tools/sitl_gazebo/worlds/" +
self.model + ".world"] self.model + ".world"]
def add_to_env_if_set(self, var: str) -> None:
if var in os.environ:
self.env[var] = os.environ[var]
class GzclientRunner(Runner): class GzclientRunner(Runner):
def __init__(self, def __init__(self,