B
    18\                 @   sl   d Z ddlZddlZddlZddlmZ ddlZddlmZmZ e  G dd dej	Z
edkrhe  dS )z7Tests for the gprof2html script in the Tools directory.    N)mock)skip_if_missingimport_toolc               @   s   e Zd Zdd Zdd ZdS )Gprof2htmlTestsc                s0   t d| _tj  fdd}| | g t_d S )N
gprof2htmlc                  s
    t _d S )N)sysargv )oldargvr	   c/oak/stanford/groups/akundaje/marinovg/programs/Python-3.7.3/Lib/test/test_tools/test_gprof2html.pyfixup   s    z$Gprof2htmlTests.setUp.<locals>.fixup)r   gprofr   r   
addCleanup)selfr   r	   )r
   r   setUp   s
    

zGprof2htmlTests.setUpc          
   C   sr   t j| jdJ}t 6}tj|d}t	|d
  d|gt_| j  W d Q R X W d Q R X | |j	j d S )NZ
webbrowserabcwr   )r   patchobjectr   tempfileTemporaryDirectoryospathjoinopencloser   r   main
assertTruecalled)r   ZwmockZtmpdirfnr	   r	   r   
test_gprof   s    

zGprof2htmlTests.test_gprofN)__name__
__module____qualname__r   r    r	   r	   r	   r   r      s   r   __main__)__doc__r   r   unittestr   r   Ztest.test_toolsr   r   TestCaser   r!   r   r	   r	   r	   r   <module>   s   