def setUp(self) -> None:
auto_discover_drivers()
config: Config = Config(LABBY_CONFIG_YAML)
server: Server = Server(config)
def _handle(msg: EncodedData) -> None:
response_bytes = ServerRequest.handle_from_msgpack(server, msg)
self.req_mock.return_value.recv.return_value = response_bytes
self.req_patch = patch("labby.client.Req0")
self.req_mock = self.req_patch.start()
self.req_mock.return_value.send.side_effect = _handle
self.client = Client("foobar")
def run(cls, trigger: str, argv: Sequence[str]) -> int:
try:
command_klass = ALL_COMMANDS[trigger]
# pyre-ignore[16]: command_klass has no __orig_bases__ attribute
args_klass = get_args(command_klass.__orig_bases__[0])[0]
args = args_klass(prog=f"labby {trigger}").parse_args(argv)
auto_discover_drivers()
with open(args.config, "r") as config_file:
config = Config(config_file.read())
# pyre-ignore[45]: cannot instantiate Command with abstract method
command = command_klass(config)
return command.main(args)
except pynng.exceptions.Timeout:
# this had to be an inline import so the tests would use the
# WASABI_LOG_FRIENDLY env variable correctly ¯\_(ツ)_/¯
from wasabi import msg
msg.fail("Timeout")
msg.text("The labby server did not respond. Are you sure it is started?")
return 1
def setUp(self) -> None:
auto_discover_drivers()