File: //lib/python3/dist-packages/hyperlink/test/__pycache__/test_scheme_registration.cpython-38.pyc
U
�UY � @ sH d dl mZ ddlmZ ddlmZ ddlmZmZ G dd� de�Zd S )
� )�unicode_literals� )�_url� )�HyperlinkTestCase)�register_scheme�URLc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zdd� Zd
d� Z dd� Z
dS )�TestSchemeRegistrationc C s t tj�| _ttj�| _d S �N)�dictr �SCHEME_PORT_MAP�_orig_scheme_port_map�set�NO_NETLOC_SCHEMES�_orig_no_netloc_schemes��self� r �I/usr/lib/python3/dist-packages/hyperlink/test/test_scheme_registration.py�setUp s zTestSchemeRegistration.setUpc C s | j t_| jt_d S r
)r
r r r r r r r r �tearDown s zTestSchemeRegistration.tearDownc C s� t dddd� t�d�}|jdks&t�|jdks4t�|jdksBt�t�d�}|jdd�}|�� dksht�t�d �}|�� dks�t�t d
dd� td
d
�}|jdd�}|�� dks�t�d S )NZdeltronTi� ��uses_netloc�default_portzdeltron://example.comzdeltron:zexample.com)Zhostzdeltron://example.com:3030Znonetroni� �r ��schemeznonetron://example.com) r r Z from_textr �AssertionErrorZportr �replace�to_text)r Zu1Zu2Zu3�u4r r r �test_register_scheme_basic s
z1TestSchemeRegistration.test_register_scheme_basicc C s6 t ddd� tdd�}|jdd�}|�� dks2t�d S )NZ noloctronF�r r )Zexample�path)r# znoloctron:example/path)r r r r r )r r r r r �test_register_no_netloc_scheme* s
z5TestSchemeRegistration.test_register_no_netloc_schemec C s( | � t�� tdddd� W 5 Q R X d S )NZ
badnetloclessF� r ��assertRaises�
ValueErrorr r r r r �!test_register_no_netloc_with_port0 s z8TestSchemeRegistration.test_register_no_netloc_with_portc C sJ | � t�� tdd d� W 5 Q R X | � t�� tdt� d� W 5 Q R X d S )NZ badnetlocr"