a
}|gM @ s d Z ddlZddlZddlmZ ddlZddlmZ ddlZddlZ ddl
Z
ddl m Z ddl
mZ edZedZed ZG d
d d ejZG d d
d
ejZG dd dejZG dd dejZG dd dejZG dd dejZG dd dejZ G dd dejZ G dd dejZ G d d d ejZ G d d d ejZ G d d! d!ejZ!G d"d# d#ejZ"G d$d% d%ejZ#G d&d' d'ejZ$G d(d) d)ejZ%G d*d+ d+ejZ&e'd,kre(e
)ej*d-d e+g dS ).zTests for acme.messages. N)Dict)mock)
challenges) test_utilzcert.derzcsr.derzrsa512_key.pemc @ sp e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dd Z
dd Zdd ZdS ) ErrorTestz Tests for acme.messages.Error.c C s ddl m} ddl m} ddl m} ddl m} |jdddd | _dd
|d d | _|d d
d| _||dd| _ |jdd
d| j d| _
|jddd| j
gd| _ | | _ d S )Nr Error) ERROR_PREFIX
IdentifierIDENTIFIER_FQDN malformedfootitledetailr z
some title)r r typeZcustombar)typr example.comr valueZcaa)r r
identifier)r r subproblems)
acme.messagesr r r r
with_codeerrorjobj error_customr
subproblemerror_with_subproblems empty_error)selfr r r r
r$ F/usr/lib/python3.9/site-packages/acme/_internal/tests/messages_test.pysetUp s zErrorTest.setUpc C s ddl m} | jdks J d S )Nr r z about:blank)r r r r# r r$ r$ r% test_default_typ( s zErrorTest.test_default_typc C s$ ddl m} | |dks J d S )Nr r {})r r from_jsonr' r$ r$ r% test_from_json_empty, s z ErrorTest.test_from_json_emptyc C s$ ddl m} t|| j d S )Nr r )r r hashr* r to_jsonr' r$ r$ r% test_from_json_hashable0 s z!ErrorTest.test_from_json_hashablec C sF ddl m} || j }dt|jks.J | j|jd ksBJ d S )Nr r )r r r* r! r- lenr r )r# r Z parsed_errorr$ r$ r% test_from_json_with_subproblems4 s z)ErrorTest.test_from_json_with_subproblemsc C s$ d| j jksJ | jjd u s J d S )Nz!The request message was malformed)r descriptionr r# r$ r$ r% test_description< s zErrorTest.test_descriptionc C s@ ddl m} d| jjks J | jjd u s,J | jd u s.MockConstantPOSSIBLE_NAMESN)rG rH rI rM r __annotations__r$ r$ r$ r% MockConstanty s
rO ab)r rL rO const_aconst_b)r# rL rO r$ r$ r% r& v s
zConstantTest.setUpc C s( d| j ksJ d| j ks$J d S )NrP rQ )rR to_partial_jsonrS r3 r$ r$ r% test_to_partial_json s z!ConstantTest.test_to_partial_jsonc C sR | j | jdksJ ttj | jd W d n1 sD0 Y d S )NrP c)rR rO r* r< r= joseZDeserializationErrorr3 r$ r$ r% test_from_json s zConstantTest.test_from_jsonc C s t | jd d S NrP )r, rO r* r3 r$ r$ r% r. s z$ConstantTest.test_from_json_hashablec C s( dt | jksJ dt | jks$J d S )NzMockConstant(a)zMockConstant(b))reprrR rS r3 r$ r$ r% test_repr s zConstantTest.test_reprc C sJ | d}| j| jksJ | j|ks(J | j| jks8J | j|ksFJ d S rY )rO rR rS )r# Z
const_a_primer$ r$ r%
test_equality s
zConstantTest.test_equalityN)
rG rH rI rJ r& rU rX r. r[ r\ r$ r$ r$ r% rK s s
rK c @ sX e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dS )
DirectoryTestz"Tests for acme.messages.Directory.c C s0 ddl m} |dd|jdddgdd | _d S )
Nr Directoryregcert https://example.com/acme/termshttps://www.example.com/r )terms_of_servicewebsiteZcaa_identitiesnewRegZnewCertmeta)r r_ ZMetadirr# r_ r$ r$ r% r& s zDirectoryTest.setUpc C s ddl m} |ddi d S Nr r^ r r )r r_ rj r$ r$ r% !test_init_wrong_key_value_success s z/DirectoryTest.test_init_wrong_key_value_successc C s d| j d ksJ d S )Nr` rg )ri r3 r$ r$ r% test_getitem s zDirectoryTest.test_getitemc C s: t t | jd W d n1 s,0 Y d S Nr )r< r= KeyErrorri __getitem__r3 r$ r$ r% !test_getitem_fails_with_key_error s z/DirectoryTest.test_getitem_fails_with_key_errorc C s d| j jksJ d S )Nr` )ri rg r3 r$ r$ r% test_getattr s zDirectoryTest.test_getattrc C s: t t | jd W d n1 s,0 Y d S rn )r< r= AttributeErrorri __getattr__r3 r$ r$ r% 'test_getattr_fails_with_attribute_error s z5DirectoryTest.test_getattr_fails_with_attribute_errorc C s( | j dddddgddks$J d S )Nr` ra rb rc r )ZtermsOfServicere Z
caaIdentitiesrf )ri r- r3 r$ r$ r% test_to_json s zDirectoryTest.test_to_jsonc C s ddl m} |ddi d S rk )r r_ r* rj r$ r$ r% 2test_from_json_deserialization_unknown_key_success s z@DirectoryTest.test_from_json_deserialization_unknown_key_successc C s6 d}| j jD ] }|dkr | j j| dk}q |s2J d S )NFrd rb )ri rh )r# resultkr$ r$ r% test_iter_meta s
z DirectoryTest.test_iter_metaN)
rG rH rI rJ r& rl rm rq rr ru rv rw rz r$ r$ r$ r% r] s r] c @ s e Zd Zdd Zdd ZdS )ExternalAccountBindingTestc C s> ddl m} tjjt d| _d| _d| _ |ddi| _
d S )Nr r^ keykid-for-testinghmac-key-for-testing
newAccounthttp://url/acme/new-account) r r_ rW jwkJWKRSAKEY
public_keyr} kidhmac_keyri rj r$ r$ r% r& s z ExternalAccountBindingTest.setUpc C sT ddl m} || j| j| j| j}t|dks4J t| tg dksPJ d S )Nr ExternalAccountBinding Z protectedZpayloadZ signature)
r r from_datar} r r ri r0 sortedkeys)r# r eabr$ r$ r% test_from_data s z)ExternalAccountBindingTest.test_from_dataN)rG rH rI r& r r$ r$ r$ r% r{ s r{ c @ sX e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dS )RegistrationTestz%Tests for acme.messages.Registration.c C sj t jjt d}d}d}ddlm} ||||d| _| | _|||d| _ | j
| _ | | j d< d S ) Nr| )mailto:admin@foo.comtel:1234z https://letsencrypt.org/termsr Registration)r} contact agreement)r r r} r} )
rW r r r r r r r` Zreg_nonejobj_tocopy jobj_fromr- )r# r} r r r r$ r$ r% r& s zRegistrationTest.setUpc C s, ddl m} |jddd}|jdks(J d S )Nr r 1234
admin@foo.com)Zphoneemail)r r )r r r r )r# r r` r$ r$ r% r s z RegistrationTest.test_from_datac
C s ddl m} ddl m} ddl m} tjjt d}d}d}|dd i}| ||||}|j d
|d } | j
d ksvJ t | j
t g d
ksJ d S )Nr r^ r NewRegistrationr| r~ r r r r )r external_account_binding)r r )r r_ r r rW r r r r r r r r r )
r# r_ r r r} r r Z directoryr r` r$ r$ r% (test_new_registration_from_data_with_eab s
z9RegistrationTest.test_new_registration_from_data_with_eabc C s d| j jksJ d S )N)r )r` Zphonesr3 r$ r$ r% test_phones s z RegistrationTest.test_phonesc C s d| j jksJ d S )N)r )r` Zemailsr3 r$ r$ r% test_emails s z RegistrationTest.test_emailsc C s | j | j ksJ d S N)r r` rT r3 r$ r$ r% rU s z%RegistrationTest.test_to_partial_jsonc C s& ddl m} | j|| jks"J d S Nr r )r r r` r* r r# r r$ r$ r% rX s z RegistrationTest.test_from_jsonc C s ddl m} t|| j d S r )r r r, r* r r r$ r$ r% r. ! s z(RegistrationTest.test_from_json_hashablec C s| ddl m} | }|dd}|jdks*J |jdks8J d| vsHJ d| vsXJ d| v shJ d| v sxJ d S )Nr r r$ )r r )r r r rT Zfields_to_partial_json)r# r Z
empty_new_regZnew_reg_with_contactr$ r$ r% test_default_not_transmitted% s
z-RegistrationTest.test_default_not_transmittedN)
rG rH rI rJ r& r r r r rU rX r. r r$ r$ r$ r% r s r c @ s e Zd ZdZdd ZdS )UpdateRegistrationTestz+Tests for acme.messages.UpdateRegistration.c C s: ddl m} d}d| ks"J | ||ks6J d S )Nr )UpdateRegistrationz{"resource": "reg"}r) )r r
json_dumpsZ
json_loads)r# r Zjstringr$ r$ r%
test_empty6 s z!UpdateRegistrationTest.test_emptyN)rG rH rI rJ r r$ r$ r$ r% r 3 s r c @ s e Zd ZdZdd Zdd ZdS )RegistrationResourceTestz-Tests for acme.messages.RegistrationResource.c C s, ddl m} |tjjtjjtjjd| _d S )Nr )RegistrationResourcebodyurird )r r r sentinelr r rd regr)r# r r$ r$ r% r& @ s
z RegistrationResourceTest.setUpc C s* | j tjjtjjtjjdks&J d S )Nr )r r- r r r r rd r3 r$ r$ r% rU F s
z-RegistrationResourceTest.test_to_partial_jsonNrG rH rI rJ r& rU r$ r$ r$ r% r = s r c @ s e Zd ZdZdd ZdS )ChallengeResourceTestz*Tests for acme.messages.ChallengeResource.c C s. ddl m} d|tjddddjks*J d S )Nr )ChallengeResource
http://challbr z http://authz)r Z
authzr_uri)r r r Z MagicMockr )r# r r$ r$ r% test_uriQ s z ChallengeResourceTest.test_uriN)rG rH rI rJ r r$ r$ r$ r% r N s r c @ s@ e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dS )ChallengeBodyTestz&Tests for acme.messages.ChallengeBody.c C s t jtdd| _ddlm} ddlm} ddlm} || _ |j
ddd }|d
| j| j |d | _ d
| j d d|d
| _ | j
| _d| jd< ddd| jd< d S )N+evaGxfADs6pSRb2LAv9IZf17Dt3juxGJ-PCt92wr-oAtokenr
ChallengeBodyr )STATUS_INVALIDZserverInternalz%Unable to communicate with DNS server)r r )r challstatusr Zdns)urlr r r r Zinvalidr z)urn:ietf:params:acme:error:serverInternal)r r r )r DNSrW b64decoder r r r r r r challbr r r )r# r r r r r$ r$ r% r& Z s.
zChallengeBodyTest.setUpc C s | j d| j jksJ d S )Nr )r encoder r3 r$ r$ r% test_encodeu s z ChallengeBodyTest.test_encodec C s | j | j ksJ d S r )r r rT r3 r$ r$ r% rU x s z&ChallengeBodyTest.test_to_partial_jsonc C s& ddl m} | j|| jks"J d S Nr r )r r r r* r r# r r$ r$ r% rX { s z ChallengeBodyTest.test_from_jsonc C s ddl m} t|| j d S r )r r r, r* r r r$ r$ r% r. s z)ChallengeBodyTest.test_from_json_hashablec C s t d| jjksJ d S )Nr )rW r r r r3 r$ r$ r%
test_proxy s
z ChallengeBodyTest.test_proxyN)
rG rH rI rJ r& r rU rX r. r r$ r$ r$ r% r W s r c @ s( e Zd ZdZdd Zdd Zdd ZdS ) AuthorizationTestz&Tests for acme.messages.Authorization.c C s ddl m} ddl m} |d|tjddd|d|tjd ddf| _dd
l m} dd l m} dd l m } ||d
d}||| jd| _
| dd | jD d| _ d S )Nr r ) STATUS_VALIDhttp://challb1 IlirfxKKXAsHtmzK29Pj8Ar r r r http://challb2 DGyRejmCefe7v4NfDGDKfA
Authorizationr
r r r r r c S s g | ] }| qS r$ )r- .0r r$ r$ r%
z+AuthorizationTest.setUp..)
r r r r HTTP01r challbsr r r
authzr- r )r# r r r r r
r r$ r$ r% r& s,
zAuthorizationTest.setUpc C s ddl m} || j d S Nr r )r r r* r r# r r$ r$ r% rX s z AuthorizationTest.test_from_jsonc C s ddl m} t|| j d S r )r r r, r* r r r$ r$ r% r. s z)AuthorizationTest.test_from_json_hashableN)rG rH rI rJ r& rX r. r$ r$ r$ r% r s r c @ s e Zd ZdZdd ZdS )AuthorizationResourceTestz.Tests for acme.messages.AuthorizationResource.c C s4 ddl m} |tjjtjjd}t|tjs0J d S )Nr AuthorizationResourcer r ) r r r r r r
isinstancerW JSONDeSerializable)r# r Zauthzrr$ r$ r% test_json_de_serializable s z3AuthorizationResourceTest.test_json_de_serializableN)rG rH rI rJ r r$ r$ r$ r% r s r c @ s e Zd ZdZdd Zdd ZdS )CertificateRequestTestz+Tests for acme.messages.CertificateRequest.c C s ddl m} |td| _d S )Nr CertificateRequest)Zcsr)r r CSRreqr# r r$ r$ r% r& s z CertificateRequestTest.setUpc C s< t | jtjsJ ddlm} | j|| j ks8J d S )Nr r )r r rW r r r r* r- r r$ r$ r% r s z0CertificateRequestTest.test_json_de_serializableNrG rH rI rJ r& r r$ r$ r$ r% r s r c @ s e Zd ZdZdd Zdd ZdS )CertificateResourceTestz0Tests for acme.messages.CertificateResourceTest.c C s* ddl m} |ttjjdtjjd| _d S )Nr CertificateResourcer$ )r r Zauthzrscert_chain_uri)r r CERTr r r r certrr# r r$ r$ r% r& s
z CertificateResourceTest.setUpc C s< t | jtjsJ ddlm} | j|| j ks8J d S )Nr r )r r rW r r r r* r- r r$ r$ r% r s z1CertificateResourceTest.test_json_de_serializableNr r$ r$ r$ r% r s r c @ s e Zd ZdZdd Zdd ZdS )RevocationTestz'Tests for acme.messages.RevocationTest.c C s ddl m} |td| _d S )Nr
Revocation)Z certificate)r r r revr# r r$ r$ r% r& s zRevocationTest.setUpc C s$ ddl m} t|| j d S )Nr r )r r r, r* r r- r r$ r$ r% r. s z&RevocationTest.test_from_json_hashableN)rG rH rI rJ r& r. r$ r$ r$ r% r s r c @ s( e Zd ZdZdd Zdd Zdd ZdS ) OrderResourceTestz&Tests for acme.messages.OrderResource.c C s& ddl m} |tjjtjjd| _d S )Nr
OrderResource)r r )r r r r r r r )r# r r$ r$ r% r& s zOrderResourceTest.setUpc C s& | j tjjtjjd dks"J d S )N)r r authorizations)r r- r r r r r3 r$ r$ r% rU s
z&OrderResourceTest.test_to_partial_jsonc C s ddl m} ddl m} |d|tjddd|d|tjd ddf}dd
l m} dd l m} dd l m} dd
l m } ||dd}|d|||dd} ddl m
}
|
|f|t dd |D d} ddl m } | d| d| fd}
|
|
| |
d S )Nr r )STATUS_PENDINGr r r r r r r r r
r r r z
http://authz1r r )Orderc s s | ] }|j V qd S r r r r$ r$ r% r z>OrderResourceTest.test_json_de_serializable..) identifiersr r r z
http://order1s test blob)r r Zcsr_pemr )r r r r r r r r r r
r tupler Z assertEqualr* r- )r# r r r r r r r
r r r r r Zorderrr$ r$ r% r sF
z+OrderResourceTest.test_json_de_serializableN)rG rH rI rJ r& rU r r$ r$ r$ r% r s r c @ s e Zd ZdZdd Zdd ZdS ) NewOrderTestz!Tests for acme.messages.NewOrder.c C s ddl m} |tjjd| _d S )Nr )NewOrder)r )r r r r r r` )r# r r$ r$ r% r& s zNewOrderTest.setUpc C s | j dtjjiksJ d S )Nr )r` r- r r r r3 r$ r$ r% rU s z!NewOrderTest.test_to_partial_jsonNr r$ r$ r$ r% r s r c @ s e Zd ZdZdd ZdS )JWSPayloadRFC8555CompliantzFTest for RFC8555 compliance of JWS generated from resources/challengesc C s2 ddl m} | }|jdd }|dks.J d S )Nr )NewAuthorization )indents {})r r r r )r# r Z new_orderr r$ r$ r% test_message_payload s z/JWSPayloadRFC8555Compliant.test_message_payloadN)rG rH rI rJ r r$ r$ r$ r% r s r __main__r/ ),rJ
contextlibsystypingr Zunittestr warningsZjosepyrW r< Zacmer Zacme._internal.testsr Zload_comparable_certr Zload_comparable_csrr Zload_rsa_private_keyr ZTestCaser rK r] r{ r r r r r r r r r r r r r rG exitmainargv__file__r$ r$ r$ r% s@
_&8P
1%
.