o
    3Ih                      @  sJ  d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl	m
Z
mZ ddlZddlZddlZddlZddlZddlmZmZ dd	lmZ ddlZe
r^dd
lmZmZmZ ddlmZ ddlZddlmZ ddlZdZ e!dZ"dd Z#eG dd dZ$eG dd dZ%eG dd dZ&d&ddZ'ed d!G d"d# d#Z(d$d% Z)dS )'z=
Python representations of the JSON Schema Test Suite tests.
    )annotations)suppress)partial)Path)TYPE_CHECKINGAnyN)fieldfrozen)Registry)IterableMappingSequence)Schema)_VALIDATORSzhttp://localhost:1234z[\W\- ]+c                  C  sB   t jd} | d urt| S ttjjjd } |  std| S )NJSON_SCHEMA_TEST_SUITEjsonzCan't find the JSON-Schema-Test-Suite directory. Set the 'JSON_SCHEMA_TEST_SUITE' environment variable or run the tests from alongside a checkout of the suite.)	osenvirongetr   
jsonschema__file__parentis_dir
ValueError)root r   W/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/jsonschema/tests/_suite.py_find_suite"   s   r   c                   @  s4   e Zd ZU eedZded< dddZdd
dZdS )Suite)factoryr   _rootrunnerpyperf.Runnerc                 C  s,   t  D ]\}}| j|dj||d qd S )N)name)r!   	Validator)r   itemsversion	benchmark)selfr!   r#   r$   r   r   r   r'   :   s   zSuite.benchmarkreturnVersionc                 C  sX   t | }||j}tj|}t jt| j	d ||d|d}t
|| j	d | |dS )Nremotes)r   r#   uri)default_specificationtests)r#   pathr+   )r   ID_OFMETA_SCHEMAreferencingr   specification_withr
   with_contents
remotes_inr    r*   )r(   r#   r$   r,   specificationregistryr   r   r   r&   A   s   zSuite.versionNr!   r"   )r)   r*   )	__name__
__module____qualname__r   r   r    __annotations__r'   r&   r   r   r   r   r   4   s   
 
r   c                   @  s^   e Zd ZU ded< ded< ded< dd ZdddZdddZdddZdd ZdddZ	dS )r*   r   _path%referencing.jsonschema.SchemaRegistry_remotesstrr#   c                 K  s"   |   D ]
}|jdi | qd S Nr   )casesr'   )r(   kwargscaser   r   r   r'   Y   s   zVersion.benchmarkr)   Iterable[_Case]c                 C     | j | jddS )N*.jsonpaths	_cases_inr=   globr(   r   r   r   rB   ]      zVersion.casesc                 C  rF   )Nzoptional/format/*.jsonrH   rJ   rM   r   r   r   format_cases`   rN   zVersion.format_casesc                 C  s   | j | jd | d gdS )Noptionalz.jsonrH   )rK   r=   )r(   r#   r   r   r   optional_cases_ofc   s   zVersion.optional_cases_ofc                   s     dd| j dd }dd  fdd|D D }t|tjf|}tt t	 |_
W d    |S 1 s9w   Y  |S )	Nr#   Test- c                 S  s   i | ]}|j |qS r   )r9   ).0methodr   r   r   
<dictcomp>h   s    z0Version.to_unittest_testcase.<locals>.<dictcomp>c                 3  s6    | ]}|D ]}|j D ]}|jdi  V  qqqd S rA   )r.   to_unittest_method)rU   grouprD   testrC   r   r   	<genexpr>j   s    z/Version.to_unittest_testcase.<locals>.<genexpr>)popr#   titlereplacetypeunittestTestCaser   	Exception)_someone_save_us_the_module_of_the_callerr:   )r(   groupsrC   r#   methodsclsr   r[   r   to_unittest_testcasef   s   
	


zVersion.to_unittest_testcaserI   Iterable[Path]c                 c  s@    |D ]}t |jddD ]}tj|| |j| jdV  qqd S )Nzutf-8)encoding)r&   subjectr+   )r   loads	read_text_Case	from_dictstemr?   )r(   rI   r/   rD   r   r   r   rK   {   s   
zVersion._cases_inN)r)   rE   )r#   r@   r)   rE   )rI   ri   r)   rE   )
r9   r:   r;   r<   r'   rB   rO   rQ   rh   rK   r   r   r   r   r*   Q   s   
 


r*   c                   @  sd   e Zd ZU ded< ded< ded< ded< ded	< d
Zded< dZded< edd ZdddZd
S )rn   r*   r&   r@   rk   descriptionMapping[str, Any] | boolschemazlist[_Test]r.   N
str | Nonecommentr   zSequence[dict[str, str]]r6   c                   s6     |  fdd dD }| dd|i S )Nc              
     s6   g | ]}t d d   d  d  d d|qS )r&   rk   rq   rs   )r&   rk   case_descriptionrs   r+   r   )_Test)rU   rZ   datar+   r   r   
<listcomp>   s    z#_Case.from_dict.<locals>.<listcomp>r.   r   )updater]   )rg   ry   r+   rC   r.   r   rx   r   ro      s
   

z_Case.from_dictr!   r"   c                 K  s,   | j D ]}||jt|jfi | qd S N)r.   
bench_funcfully_qualified_namer   validate_ignoring_errors)r(   r!   rC   rZ   r   r   r   r'      s   
z_Case.benchmarkr8   )	r9   r:   r;   r<   ru   r6   classmethodro   r'   r   r   r   r   rn      s   
 
rn   r   r   r#   r@   r,   r)   Iterable[tuple[str, Schema]]c                 c  s~    |  dD ]6}t| }t|| dd}|dv r#|jdks1d|vr2|dr2||s2qt	 d| |fV  qd S )NrG   \/>   draft3draft4treez$schemadraft)
rglobr   rl   rm   r@   relative_tor_   rp   
startswithMAGIC_REMOTE_URL)r   r#   r,   eachrs   relativer   r   r   r5      s   

r5   F)reprc                   @  s   e Zd ZU ded< ded< ded< ded< ded< d	ed
< ded< ded< dZded< dd Zedd Zdd fddZdd Z	dd Z
dS )rw   r*   r&   r@   rk   rv   rq   r   ry   rr   rs   boolvalidr>   r?   Nrt   ru   c                 C  s   d| j  dS )Nz<Test >)r~   rM   r   r   r   __repr__   s   z_Test.__repr__c                 C  s   d | jj| j| j| jgS )Nz > )joinr&   r#   rk   rv   rq   rM   r   r   r   r~      s   z_Test.fully_qualified_namec                 C  s   d S r|   r   )rZ   r   r   r   <lambda>   s    z_Test.<lambda>c              	     s   j r fdd}n fdd}ddtdjtdjtdjg|_|}|d u s<tj	
dddkr>|S tj	
dddkrLt|S t||S )	Nc                   s   j di   d S rA   )validatethisrC   r(   r   r   fn   rN   z$_Test.to_unittest_method.<locals>.fnc                   sB   |  tj jdi   W d    d S 1 sw   Y  d S rA   )assertRaisesr   ValidationErrorr   r   r   r   r   r      s   "_rZ   JSON_SCHEMA_DEBUG0JSON_SCHEMA_EXPECTED_FAILURES)r   r   _DELIMITERSsubrk   rv   rq   r9   r   r   r   ra   expectedFailureskip)r(   r   rC   r   reasonr   r   r   rX      s    
z_Test.to_unittest_methodc                 K  sN   | | j |d| j| jd|}tjdddkrt  |j| jd d S )N)rs   r7   r   r   )instancer   )	check_schemars   r?   r   r   r   
breakpointr   ry   )r(   r$   rC   	validatorr   r   r   r      s   z_Test.validatec                 C  s<   t tj | j|d W d    d S 1 sw   Y  d S )N)r$   )r   r   r   r   )r(   r$   r   r   r   r     s   "z_Test.validate_ignoring_errors)r9   r:   r;   r<   ru   r   propertyr~   rX   r   r   r   r   r   r   rw      s    
 

rw   c                   C  s   t djd S )a  
    The FQON of the module 2nd stack frames up from here.

    This is intended to allow us to dynamically return test case classes that
    are indistinguishable from being defined in the module that wants them.

    Otherwise, trial will mis-print the FQON, and copy pasting it won't re-run
    the class that really is running.

    Save us all, this is all so so so so so terrible.
       r9   )sys	_getframe	f_globalsr   r   r   r   rd     s   rd   )r   r   r#   r@   r,   r@   r)   r   )*__doc__
__future__r   
contextlibr   	functoolsr   pathlibr   typingr   r   r   r   rer   ra   attrsr   r	   r2   r
   referencing.jsonschemacollections.abcr   r   r   r   pyperfjsonschema.validatorsr   r   r   compiler   r   r   r*   rn   r5   rw   rd   r   r   r   r   <module>   sB    
4
"H