Gecko [ 167.86.127.34 ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
__init__.py	727 B	-rw-r--r--
challenges.py	19.8 KB	-rw-r--r--
challenges_test.py	21.17 KB	-rw-r--r--
client.py	46.26 KB	-rw-r--r--
client_test.py	56.84 KB	-rw-r--r--
crypto_util.py	10.99 KB	-rw-r--r--
crypto_util_test.py	9.98 KB	-rw-r--r--
errors.py	3.57 KB	-rw-r--r--
errors_test.py	1.48 KB	-rw-r--r--
fields.py	1.7 KB	-rw-r--r--
fields_test.py	2.03 KB	-rw-r--r--
jose_test.py	1.92 KB	-rw-r--r--
jws.py	2.09 KB	-rw-r--r--
jws_test.py	2.03 KB	-rw-r--r--
magic_typing.py	534 B	-rw-r--r--
magic_typing_test.py	1.42 KB	-rw-r--r--
messages.py	19.11 KB	-rw-r--r--
messages_test.py	16.25 KB	-rw-r--r--
standalone.py	11.09 KB	-rw-r--r--
standalone_test.py	10.54 KB	-rw-r--r--
test_util.py	3.12 KB	-rw-r--r--
util.py	166 B	-rw-r--r--
util_test.py	456 B	-rw-r--r--

Code Editor : test_util.py

"""Test utilities.

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

"""
import os
import sys
import pkg_resources
import unittest

from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives import serialization
import josepy as jose
from OpenSSL import crypto

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], crypto.FILETYPE_PEM, crypto.FILETYPE_ASN1)
    return crypto.load_certificate(loader, load_vector(*names))

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

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

def load_comparable_csr(*names):
    """Load ComparableX509 certificate request."""
    return jose.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 jose.ComparableRSAKey(loader(
        load_vector(*names), password=None, backend=default_backend()))

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

def skip_unless(condition, reason):  # pragma: no cover
    """Skip tests unless a condition holds.

This implements the basic functionality of unittest.skipUnless
    which is only available on Python 2.7+.

:param bool condition: If ``False``, the test will be skipped
    :param str reason: the reason for skipping the test

:rtype: callable
    :returns: decorator that hides tests unless condition is ``True``

"""
    if hasattr(unittest, "skipUnless"):
        return unittest.skipUnless(condition, reason)
    elif condition:
        return lambda cls: cls
    else:
        return lambda cls: None

def broken_on_windows(function):
    """Decorator to skip temporarily a broken test on Windows."""
    reason = 'Test is broken and ignored on windows but should be fixed.'
    return unittest.skipIf(
        sys.platform == 'win32'
        and os.environ.get('SKIP_BROKEN_TESTS_ON_WINDOWS', 'true') == 'true',
        reason)(function)

${this.title}

Code Editor : test_util.py