REDROOM

Navigation

Breadcrumbs

Create

Preview: test_util.py Size: 2.57 KB

/lib/python3.6/site-packages/josepy/test_util.py

"""Test utilities.

.. warning:: This module is not part of the public API.

"""
import os

import OpenSSL
import pkg_resources
from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives import serialization

from josepy import ComparableRSAKey, ComparableX509
from josepy.util import ComparableECKey


def vector_path(*names):
    """Path to a test vector."""
    return pkg_resources.resource_filename(
        __name__, os.path.join('testdata', *names))


def load_vector(*names):
    """Load contents of a test vector."""
    # luckily, resource_string opens file in binary mode
    return pkg_resources.resource_string(
        __name__, os.path.join('testdata', *names))


def _guess_loader(filename, loader_pem, loader_der):
    _, ext = os.path.splitext(filename)
    if ext.lower() == '.pem':
        return loader_pem
    elif ext.lower() == '.der':
        return loader_der
    else:  # pragma: no cover
        raise ValueError("Loader could not be recognized based on extension")


def load_cert(*names):
    """Load certificate."""
    loader = _guess_loader(
        names[-1], OpenSSL.crypto.FILETYPE_PEM, OpenSSL.crypto.FILETYPE_ASN1)
    return OpenSSL.crypto.load_certificate(loader, load_vector(*names))


def load_comparable_cert(*names):
    """Load ComparableX509 cert."""
    return ComparableX509(load_cert(*names))


def load_csr(*names):
    """Load certificate request."""
    loader = _guess_loader(
        names[-1], OpenSSL.crypto.FILETYPE_PEM, OpenSSL.crypto.FILETYPE_ASN1)
    return OpenSSL.crypto.load_certificate_request(loader, load_vector(*names))


def load_comparable_csr(*names):
    """Load ComparableX509 certificate request."""
    return ComparableX509(load_csr(*names))


def load_rsa_private_key(*names):
    """Load RSA private key."""
    loader = _guess_loader(names[-1], serialization.load_pem_private_key,
                           serialization.load_der_private_key)
    return ComparableRSAKey(loader(
        load_vector(*names), password=None, backend=default_backend()))


def load_ec_private_key(*names):
    """Load EC private key."""
    loader = _guess_loader(names[-1], serialization.load_pem_private_key,
                           serialization.load_der_private_key)
    return ComparableECKey(loader(
        load_vector(*names), password=None, backend=default_backend()))


def load_pyopenssl_private_key(*names):
    """Load pyOpenSSL private key."""
    loader = _guess_loader(
        names[-1], OpenSSL.crypto.FILETYPE_PEM, OpenSSL.crypto.FILETYPE_ASN1)
    return OpenSSL.crypto.load_privatekey(loader, load_vector(*names))

Name	Size	Perms	Modified	Actions
testdata DIR	-	drwxr-xr-x	2025-04-13 08:21:05	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2025-04-13 08:21:05	Edit Download recursive recursive
b64.py	1.40 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
b64_test.py	2.20 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
errors.py	762 B	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
errors_test.py	463 B	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
interfaces.py	7.51 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
interfaces_test.py	3.51 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
json_util.py	15.35 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
json_util_test.py	13.83 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
jwa.py	8.17 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
jwa_test.py	8.80 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
jwk.py	13.12 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
jwk_test.py	12.69 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
jws.py	13.84 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
jws_test.py	8.32 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
magic_typing.py	529 B	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
magic_typing_test.py	1.05 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
test_util.py	2.57 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
util.py	8.37 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
util_test.py	8.15 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
__init__.py	1.66 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive

Directory Contents