← Back
Editing: 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"