U
    lufú  ã                   @   sŒ   d Z ddlZddlZddlmZ ddlmZmZm	Z	 ddl
mZmZ erTddlmZ erhddlmZmZ G d	d
„ d
eƒZG dd„ deƒZdS )zTests of the builder registry.é    N)ÚBeautifulSoup)Úbuilder_registryÚHTMLParserTreeBuilderÚTreeBuilderRegistryé   )ÚHTML5LIB_PRESENTÚLXML_PRESENT)ÚHTML5TreeBuilder)ÚLXMLTreeBuilderForXMLÚLXMLTreeBuilderc                   @   s0   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
S )ÚTestBuiltInRegistryz@Test the built-in registry with the default builders registered.c                 C   s\   t  dd¡tkst‚tr@t  dd¡tks,t‚t  dd¡tks@t‚trXt  dd¡tksXt‚d S )NÚstrictÚhtmlÚfastZ
permissiveÚxmlÚhtml5lib)	ÚregistryÚlookupr   ÚAssertionErrorr   r   r
   r   r	   ©Úself© r   úC/tmp/pip-unpacked-wheel-pg4vfqbd/bs4/tests/test_builder_registry.pyÚtest_combination!   s    z$TestBuiltInRegistry.test_combinationc                 C   sj   t r*t d¡tkst‚t d¡tksft‚n<t d¡d ks<t‚trTt d¡tksft‚nt d¡tksft‚d S )Nr   r   )	r   r   r   r   r   r
   r   r	   r   r   r   r   r   Útest_lookup_by_markup_type)   s    z.TestBuiltInRegistry.test_lookup_by_markup_typec                 C   sX   t r,t dd¡tkst‚t dd¡tks,t‚trBt d¡tksBt‚t d¡tksTt‚d S )NÚlxmlr   r   r   zhtml.parser)	r   r   r   r
   r   r   r   r	   r   r   r   r   r   Útest_named_library4   s    z&TestBuiltInRegistry.test_named_libraryc              	   C   sZ   t jdd"}tddd tdddgd W 5 Q R X t t¡ tddd W 5 Q R X d S )NT)ÚrecordÚ r   )Úfeaturesr   zno-such-feature)ÚwarningsÚcatch_warningsr   ÚpytestZraisesÚ
ValueError)r   Úwr   r   r   Ú*test_beautifulsoup_constructor_does_lookup=   s    
z>TestBuiltInRegistry.test_beautifulsoup_constructor_does_lookupN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r%   r   r   r   r   r      s
   	r   c                   @   sX   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Z	dd„ Z
dd„ Zdd„ ZdS )ÚTestRegistryz.Test the TreeBuilderRegistry class in general.c                 C   s   t ƒ | _d S ©N)r   r   r   r   r   r   Úsetup_methodQ   s    zTestRegistry.setup_methodc                 G   s,   t dd |¡ tfd|iƒ}| j |¡ |S )NZBuilder_Ú_r   )ÚtypeÚjoinÚobjectr   Úregister)r   Zfeature_listÚclsr   r   r   Úbuilder_for_featuresT   s     ÿz!TestRegistry.builder_for_featuresc                 C   s2   |   ¡ }| j d¡d kst‚| j ¡ |ks.t‚d S )NÚfoo©r3   r   r   r   ©r   Zbuilderr   r   r   Útest_register_with_no_features[   s    z+TestRegistry.test_register_with_no_featuresc                 C   s8   |   dd¡}| j d¡|ks t‚| j d¡|ks4t‚d S ©Nr4   Úbarr5   r6   r   r   r   Ú0test_register_with_features_makes_lookup_succeedf   s    z=TestRegistry.test_register_with_features_makes_lookup_succeedc                 C   s$   |   dd¡}| j d¡d ks t‚d S ©Nr4   r9   Úbazr5   r6   r   r   r   Ú4test_lookup_fails_when_no_builder_implements_featurek   s    zATestRegistry.test_lookup_fails_when_no_builder_implements_featurec                 C   s*   |   d¡}|   d¡}| j ¡ |ks&t‚d S r8   r5   ©r   Zbuilder1Zbuilder2r   r   r   ÚCtest_lookup_gets_most_recent_registration_when_no_feature_specifiedo   s    

zPTestRegistry.test_lookup_gets_most_recent_registration_when_no_feature_specifiedc                 C   s   | j  ¡ d kst‚d S r+   )r   r   r   r   r   r   r   Ú2test_lookup_fails_when_no_tree_builders_registeredt   s    z?TestRegistry.test_lookup_fails_when_no_tree_builders_registeredc                 C   sv   |   d¡}|   d¡}|   ddd¡}|   ddd¡}|   d¡}|   d¡}| j dd¡|ksZt‚| j ddd¡|ksrt‚d S )Nr4   r9   r<   Zquuxr5   )r   Zhas_oneZhas_the_otherZhas_both_earlyZhas_both_lateZ	lacks_oner   r   r   Ú<test_lookup_gets_most_recent_builder_supporting_all_featuresw   s    



zITestRegistry.test_lookup_gets_most_recent_builder_supporting_all_featuresc                 C   s2   |   dd¡}|   dd¡}| j dd¡d ks.t‚d S r;   r5   r>   r   r   r   Ú:test_lookup_fails_when_cannot_reconcile_requested_features†   s    zGTestRegistry.test_lookup_fails_when_cannot_reconcile_requested_featuresN)r&   r'   r(   r)   r,   r3   r7   r:   r=   r?   r@   rA   rB   r   r   r   r   r*   N   s   r*   )r)   r"   r    Zbs4r   Zbs4.builderr   r   r   r   r   r   r   r	   r
   r   r0   r   r*   r   r   r   r   Ú<module>   s   0