Class TestHMAC

tests/test_jws.py:101–139

class TestHMAC:
    def testHMAC256(self, payload):
        token = jws.sign(payload, "secret", algorithm=ALGORITHMS.HS256)
        assert jws.verify(token, "secret", ALGORITHMS.HS256) == payload

    def testHMAC384(self, payload):
        token = jws.sign(payload, "secret", algorithm=ALGORITHMS.HS384)
        assert jws.verify(token, "secret", ALGORITHMS.HS384) == payload

    def testHMAC512(self, payload):
        token = jws.sign(payload, "secret", algorithm=ALGORITHMS.HS512)
        assert jws.verify(token, "secret", ALGORITHMS.HS512) == payload

    def test_wrong_alg(self, payload):
        token = jws.sign(payload, "secret", algorithm=ALGORITHMS.HS256)
        with pytest.raises(JWSError):
            jws.verify(token, "secret", ALGORITHMS.HS384)

    def test_wrong_key(self, payload):
        token = jws.sign(payload, "secret", algorithm=ALGORITHMS.HS256)
        with pytest.raises(JWSError):
            jws.verify(token, "another", ALGORITHMS.HS256)

    def test_unsupported_alg(self, payload):
        with pytest.raises(JWSError):
            jws.sign(payload, "secret", algorithm="SOMETHING")

    def test_add_headers(self, payload):
        additional_headers = {"test": "header"}

        expected_headers = {
            "test": "header",
            "alg": "HS256",
            "typ": "JWT",
        }

        token = jws.sign(payload, "secret", headers=additional_headers)
        header, payload, signing_input, signature = jws._load(token)
        assert expected_headers == header


rsa_private_key = """-----BEGIN RSA PRIVATE KEY-----