def test_excp(self, log_excp, exit):
main = create_mock()
main.side_effect = KeyError
# Call
main_wrapper(main)
# Tests
ntools.ok_(log_excp.called)
ntools.ok_(exit.called)
:param int exp_time: expiration time of the drkey secret
:return: the according drkey secret
:rtype: DRKeySecretValue
"""
sv = self.drkey_secrets.get(exp_time)
if not sv:
sv = DRKeySecretValue(
kdf(self.config.master_as_key, b"Derive DRKey Key"), exp_time)
self.drkey_secrets[sv.exp_time] = sv
return sv
def _init_metrics(self):
super()._init_metrics()
for type_ in ("trc", "cc", "drkey"):
REQS_TOTAL.labels(**self._labels, type=type_).inc(0)
IS_MASTER.labels(**self._labels).set(0)
def run(self):
"""
Run an instance of the Cert Server.
"""
threading.Thread(target=thread_safety_net,
args=(self.worker, ),
name="CS.worker",
daemon=True).start()
super().run()
if __name__ == "__main__":
main_wrapper(main_default, CertServer)
NAME = "SIBRA"
def _check_result(self, client, server):
if client.success == "skip":
return True
return super()._check_result(client, server)
def _create_server(self, data, finished, addr):
return SibraServer(self._run_sciond(addr), data, finished, addr)
def _create_client(self, data, finished, src, dst, port):
return SibraClient(self._run_sciond(src), data, finished, src, dst,
port)
def main():
args, srcs, dsts = setup_main("sibra_ext_test")
if args.wait:
logging.info("Waiting %ss", args.wait)
time.sleep(args.wait)
SIBRATest(args.client,
args.server,
srcs,
dsts,
local=False,
max_runs=args.runs).run()
if __name__ == "__main__":
main_wrapper(main)
def test_basic(self):
main = create_mock()
# Call
main_wrapper(main, "arg1", arg2="arg2")
# Tests
main.assert_called_once_with("arg1", arg2="arg2")
