from contextlib import contextmanager import os from unittest.mock import Mock from OpenSSL import crypto from cryptography.hazmat.backends import default_backend from flask import template_rendered from atst.utils.notification_sender import NotificationSender @contextmanager def captured_templates(app): recorded = [] def record(sender, template, context, **extra): recorded.append((template, context)) template_rendered.connect(record, app) try: yield recorded finally: template_rendered.disconnect(record, app) class FakeLogger: def __init__(self): self.messages = [] self.extras = [] def log(self, _lvl, msg, *args, **kwargs): self._log(_lvl, msg, *args, **kwargs) def info(self, msg, *args, **kwargs): self._log("info", msg, *args, **kwargs) def warning(self, msg, *args, **kwargs): self._log("warning", msg, *args, **kwargs) def error(self, msg, *args, **kwargs): self._log("error", msg, *args, **kwargs) def _log(self, _lvl, msg, *args, **kwargs): self.messages.append(msg) if "extra" in kwargs: self.extras.append(kwargs["extra"]) FakeNotificationSender = lambda: Mock(spec=NotificationSender) def parse_for_issuer_and_next_update(crl): with open(crl, "rb") as crl_file: parsed = crypto.load_crl(crypto.FILETYPE_ASN1, crl_file.read()) return parsed.get_issuer().der() def make_crl_list(x509_obj, x509_path): issuer = x509_obj.issuer.public_bytes(default_backend()) filename = os.path.basename(x509_path) return [(filename, issuer.hex())]