def test_dt_mult_handling():
"""
Tests that we properly track the dtmult value. BlueSky resets this value to 1 after
a reset. We can't query for the current value however, so have to track it ourselves
"""
bs_client_mock = mock.Mock()
bs_sim_controls = BlueSkySimulatorControls(bs_client_mock)
dt_mult = 4
bs_client_mock.send_stack_cmd.return_value = [f"Speed set to {dt_mult}"]
err = bs_sim_controls.set_speed(dt_mult)
assert not err
bs_client_mock.sim_info_stream_data = _TEST_SIMINFO
with mock.patch(
"bluebird.sim_client.bluesky.bluesky_simulator_controls.in_agent_mode"
) as in_agent_mode_mock:
in_agent_mode_mock.return_value = True
props = bs_sim_controls.properties
assert props.speed == dt_mult
bs_client_mock.reset_sim.return_value = None
err = bs_sim_controls.reset()
assert not err
props = bs_sim_controls.properties
assert props.speed == 1
def test_step():
"""Test step"""
bs_client_mock = mock.Mock()
bs_sim_controls = BlueSkySimulatorControls(bs_client_mock)
bs_client_mock.step.return_value = "Error"
err = bs_sim_controls.step()
assert err == "Error"
bs_client_mock.reset_mock()
bs_client_mock.step.return_value = None
err = bs_sim_controls.step()
assert not err
bs_client_mock.step.assert_called_once()
def test_stop():
"""Test stop"""
bs_client_mock = mock.Mock()
bs_sim_controls = BlueSkySimulatorControls(bs_client_mock)
bs_client_mock.send_stack_cmd.return_value = "Error"
err = bs_sim_controls.stop()
assert err == "Error"
bs_client_mock.reset_mock()
bs_client_mock.send_stack_cmd.return_value = None
err = bs_sim_controls.stop()
assert not err
bs_client_mock.send_stack_cmd.assert_called_once_with("STOP")
def test_set_seed():
"""Test set_seed"""
bs_client_mock = mock.Mock()
bs_sim_controls = BlueSkySimulatorControls(bs_client_mock)
bs_client_mock.send_stack_cmd.return_value = "Error"
err = bs_sim_controls.set_seed(0)
assert err == 'No confirmation received from BlueSky. Received: "Error"'
bs_client_mock.reset_mock()
seed = 4444
bs_client_mock.send_stack_cmd.return_value = [f"Seed set to {seed}"]
err = bs_sim_controls.set_seed(seed)
assert not err
bs_client_mock.send_stack_cmd.assert_called_once_with(
f"SEED {seed}", response_expected=True)
def test_set_speed():
"""Test set_speed"""
bs_client_mock = mock.Mock()
bs_sim_controls = BlueSkySimulatorControls(bs_client_mock)
bs_client_mock.send_stack_cmd.return_value = "Error"
err = bs_sim_controls.set_speed(1.23)
assert err == 'No confirmation received from BlueSky. Received: "Error"'
bs_client_mock.reset_mock()
dt_mult = 10.1
bs_client_mock.send_stack_cmd.return_value = [f"Speed set to {dt_mult}"]
err = bs_sim_controls.set_speed(dt_mult)
assert not err
bs_client_mock.send_stack_cmd.assert_called_once_with(
f"DTMULT {dt_mult}", response_expected=True)
def test_abstract_class_implemented():
"""Tests that BlueSkyAircraftControls implements the abstract base class"""
# Test basic instantiation
BlueSkySimulatorControls(mock.Mock())
# Test ABC exactly implemented
assert AbstractSimulatorControls.__abstractmethods__ == {
x for x in dir(BlueSkySimulatorControls) if not x.startswith("_")
}
def test_load_scenario():
"""Test load_scenario"""
bs_client_mock = mock.Mock()
bs_sim_controls = BlueSkySimulatorControls(bs_client_mock)
# Test with no sector set
existing_scenario = props.Scenario("TEST", [])
with pytest.raises(AssertionError):
bs_sim_controls.load_scenario(existing_scenario)
# Test error from scenario parsing
assert not bs_sim_controls.load_sector(_TEST_SECTOR)
existing_scenario = props.Scenario("TEST", ["a", "b", "c"])
err = bs_sim_controls.load_scenario(existing_scenario)
assert err.startswith("Could not parse a BlueSky scenario")
# Test upload_new_scenario - error response
bs_client_mock.upload_new_scenario.return_value = "Error 1"
err = bs_sim_controls.load_scenario(_TEST_SCENARIO)
assert err == "Error 1"
# Test load_scenario - error response
bs_client_mock.upload_new_scenario.return_value = None
bs_client_mock.load_scenario.return_value = "Error 2"
err = bs_sim_controls.load_scenario(_TEST_SCENARIO)
assert err == "Error 2"
# Test load_scenario - valid response
bs_client_mock.load_scenario.return_value = None
err = bs_sim_controls.load_scenario(_TEST_SCENARIO)
assert not err
def test_properties():
"""Test the properties property"""
bs_client_mock = mock.Mock()
bs_sim_controls = BlueSkySimulatorControls(bs_client_mock)
# Test error when no data received
bs_client_mock.sim_info_stream_data = None
bs_sim_controls.properties == "aaa"
# Test properties when we have data - Agent mode
bs_client_mock.sim_info_stream_data = _TEST_SIMINFO
Settings.SIM_MODE = props.SimMode.Agent
assert bs_sim_controls.properties == props.SimProperties(
dt=0.05,
scenario_name=None,
scenario_time=1234,
sector_name=None,
seed=None,
speed=1.0,
state=props.SimState.RUN,
utc_datetime=datetime(2020, 1, 2, 12, 34, 56),
)
# Test properties when we have data - Sandbox mode
bs_client_mock.sim_info_stream_data = _TEST_SIMINFO
Settings.SIM_MODE = props.SimMode.Sandbox
assert bs_sim_controls.properties == props.SimProperties(
dt=0.05,
scenario_name=None,
scenario_time=1234,
sector_name=None,
seed=None,
speed=1.12,
state=props.SimState.RUN,
utc_datetime=datetime(2020, 1, 2, 12, 34, 56),
)
