Your IP : 216.73.216.196
B
� 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/python37/lib64/python3.7/test/test_webbrowser.py�poll s zPopenMock.pollNc C s dS )Nr 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� x(|D ] } | � | |� |�
|� | �� qbW | � ||� dS )aj Given a web browser instance method name along with arguments and
keywords for same (which defaults to the single argument URL), creates
a browser instance from the class pointed to by self.browser, calls the
indicated instance method with the indicated arguments, and compares
the resulting options and arguments passed to Popen by the browser
instance against the 'options' and 'args' lists. Options are compared
in a position independent fashion, and the arguments are compared in
sequence order to whatever is left over after removing the options.
�Popen)�namer N)
r r Zpatch�
subprocess�
browser_class�CMD_NAME�getattrr � 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 )Nr# )r r )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 )Nr# )r r )r r! )r r r r r$ I s zChromeCommandTest.test_openc C s | j dtdd�g tgd� d S )Nr# F)� 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-window)r r )r r! )r r r r �
test_open_newS s z ChromeCommandTest.test_open_newc C s | j dg tgd� d S )N� open_new_tab)r r )r r! )r r r r �test_open_new_tabX s z#ChromeCommandTest.test_open_new_tabN)
r r r
r% ZChromer r$ r+ r- r/ 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 )Nr# )r r )r r! )r r r r r$ b s z MozillaCommandTest.test_openc C s | j dtdd�g tgd� d S )Nr# F)r) )r r r )r r* r! )r r r r r+ g s z1MozillaCommandTest.test_open_with_autoraise_falsec C s | j dg dtgd� d S )Nr, z -new-window)r r )r r! )r r r r r- l s z MozillaCommandTest.test_open_newc C s | j dg dtgd� d S )Nr. z-new-tab)r r )r r! )r r r r r/ q s z$MozillaCommandTest.test_open_new_tabN)
r r r
r% ZMozillar r$ r+ r- r/ r r r r r0 ^ s
r0 c @ s2 e Zd ZejZdd� Zdd� Zdd� Zdd� Z d S )
�NetscapeCommandTestc C s | j dddgd�t�gd� d S )Nr# z-raisez-remotez openURL({}))r r )r �formatr! )r r r r r$ { s z NetscapeCommandTest.test_openc C s( | j dtdd�ddgd�t�gd� d S )Nr# F)r) z-noraisez-remotez openURL({}))r r r )r r* r2 r! )r r r r r+ � s z2NetscapeCommandTest.test_open_with_autoraise_falsec C s | j dddgd�t�gd� d S )Nr, z-raisez-remotezopenURL({},new-window))r r )r r2 r! )r r r r r- � s z!NetscapeCommandTest.test_open_newc C s | j dddgd�t�gd� d S )Nr. z-raisez-remotezopenURL({},new-tab))r r )r r2 r! )r r r r r/ � s z%NetscapeCommandTest.test_open_new_tabN)
r r r
r% ZNetscaper r$ r+ r- r/ r r r r r1 w s
r1 c @ s2 e Zd ZejZdd� Zdd� Zdd� Zdd� Z d S )
�GaleonCommandTestc C s | j ddgtgd� d S )Nr# z-n)r r )r r! )r r r r r$ � s zGaleonCommandTest.test_openc C s" | j dtdd�ddgtgd� d S )Nr# F)r) z-noraisez-n)r r r )r r* r! )r r r r r+ � s z0GaleonCommandTest.test_open_with_autoraise_falsec C s | j ddgtgd� d S )Nr, z-w)r r )r r! )r r r r r- � s z GaleonCommandTest.test_open_newc C s | j ddgtgd� d S )Nr. z-w)r r )r r! )r r r r r/ � s z#GaleonCommandTest.test_open_new_tabN)
r r r
r% ZGaleonr r$ r+ r- r/ r r r r r3 � s
r3 c @ s2 e Zd ZejZdd� Zdd� Zdd� Zdd� Z d S )
�OperaCommandTestc C s | j dg tgd� d S )Nr# )r r )r r! )r r r r r$ � s zOperaCommandTest.test_openc C s | j dtdd�g tgd� d S )Nr# F)r) )r r r )r r* r! )r r r r r+ � s z/OperaCommandTest.test_open_with_autoraise_falsec C s | j ddgtgd� d S )Nr, z --new-window)r r )r r! )r r r r r- � s z OperaCommandTest.test_open_newc C s | j dg tgd� d S )Nr. )r r )r r! )r r r r r/ � s z"OperaCommandTest.test_open_new_tabN)
r r r
r% ZOperar r$ r+ r- r/ r r r r r4 � s
r4 c @ s2 e Zd ZejZdd� Zdd� Zdd� Zdd� Z d S )
�ELinksCommandTestc C s | j ddgd�t�gd� d S )Nr# z-remotez openURL({}))r r )r r2 r! )r r r r r$ � s
zELinksCommandTest.test_openc C s | j ddgd�t�gd� d S )Nr# z-remotez openURL({}))r r )r r2 r! )r r r r r+ � s z0ELinksCommandTest.test_open_with_autoraise_falsec C s | j ddgd�t�gd� d S )Nr, z-remotezopenURL({},new-window))r r )r r2 r! )r r r r r- � s z ELinksCommandTest.test_open_newc C s | j ddgd�t�gd� d S )Nr. z-remotezopenURL({},new-tab))r r )r r2 r! )r r r r r/ � s z#ELinksCommandTest.test_open_new_tabN)
r r r
r% ZElinksr r$ r+ r- r/ r r r r r5 � s
r5 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_browsers)r r r r �setUp� s z BrowserRegistrationTest.setUpc C s | j t_| jt_d S )N)r8 r% r7 r: r9 )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 r= �Example1�example1Zexample2)� preferred)r r% r7 r9 �register)r r@ r= Zexpected_tryorderZexpected_browsers�instancer r r �_check_registration� s*
z+BrowserRegistrationTest._check_registrationc C s | j dd� d S )NF)r@ )rC )r r r r �
test_register s z%BrowserRegistrationTest.test_registerc C s | j d d� d S )N)r@ )rC )r r r r �test_register_default s z-BrowserRegistrationTest.test_register_defaultc C s | j dd� d S )NT)r@ )rC )r r r r �test_register_preferred s z/BrowserRegistrationTest.test_register_preferredN) r r r
r; r< rC rD rE rF r r r r r6 � s r6 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>.ExampleBrowserN)r r r
r r r r r= s r= r> ���r? )
r �import_fresh_module� assertIsNoner7 � assertFalser9 rA Z
assertTruer r )r r% r= r r r rD s
zImportTest.test_registerc C sT t �d�}| �|j� | �|j� | �|j�� |�d� W d Q R X | � |j� d S )Nr% Z fakebrowser)
r rI rJ r7 rK r9 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 rI �os�path�basename�sys�
executable�lowerrA r& rM )r r% r r r r �test_synthesize' s
zImportTest.test_synthesizec
C s� t �d�}y|�� j}W n6 |jtfk
rN } z| �t|�� W d d }~X Y nX t �� � }||d< t �d�}|�� W d Q R X d S )Nr% �BROWSER) r rI rM r rL �AttributeError�skipTest�str�EnvironmentVarGuard)r r% r �err�envr r r �test_environment- s
zImportTest.test_environmentc
C s� t �d�}y |�� |�|jd �j}W n8 |jttfk
r` } z| �t |�� W d d }~X Y nX t �
� �*}||d< t �d�}| � |�� j|� W d Q R X t �
� �.}t j
|d< t �d�}| � |�� jt j
� W d Q R X d S )Nr% rH rV )r rI rM r7 r rL rW �
IndexErrorrX rY rZ r rR rS )r r% Zleast_preferred_browserr[ r\ r r r �test_environment_preferred8 s
z%ImportTest.test_environment_preferredN)r r r
rD rN rU r] r_ r r r r rG s
rG �__main__)r% ZunittestrO rR r r r r r! r Z MagicMockr r ZTestCaser" r' r( r0 r1 r3 r4 r5 r6 rG r �mainr r r r �<module> s,
5<