def test_AitConfig():
config = cfg.AitConfig(data=YAML())
path = "bin/ait-orbits"
assert_AitConfig(config, path)
with TestFile(YAML(), "w+") as filename:
config = cfg.AitConfig(filename)
assert_AitConfig(config, path, filename)
config.reload()
assert_AitConfig(config, path, filename)
def test_successful_outbound_stream_creation(self, create_handler_mock,
server_init_mock,
broker_class_mock):
""" Tests that all types of outbound streams are successfully created """
# Testing creation of outbound stream with ZMQ input/output
server = Server()
server.broker = ait.core.server.broker.Broker()
config = {
'name': 'some_stream',
'handlers': [{
'name': 'some-handler'
}]
}
created_stream = server._create_outbound_stream(config)
assert type(created_stream) == ait.core.server.stream.ZMQStream
assert created_stream.name == 'some_stream'
assert type(created_stream.handlers) == list
# Testing creation of outbound stream with port output
config = cfg.AitConfig(config={'name': 'some_stream', 'output': 3333})
created_stream = server._create_outbound_stream(config)
assert type(created_stream) == ait.core.server.stream.PortOutputStream
assert created_stream.name == 'some_stream'
assert created_stream.out_port == 3333
assert created_stream.handlers == []
def test_successful_outbound_stream_creation(
self, create_handler_mock, server_stream_plugin_mock_mock,
broker_class_mock):
"""Tests that all types of outbound streams are successfully created"""
# Testing creation of outbound stream with ZMQ input/output
server = Server()
server.broker = ait.core.server.broker.Broker()
config = {
"name": "some_stream",
"handlers": [{
"name": "some-handler"
}]
}
created_stream = server._create_outbound_stream(config)
assert type(created_stream) == ait.core.server.stream.ZMQStream
assert created_stream.name == "some_stream"
assert type(created_stream.handlers) == list
# Testing creation of outbound stream with port output
config = cfg.AitConfig(config={"name": "some_stream", "output": 3333})
created_stream = server._create_outbound_stream(config)
assert type(created_stream) == ait.core.server.stream.PortOutputStream
assert created_stream.name == "some_stream"
assert created_stream.out_port == 3333
assert created_stream.handlers == []
def get_packet_and_defn():
first_stream = ait.config.get('gui.telemetry')[0]
stream = cfg.AitConfig(config=first_stream).get('stream')
name = stream.get('name', '<unnamed>')
pkt_defn = tlm.getDefaultDict().get(name, None)
pkt = tlm.Packet(pkt_defn)
return pkt_defn, pkt
def test_datapaths ():
"""
default:
ISS:
apiport: 9090
bticard: 0
desc: ISS PL/MDM Simulator
path: bin/ait-orbits
rtaddr: 15
"""
# check data paths work from YAML()
config = cfg.AitConfig(data=YAML())
assert len(config._datapaths) > 0
# check if data paths do not exist
config = cfg.AitConfig(data=test_datapaths.__doc__)
try:
paths = config._datapaths
assert False
except cfg.AitConfigMissing as e:
assert True
def test_addPathVariables():
config = cfg.AitConfig(data=YAML())
before = config._pathvars
before_len = len(before.keys())
pathvars = {'x': 'test-x', 'y': 'test-y'}
config.addPathVariables(pathvars)
after = config._pathvars
after_len = len(after.keys())
assert before_len < after_len
assert 'x' in after.keys()
assert 'y' in after.keys()
def test_add_path_variables():
config = cfg.AitConfig(data=YAML())
before = config._pathvars
before_len = len(before.keys())
pathvars = {"x": "test-x", "y": "test-y"}
config.add_path_variables(pathvars)
after = config._pathvars
after_len = len(after.keys())
assert before_len < after_len
assert "x" in after.keys()
assert "y" in after.keys()
def testGetDefaultDictWithExtension_setup_teardown():
cfg_yml = '''
default:
extensions:
ait.core.cmd.CmdDict: tests.ait.core.test_cmd.TestCmdDict
'''
saved_config = ait.config
ait.config = cfg.AitConfig(data=cfg_yml)
setattr(cmd, 'DefaultDict', None) # Reset DefaultDict being cached
importlib.reload(cmd) # Recreate createCmdDict method
yield
ait.config = saved_config
setattr(cmd, 'DefaultDict', None) # Reset DefaultDict being cached
importlib.reload(cmd) # Recreate createCmdDict method
importlib.reload(tlm) # Also reload tlm default schema
def tearDown(self):
ait.config = cfg.AitConfig()
if os.path.exists(self.test_yaml_file):
os.remove(self.test_yaml_file)
