3
�}:a�2 � @ s� d Z ddlZddlZddlmZmZmZmZ ejd�Z ej
d�Z ej
d�Z ej
d�Zej
d�Zej
d �ZG d
d � d ej�ZG d d
� d
�ZG dd� deje�ZG dd� deje�ZG dd� deje�Zedkr�ej� dS )zTests for josepy.jwk.� N)�errors� json_util� test_util�utilzdsa512_key.pemzrsa256_key.pemzrsa512_key.pemzec_p256_key.pemzec_p384_key.pemzec_p521_key.pemc @ s e Zd ZdZdd� Zdd� ZdS )�JWKTestzTests for josepy.jwk.JWK.c C s" ddl m} | jtj|jt� d S )Nr )�JWK)�
josepy.jwkr � assertRaisesr �Error�load�DSA_PEM)�selfr � r � /usr/lib/python3.6/jwk_test.py� test_load s zJWKTest.test_loadc C s" ddl m} | jtj|jt� d S )Nr )�JWKRSA)r r r r r
r r )r
r r r r � test_load_subclass_wrong_type s z%JWKTest.test_load_subclass_wrong_typeN)�__name__�
__module__� __qualname__�__doc__r r r r r r r s r c @ s* e Zd ZU dZeZedd� Zdd� ZdS )�JWKTestBaseMixinz"Mixin test for JWK subclass tests.c C s | j | j| jj� � d S )N)� assertEqual�
thumbprint�jwk)r
r r r �test_thumbprint_private s z(JWKTestBaseMixin.test_thumbprint_privatec C s | j | j| jj� j� � d S )N)r r r �
public_key)r
r r r �test_thumbprint_public# s z'JWKTestBaseMixin.test_thumbprint_publicN) r r r r �NotImplementedr �bytesr r r r r r r s
r c @ sD e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dS )�
JWKOctTestz Tests for josepy.jwk.JWKOct.s �羨`��p�}��>�I�͒'�o�1�(��i c C s. ddl m} |dd�| _dtjd�d�| _d S )Nr )�JWKOcts foo)�key�oct)�kty�k)r r! r r Zencode_b64jose�jobj)r
r! r r r �setUp. s zJWKOctTest.setUpc C s | j | jj� | j� d S )N)r r �to_partial_jsonr& )r
r r r �test_to_partial_json3 s z JWKOctTest.test_to_partial_jsonc C s& ddl m} | j| j|j| j�� d S )Nr )r! )r r! r r � from_jsonr& )r
r! r r r �test_from_json6 s zJWKOctTest.test_from_jsonc C s ddl m} t|j| j�� d S )Nr )r! )r r! �hashr* r& )r
r! r r r �test_from_json_hashable: s z"JWKOctTest.test_from_json_hashablec C s$ ddl m} | j| j|jd�� d S )Nr )r! s foo)r r! r r r )r
r! r r r r > s zJWKOctTest.test_loadc C s | j | jj� | j� d S )N)ZassertIsr r )r
r r r �test_public_keyB s zJWKOctTest.test_public_keyN) r r r r r r' r) r+ r- r r. r r r r r ' s r c @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd S ) �
JWKRSATestz Tests for josepy.jwk.JWKRSA.s �K�#3�ʘ�ˀ�< �H�*l�$�&|O�� �Yc C s� ddl m} |tj� d�| _dddd�| _|tj� jd�| _|tj� d�| _ dddd�| _
|td�| _ | jj � | _
d | j
d
< | jj � | _| jjd d d d
ddd�� | j | _d S )Nr )r )r" �RSA�AQABz+m2Fylv-Uz7trgTW8EBHP3FQSMeZs2GNQ6VRo1sIVJEk)r$ �e�nzVrHVztFHtH92ucFJD_N_HW9AsdRsUuHUBBBDlHwNlRd3fp580rv2-6QWE30cWgdmJS86ObRz6lUTor4R0T-3C5Qz+lPQED_EPTV0UIBfNI3KP2d9Jlrc2mrMllmf946bu-CE�dZzUVNZn4lLLBD1R6NE8TKNQZwcfKfc7kl5jfqXArCRSURQZCWJFq43QvT5Bm5iN8n1okQZbHh2u7etM8LKKCF2pY2UdQZoi45cEkbVoJjAbnQpFY87Q)r4 �p�qZdpZdqZqi)r r �
RSA256_KEYr �jwk256�
jwk256json�_wrapped�jwk256_not_comparable�
RSA512_KEY�jwk512�
jwk512json�private�copy�private_json_small� private_json�updater )r
r r r r r' M s.
zJWKRSATest.setUpc C s& | j | jjtj� | j| j| j� d S )N)�assertIsInstancer; r" r ZComparableRSAKeyr r8 )r
r r r �test_init_auto_comparablen s z$JWKRSATest.test_init_auto_comparablec C s" ddl m} | jd|jd�� d S )Nr )r �AA)r r r �
_encode_param)r
r r r r �test_encode_param_zeror s z!JWKRSATest.test_encode_param_zeroc C s$ | j | j| j� | j | j| j� d S )N)r r8 r= )r
r r r � test_equalsx s zJWKRSATest.test_equalsc C s$ | j | j| j� | j | j| j� d S )N)�assertNotEqualr8 r= )r
r r r �test_not_equals| s zJWKRSATest.test_not_equalsc C s* ddl m} | j| j|jtjd��� d S )Nr )r zrsa256_key.pem)r r r r? r r � load_vector)r
r r r r r � s zJWKRSATest.test_loadc C s | j | j| jj� � d S )N)r r8 r? r )r
r r r r. � s zJWKRSATest.test_public_keyc C s@ | j | jj� | j� | j | jj� | j� | j | jj� | j� d S )N)r r8 r( r9 r= r> r? rB )r
r r r r) � s z JWKRSATest.test_to_partial_jsonc C sR ddl m} | j| j|j| j�� | j| j|j| j�� | j| j|j| j �� d S )Nr )r )
r r r r8 r* r9 r= r> r? rB )r
r r r r r+ � s zJWKRSATest.test_from_jsonc C s& ddl m} | j| j|j| j�� d S )Nr )r )r r r r? r* rA )r
r r r r � test_from_json_private_small� s z'JWKRSATest.test_from_json_private_smallc C s, ddl m} | jd= | jtj|j| j� d S )Nr )r r6 )r r rB r r r
r* )r
r r r r �%test_from_json_missing_one_additional� s z0JWKRSATest.test_from_json_missing_one_additionalc C s ddl m} t|j| j�� d S )Nr )r )r r r, r* r9 )r
r r r r r- � s z"JWKRSATest.test_from_json_hashablec C sD ddl m} | jtj|jdddd�� | jtj|jdddd�� d S )Nr )r r0 r1 � )r$ r2 r3 �1)r r r r �DeserializationErrorr* )r
r r r r � test_from_json_non_schema_errors� s
z+JWKRSATest.test_from_json_non_schema_errorsc C s0 ddl m} |jd�}| jtj|j� �d� d S )Nr )r a� {
"kty": "RSA",
"kid": "bilbo.baggins@hobbiton.example",
"use": "sig",
"n": "n4EPtAOCc9AlkeQHPzHStgAbgs7bTZLwUBZdR8_KuKPEHLd4rHVTeT-O-XV2jRojdNhxJWTDvNd7nqQ0VEiZQHz_AJmSCpMaJMRBSFKrKb2wqVwGU_NsYOYL-QtiWN2lbzcEe6XC0dApr5ydQLrHqkHHig3RBordaZ6Aj-oBHqFEHYpPe7Tpe-OfVfHd1E6cS6M1FZcD1NNLYD5lFHpPI9bTwJlsde3uhGqC0ZCuEHg8lhzwOHrtIQbS0FVbb9k3-tVTU4fg_3L_vniUFAKwuCLqKnS2BYwdq_mzSnbLY7h_qixoR7jig3__kRhuaxwUkRz5iaiQkqgc5gHdrNP5zw",
"e": "AQAB"
}s@ f63838e96077ad1fc01c3f8405774dedc0641f558ebb4b40dccf5f9b6d66a932)r r Z
json_loadsr �binasciiZhexlifyr )r
r r" r r r �test_thumbprint_go_jose� s z"JWKRSATest.test_thumbprint_go_joseN)r r r r r r' rE rH rI rK r r. r) r+ rM rN r- rR rT r r r r r/ F s !r/ c @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!S )"� JWKECTestzTests for josepy.jwk.JWKEC.s �L��� lF�J��$ �j�O d�R�������c C s� ddl m} |tj� d�| _|tj� d�| _|tj� d�| _|tj� j d�| _
ddddd�| _ dd d
d d�| _ dd d
dd�| _
|td�| _dddddd�| _| j| _d S )Nr )�JWKEC)r" �ECzP-256z+jjQtV-fA7J_tK8dPzYq7jRPNjF8r5p6LW2R25S2Gw5Uz+EPAw8_8z7PYKsHH6hlGSlsWxFoFl7-0vM0QRGbmnvCc)r$ �crv�x�yzP-384Z@tIhpNtEXkadUbrY84rYGgApFM1X_3l3EWQRuOP1IWtxlTftrZQwneJZF0k0eRn00z@KW2Gp-TThDXmZ-9MJPnD8hv-X130SVvfZRl1a04HPVwIbvLe87mvA_iuOa-myUyvzP-521ZXAFkdl6cKzBmP18U8fffpP4IZN2eED45hDcwRPl5ZeClwHcLtnMBMuWYFFO_Nzm6DL2MhpN0zI2bcMLJd95aY2tPszXAYvZq3wByjt7nQd8nYMqhFNCL3j_-U6GPWZet1hYBY_XZHrC4yIV0R4JnssRAY9eqc1EElpCc4hziis1jiV1iR4WZ+xReNQBKqqTthG8oTmBdhp4EQYImSK1dVqfa2yyMn2rc)r4 rY rZ rX r$ )r rV � EC_P256_KEYr r8 � EC_P384_KEY�jwk384� EC_P521_KEY�jwk521r: r; r9 �
jwk384json�
jwk521jsonr? rB r )r
rV r r r r' � s0
zJWKECTest.setUpc C s& | j | jjtj� | j| j| j� d S )N)rD r; r" r ZComparableECKeyr r8 )r
r r r rE � s z#JWKECTest.test_init_auto_comparablec C s$ ddl m} | jd|jdd�� d S )Nr )rV rF � )r rV r rG )r
rV r r r rH � s z JWKECTest.test_encode_param_zeroc C s4 | j | j| j� | j | j| j� | j | j| j� d S )N)r r8 r] r_ )r
r r r rI � s zJWKECTest.test_equalsc C sd | j | j| j� | j | j| j� | j | j| j� | j | j| j� | j | j| j� | j | j| j� d S )N)rJ r8 r] r_ )r
r r r rK � s zJWKECTest.test_not_equalsc C s* ddl m} | j| j|jtjd��� d S )Nr )rV zec_p256_key.pem)r rV r r? r r rL )r
rV r r r r � s zJWKECTest.test_loadc C s | j | j| jj� � d S )N)r r8 r? r )r
r r r r. s zJWKECTest.test_public_keyc C sT | j | jj� | j� | j | jj� | j� | j | jj� | j� | j | jj� | j � d S )N)
r r8 r( r9 r] r` r_ ra r? rB )r
r r r r) s z JWKECTest.test_to_partial_jsonc C sh ddl m} | j| j|j| j�� | j| j|j| j�� | j| j|j| j �� | j| j
|j| j �� d S )Nr )r ) r r r r8 r* r9 r] r` r_ ra r? rB )r
r r r r r+ s zJWKECTest.test_from_jsonc C s* ddl m} | jd= | jt|j| j� d S )Nr )r rY )r r rB r �KeyErrorr* )r
r r r r �#test_from_json_missing_x_coordinate s z-JWKECTest.test_from_json_missing_x_coordinatec C s* ddl m} | jd= | jt|j| j� d S )Nr )r rZ )r r rB r rc r* )r
r r r r �#test_from_json_missing_y_coordinate s z-JWKECTest.test_from_json_missing_y_coordinatec C s ddl m} t|j| j�� d S )Nr )r )r r r, r* r9 )r
r r r r r- s z!JWKECTest.test_from_json_hashablec C sH ddl m} | jtj|jddddd�� | jtj|jdddd d�� d S )
Nr )r rW zP-256r1 z+m2Fylv-Uz7trgTW8EBHP3FQSMeZs2GNQ6VRo1sIVJEk)r$ rX rY rZ z+jjQtV-fA7J_tK8dPzYq7jRPNjF8r5p6LW2R25S2Gw5UrP )r r r r rQ r* )r
r r r r rR " s
z*JWKECTest.test_from_json_non_schema_errorsc C s, ddl m} | jtj|jddddd�� d S )Nr )r rW zP-255z+jjQtV-fA7J_tK8dPzYq7jRPNjF8r5p6LW2R25S2Gw5Uz+EPAw8_8z7PYKsHH6hlGSlsWxFoFl7-0vM0QRGbmnvCc)r$ rX rY rZ )r r r r rQ r* )r
r r r r �test_unknown_crv_name+ s z JWKECTest.test_unknown_crv_namec C sl ddl m}m} dd l}d}|j|�}|j� }|jj|d �}| j|d d� | jt |�d� |j
|� d S )Nr )rV r s� -----BEGIN EC PRIVATE KEY-----
MHcCAQEEICZ7LCI99Na2KZ/Fq8JmJROakGJ5+J7rHiGSPoO36kOAoAoGCCqGSM49
AwEHoUQDQgAEGS5RvStca15z2FEanCM3juoX7tE/LB7iD44GWawGE40APAl/iZuH
31wQfst4glTZpxkpEI/MzNZHjiYnqrGeSw==
-----END EC PRIVATE KEY-----rZ � ) r rV r �josepyr r( r Zdecode_b64joser �lenr* )r
rV r rh �datar" rZ r r r � test_encode_y_leading_zero_p2563 s
z)JWKECTest.test_encode_y_leading_zero_p256N)r r r r r r' rE rH rI rK r r. r) r+ rd re r- rR rf rk r r r r rU � s" ! rU �__main__)r rS Zunittestrh r r r r rL r Zload_rsa_private_keyr7 r< Zload_ec_private_keyr[ r\ r^ ZTestCaser r r r/ rU r �mainr r r r �<module> s"
u