Your IP : 216.73.217.32
U
i�f�) � @ sB d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ dZdZ G dd� dej
�Z G dd � d �Z G d
d � d e ej
�ZG d d
� d
e ej
�ZG dd� de ej
�ZG dd� de ej
�ZG dd� de ej
�ZG dd� de ej
�ZG dd� de ej
�ZG dd� de ej
�ZG dd� dej
�ZG d d � d ej
�Zed k�r>e�� dS ) � N)�mock)�supportzhttp://www.example.com�testc @ s e Zd Zdd� Zddd�ZdS )� PopenMockc C s dS �Nr � ��selfr r �9/opt/alt/python38/lib64/python3.8/test/test_webbrowser.py�poll s zPopenMock.pollNc C s dS r r )r Zsecondsr r r
�wait s zPopenMock.wait)N)�__name__�
__module__� __qualname__r r r r r r
r s r c @ s e Zd Zegi d�dd�ZdS )�CommandTestMixin)�args�kwc
C s� t � }t�| td|� | jtd�}t||�||� tjjd d }| � |d t� |�
d� |D ] } | � | |� |�
|� | �� q`| � ||� d S )N�Popen)�namer )
r r Zpatch�
subprocess�
browser_class�CMD_NAME�getattrr Z call_args� assertEqual�pop�assertIn�index)
r Zmethr r �options� arguments�popen�browserZ
popen_argsZoptionr r r
�_test s
zCommandTestMixin._testN)r
r r �URLr! r r r r
r s r c @ s e Zd ZejZdd� ZdS )�GenericBrowserCommandTestc C s | j dg tgd� d S �N�open�r r �r! r" r r r r
� test_open5 s �z#GenericBrowserCommandTest.test_openN)r
r r �
webbrowser�GenericBrowserr r( r r r r
r# 1 s r# c @ s e Zd ZejZdd� ZdS )� BackgroundBrowserCommandTestc C s | j dg tgd� d S r$ r' r r r r
r( ? s �z&BackgroundBrowserCommandTest.test_openN)r
r r r) ZBackgroundBrowserr r( r r r r
r+ ; s r+ c @ s2 e Zd ZejZdd� Zdd� Zdd� Zdd� Z d S )
�ChromeCommandTestc C s | j dg tgd� d S r$ r' r r r r
r( I s �zChromeCommandTest.test_openc C s | j dtdd�g tgd� d S �Nr% F�Z autoraise�r r r �r! �dictr" r r r r
� test_open_with_autoraise_falseN s �z0ChromeCommandTest.test_open_with_autoraise_falsec C s | j ddgtgd� d S �N�open_newz --new-windowr& r' r r r r
�
test_open_newS s �z ChromeCommandTest.test_open_newc C s | j dg tgd� d S �N� open_new_tabr& r' r r r r
�test_open_new_tabX s �z#ChromeCommandTest.test_open_new_tabN)
r
r r r) ZChromer r( r2 r5 r8 r r r r
r, E s
r, c @ s2 e Zd ZejZdd� Zdd� Zdd� Zdd� Z d S )
�MozillaCommandTestc C s | j dg tgd� d S r$ r' r r r r
r( b s �z MozillaCommandTest.test_openc C s | j dtdd�g tgd� d S r- r0 r r r r
r2 g s �z1MozillaCommandTest.test_open_with_autoraise_falsec C s | j dg dtgd� d S )Nr4 z -new-windowr& r' r r r r
r5 l s �z MozillaCommandTest.test_open_newc C s | j dg dtgd� d S )Nr7 z-new-tabr& r' r r r r
r8 q s �z$MozillaCommandTest.test_open_new_tabN)
r
r r r) ZMozillar r( r2 r5 r8 r r r r
r9 ^ s
r9 c @ s2 e Zd ZejZdd� Zdd� Zdd� Zdd� Z d S )
�NetscapeCommandTestc C s | j dddgd�t�gd� d S )Nr% �-raise�-remote� openURL({})r& �r! �formatr" r r r r
r( { s
�z NetscapeCommandTest.test_openc C s( | j dtdd�ddgd�t�gd� d S )Nr% Fr. �-noraiser< r= r/ )r! r1 r? r" r r r r
r2 � s
�z2NetscapeCommandTest.test_open_with_autoraise_falsec C s | j dddgd�t�gd� d S )Nr4 r; r< �openURL({},new-window)r& r> r r r r
r5 � s
�z!NetscapeCommandTest.test_open_newc C s | j dddgd�t�gd� d S )Nr7 r; r< �openURL({},new-tab)r& r> r r r r
r8 � s
�z%NetscapeCommandTest.test_open_new_tabN)
r
r r r) ZNetscaper r( r2 r5 r8 r r r r
r: w s
r: c @ s2 e Zd ZejZdd� Zdd� Zdd� Zdd� Z d S )
�GaleonCommandTestc C s | j ddgtgd� d S )Nr% �-nr& r' r r r r
r( � s �zGaleonCommandTest.test_openc C s" | j dtdd�ddgtgd� d S )Nr% Fr. r@ rD r/ r0 r r r r
r2 � s �z0GaleonCommandTest.test_open_with_autoraise_falsec C s | j ddgtgd� d S )Nr4 �-wr& r' r r r r
r5 � s �z GaleonCommandTest.test_open_newc C s | j ddgtgd� d S )Nr7 rE r& r' r r r r
r8 � s �z#GaleonCommandTest.test_open_new_tabN)
r
r r r) ZGaleonr r( r2 r5 r8 r r r r
rC � s
rC c @ s2 e Zd ZejZdd� Zdd� Zdd� Zdd� Z d S )
�OperaCommandTestc C s | j dg tgd� d S r$ r' r r r r
r( � s �zOperaCommandTest.test_openc C s | j dtdd�g tgd� d S r- r0 r r r r
r2 � s �z/OperaCommandTest.test_open_with_autoraise_falsec C s | j ddgtgd� d S r3 r' r r r r
r5 � s �z OperaCommandTest.test_open_newc C s | j dg tgd� d S r6 r' r r r r
r8 � s �z"OperaCommandTest.test_open_new_tabN)
r
r r r) ZOperar r( r2 r5 r8 r r r r
rF � s
rF c @ s2 e Zd ZejZdd� Zdd� Zdd� Zdd� Z d S )
�ELinksCommandTestc C s | j ddgd�t�gd� d S �Nr% r< r= r& r> r r r r
r( � s
�zELinksCommandTest.test_openc C s | j ddgd�t�gd� d S rH r> r r r r
r2 � s
�z0ELinksCommandTest.test_open_with_autoraise_falsec C s | j ddgd�t�gd� d S )Nr4 r< rA r& r> r r r r
r5 � s
�z ELinksCommandTest.test_open_newc C s | j ddgd�t�gd� d S )Nr7 r< rB r& r> r r r r
r8 � s
�z#ELinksCommandTest.test_open_new_tabN)
r
r r r) ZElinksr r( r2 r5 r8 r r r r
rG � s
rG c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�BrowserRegistrationTestc C s t j| _g t _t j| _i t _d S �N)r) � _tryorder�_saved_tryorder� _browsers�_saved_browsersr r r r
�setUp� s z BrowserRegistrationTest.setUpc C s | j t_| jt_d S rJ )rL r) rK rN rM r r r r
�tearDown� s z BrowserRegistrationTest.tearDownc C s� G dd� d�}g }i }| � tj|� | � tj|� t�d|� dg}|d g|d< | � tj|� | � tj|� |� }|d k r�tjd|||d� nt�d||� |r�ddg}nddg}||g|d< | � tj|� | � tj|� d S )Nc @ s e Zd ZdS )zCBrowserRegistrationTest._check_registration.<locals>.ExampleBrowserN�r
r r r r r r
�ExampleBrowser� s rR �Example1�example1Zexample2�� preferred)r r) rK rM �register)r rV rR Zexpected_tryorderZexpected_browsers�instancer r r
�_check_registration� s,
�
z+BrowserRegistrationTest._check_registrationc C s | j dd� d S )NFrU �rY r r r r
�
test_register s z%BrowserRegistrationTest.test_registerc C s | j d d� d S )NrU rZ r r r r
�test_register_default s z-BrowserRegistrationTest.test_register_defaultc C s | j dd� d S )NTrU rZ r r r r
�test_register_preferred s z/BrowserRegistrationTest.test_register_preferredN) r
r r rO rP rY r[ r\ r] r r r r
rI � s rI c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �
ImportTestc C s� t �d�}| �|j� | �|j� G dd� d�}|�d|� | �|j� | �|jd d� | �|j� | � d|j� | �|jd |d g� d S )Nr) c @ s e Zd ZdS )z0ImportTest.test_register.<locals>.ExampleBrowserNrQ r r r r
rR s rR rS ���rT )
r �import_fresh_module� assertIsNonerK � assertFalserM rW Z
assertTruer r )r r) rR r r r
r[ s
zImportTest.test_registerc C sT t �d�}| �|j� | �|j� | �|j�� |�d� W 5 Q R X | � |j� d S )Nr) Z fakebrowser)
r r` ra rK rb rM Z assertRaises�Error�getZassertIsNotNone)r r) r r r
�test_get s
zImportTest.test_getc C s@ t �d�}tj�tj��� }|�|d |� |�� |�
tj� d S )Nr) ) r r` �os�path�basename�sys�
executable�lowerrW r* rd )r r) r r r r
�test_synthesize' s
zImportTest.test_synthesizec
C s� t �d�}z|�� j}W n6 |jtfk
rN } z| �t|�� W 5 d }~X Y nX t �� � }||d< t �d�}|�� W 5 Q R X d S )Nr) �BROWSER) r r` rd r rc �AttributeError�skipTest�str�EnvironmentVarGuard)r r) r �err�envr r r
�test_environment- s
zImportTest.test_environmentc
C s� t �d�}z |�� |�|jd �j}W n8 |jttfk
r` } z| �t |�� W 5 d }~X Y nX t �
� �*}||d< t �d�}| � |�� j|� W 5 Q R X t �
� �.}t j
|d< t �d�}| � |�� jt j
� W 5 Q R X d S )Nr) r_ rm )r r` rd rK r rc rn �
IndexErrorro rp rq r ri rj )r r) Zleast_preferred_browserrr rs r r r
�test_environment_preferred8 s
z%ImportTest.test_environment_preferredN)r
r r r[ re rl rt rv r r r r
r^ s
r^ �__main__)r) Zunittestrf ri r r r r r" r Z MagicMockr r ZTestCaser# r+ r, r9 r: rC rF rG rI r^ r
�mainr r r r
�<module> s,
5<