3
}:aF! @ s d Z ddlZddlZddlmZ ddlZddlmZmZmZm Z m
Z
e
j dZ e j
je
jdZG dd dejZG d d
d
ejZG d d d ejZG d
d dejZG dd dejZedkrej dS )zTests for josepy.jws. N)mock)errors json_utiljwajwk test_utilzcert.pemzrsa512_key.pemc @ s e Zd ZdZdd Zdd ZdS )
MediaTypeTestz Tests for josepy.jws.MediaType.c C sF ddl m} | jd|jd | jd|jd | jtj|jd d S )Nr ) MediaTypezapplication/appappzapp;foo)
josepy.jwsr assertEqualdecode assertRaisesr DeserializationError)selfr r /usr/lib/python3.6/jws_test.py test_decode s
zMediaTypeTest.test_decodec C s4 ddl m} | jd|jd | jd|jd d S )Nr )r r
zapplication/appzapplication/app;foo)r r r encode)r r r r r test_encode s zMediaTypeTest.test_encodeN)__name__
__module__ __qualname____doc__r r r r r r r s r c @ sP e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dS )
HeaderTestz Tests for josepy.jws.Header.c C s< ddl m} |dd| _|dd| _|d d| _| | _d S )
Nr )Headerfoo)r barab)crit)r r )r r header1header2r empty)r r r r r setUp" s
zHeaderTest.setUpc C s, ddl m} | j|ddd| j| j d S )Nr )r r r r )r r )r r )r r r r! r )r r r r r test_add_non_empty) s z HeaderTest.test_add_non_emptyc C s0 | j | j| j| j | j | j| j| j d S )N)r r! r# )r r r r test_add_empty. s zHeaderTest.test_add_emptyc C s | j t| jj| j d S )N)r TypeErrorr! __add__r" )r r r r test_add_overlapping_error2 s z%HeaderTest.test_add_overlapping_errorc C s | j t| jjd d S )NZxxx)r r' r! r( )r r r r test_add_wrong_type_error5 s z$HeaderTest.test_add_wrong_type_errorc C s* ddl m} | jtj|jdddgi d S )Nr )r r r r )r r r r r from_json)r r r r r test_crit_decode_always_errors8 s z)HeaderTest.test_crit_decode_always_errorsc C s ddl m} |ttfd}|j }tjjtjjtj}t j
|}| j |d||gi | j ||j | t j
d| |d d< | j
tj|j | d S )Nr )r )x5cr- s xxx)r r CERTto_partial_jsonOpenSSLZcryptoZdump_certificateZ
FILETYPE_ASN1wrappedbase64Z b64encoder r+ r r r )r r headerZjobjZ cert_asn1Zcert_b64r r r test_x5c_decoding= s
z HeaderTest.test_x5c_decodingc C s: | j d| jj | j d| jj | jtj| jj d S )Nr r )r r! Zfind_keyr" r r Errorr )r r r r
test_find_keyI s zHeaderTest.test_find_keyN) r r r r r$ r% r& r) r* r, r4 r6 r r r r r s r c @ s e Zd ZdZdd Zdd ZdS )
SignatureTestz Tests for josepy.jws.Signature.c C sH ddl m} ddl m} | j|d|tjdd|jddd id d S )
Nr )r ) Signatures foo)alg) signaturer3 ZZm9vr9 RS256)r r r8 r r r; r+ )r r r8 r r r test_from_jsonR s z SignatureTest.test_from_jsonc C s& ddl m} | jtj|jddi d S )Nr )r8 r: r )r r8 r r r r+ )r r8 r r r test_from_json_no_alg_errorZ s z)SignatureTest.test_from_json_no_alg_errorN)r r r r r< r= r r r r r7 O s r7 c @ sp e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dd Z
dd Zdd ZdS )JWSTestzTests for josepy.jws.JWS.c C sv t | _| jj | _ddlm} |jd| jtjd| _ |jd| jtjt
ddgd| _ |jd| jtjt
dgd| _ d S )Nr )JWSs foo)payloadkeyr9 r r9 )r@ rA r9 Zprotect)
KEYZprivkeyZ
public_keypubkeyr r? signr r; unprotected frozenset protectedmixed)r r? r r r r$ c s
z
JWSTest.setUpc C sF | j | jjjj| j | j | jjjj| j | j | jjjj| j d S )N)r rE r: combinedr rC rG rH )r r r r test_pubkey_jwkq s zJWSTest.test_pubkey_jwkc C s | j | jj d d S )NT)assertIsrE verify)r r r r test_sign_unprotectedv s z JWSTest.test_sign_unprotectedc C s | j | jj d d S )NT)rK rG rL )r r r r test_sign_protectedy s zJWSTest.test_sign_protectedc C s | j | jj d d S )NT)rK rH rL )r r r r test_sign_mixed| s zJWSTest.test_sign_mixedc C sX | j j }| jd| ddlm} |j|}| j| j | | jdht|jj j
d S )Nsr eyJhbGciOiAiUlMyNTYifQ.Zm9v.OHdxFVj73l5LpxbFp1AmYX4yJM0Pyb_893n1zQjpim_eLS5J1F61lkvrCrCDErTEJnBGOGesJ72M7b6Ve1cAJAr )r? r9 ) rH Z
to_compactr r r? from_compactZassertNotEqualsetr: rI Z not_omitted)r Zcompactr? rH r r r test_compact_lost_unprotected s
z%JWSTest.test_compact_lost_unprotectedc C s" ddl m} | jtj|jd d S )Nr )r? .)r r? r r r rP )r r? r r r $test_from_compact_missing_components s z,JWSTest.test_from_compact_missing_componentsc C s| | j jdd}| jjdd}| jd| | jd| |d j |d< ddlm} | j|j|| j | j|j|| j d S )NT)flatrG r3 r )r? ) rG r/ rE Z assertNotInto_jsonr r? r r+ )r Zprotected_jobjZunprotected_jobjr? r r r test_json_omitempty s zJWSTest.test_json_omitemptyc C s t j| jjjt jd| jjjt j| jjjjdd}|j }|d j |d< | j | jj
dd| ddl m } | j | j|j
| d S ) Ns foozutf-8)r: r@ r3 rG r3 T)rU r )r? )r encode_b64joserH r: r3 rG r copyrV r r/ r r? r+ )r jobj_to jobj_fromr? r r r test_json_flat s
zJWSTest.test_json_flatc C sn | j jftjdd}|j }|d d j g|d< | j| j jdd| ddlm } | j| j |j
| d S )Ns foo)
signaturesr@ r] r F)rU )r? ) rH r: r rX rY rV r r/ r r? r+ )r rZ r[ r? r r r test_json_not_flat s zJWSTest.test_json_not_flatc C s( ddl m} | jtj|jf dd d S )Nr )r? r )r] r: )r r? r r r r+ )r r? r r r test_from_json_mixed_flat s z!JWSTest.test_from_json_mixed_flatc C s$ ddl m} t|j| jj d S )Nr )r? )r r? hashr+ rH rV )r r? r r r test_from_json_hashable s z JWSTest.test_from_json_hashableN)r r r r r$ rJ rM rN rO rR rT rW r\ r^ r_ ra r r r r r> ` s
r> c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
CLITestc C s t jd| _d S )Nzrsa512_key.pem)r Z vector_pathkey_path)r r r r r$ s z
CLITest.setUpc C sX ddl m} tjd8}d|j_tjd | jd|jdg W d Q R X W d Q R X d S ) Nr )CLIz sys.stdinz&{"payload": "foo", "signature": "xxx"}z
sys.stdout rL )r rd r patchread return_valuer run)r rd sinr r r test_unverified s
zCLITest.test_unverifiedc C s ddl m} tjdj}d|j_tjdL}|jdd| jdd d
d g |jj d d d |j_| j
d|jd
g W d Q R X W d Q R X d S )Nr )rd z sys.stdinr z
sys.stdoutrD z-kz-ar; z-pr re rL ) r rd r rg rh ri rj rc write
mock_callsr )r rd rk soutr r r test_json s
zCLITest.test_jsonc C s ddl m} tjdp}d|j_tjdR}|jddd| jg |jj d d d |j_| j
d|jdd
d d d| jg W d Q R X W d Q R X d S )
Nr )rd z sys.stdinr z
sys.stdoutz --compactrD z-kre rL z--ktyZRSA) r rd r rg rh ri rj rc rm rn r )r rd rk ro r r r test_compact s
zCLITest.test_compactN)r r r r$ rl rp rq r r r r rb s rb __main__)r r2 Zunittestr r0 Zjosepyr r r r r Zload_comparable_certr. ZJWKRSAloadZ load_vectorrB ZTestCaser r r7 r> rb r mainr r r r s
0c$