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")