← Back
Editing: test_visualize.cpython-38.pyc
U ��X�5 � @ s� d dl mZ d dlZd dlZd dlZd dlmZmZ d dlZddl m Z ddl m Z dd � Z d d � Zd d � Zee � d�G dd� de��ZejG dd� de��Zdd� Zdd� Zee � d�G dd� de��Zee � d�ee� d�G dd� de���Zee � d�G d d � d e��ZG d d � d e�ZG d d!� d!e�ZG d"d#� d#e�Zee � d�ee� d�ee � d$�G d%d&� d&e����Z dS )'� )�print_functionN)�TestCase�skipIf� )�MethodicalMachine� )�isTwistedInstalledc C s, z t d� W n tk r" Y dS X dS dS )z2 Is the graphviz Python module installed? ZgraphvizFTN)� __import__� ImportError� r r �>/usr/lib/python3/dist-packages/automat/_test/test_visualize.py�isGraphvizModuleInstalled s r c C s> t �� \} }t �|� ztjd| dd� W �S t �| � X dS )z+ Are the graphviz tools installed? �dotT)�stdin�shellN)�os�pipe�close� subprocess�call)�r�wr r r �isGraphvizInstalled s r c s, t � � G � fdd�dt�} | � }|�� � S )zG Create a sample L{MethodicalMachine} with some sample states. c s` e Zd Z� jdd�dd� �Z� �� dd� �Z� �� dd� �Z� �� d d � �Z e� eee g� d S ) z#sampleMachine.<locals>.SampleObjectT)�initialc S s dS )z initial stateNr ��selfr r r �begin. s z)sampleMachine.<locals>.SampleObject.beginc S s dS )z end stateNr r r r r �end1 s z'sampleMachine.<locals>.SampleObject.endc S s dS )z sample inputNr r r r r �go4 s z&sampleMachine.<locals>.SampleObject.goc S s dS )z sample outputNr r r r r �out7 s z'sampleMachine.<locals>.SampleObject.outN) �__name__� __module__� __qualname__�stater r �inputr �outputr Zuponr �Zmmr r � SampleObject- s r' )r �objectr )r' Zsor r& r � sampleMachine( s r) z!Graphviz module is not installed.c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �ElementMakerTestszL L{elementMaker} generates HTML representing the specified element. c C s ddl m} || _d S )Nr )� elementMaker)� _visualizer+ )r r+ r r r �setUpF s zElementMakerTests.setUpc C s" d}| � || jddddd�� dS )zK L{elementMaker} orders HTML attributes lexicographically. z <div a="1" b="2" c="3"></div>�div�2�1�3��b�a�cN�� assertEqualr+ �r Zexpectedr r r �test_sortsAttrsJ s ��z!ElementMakerTests.test_sortsAttrsc C s" d}| � || jddddd�� dS )z� L{elementMaker} quotes HTML attributes according to DOT's quoting rule. See U{http://www.graphviz.org/doc/info/lang.html}, footnote 1. z-<div a="1" b="a \" quote" c="a string"></div>r. z a " quoter za stringr2 Nr6 r8 r r r �test_quotesAttrsU s ��z"ElementMakerTests.test_quotesAttrsc C s d}| � || �d�� dS )zN L{elementMaker} should render an element with no attributes. z <div ></div>r. Nr6 r8 r r r � test_noAttrsb s z ElementMakerTests.test_noAttrsN)r r! r" �__doc__r- r9 r: r; r r r r r* @ s r* c @ s( e Zd ZdZe�� Ze�� Ze�� ZdS )� HTMLElementz2Holds an HTML element, as created by elementMaker.N) r r! r" r< �attrZib�name�children� attributesr r r r r= j s r= c s. � | �r| gS t | �rg S � fdd�| jD �S )zk Recursively collect all elements in an L{HTMLElement} tree that match the optional predicate. c s g | ]}t |� �D ]}|�qqS r )� findElements)�.0Zchild�result�� predicater r � <listcomp>| s �z findElements.<locals>.<listcomp>)�isLeafr@ )�elementrF r rE r rB r s �rB c C s t | t� S )z2 This HTML element is actually leaf node. )� isinstancer= �rI r r r rH � s rH c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �TableMakerTestsz� Tests that ensure L{tableMaker} generates HTML tables usable as labels in DOT graphs. For more information, read the "HTML-Like Labels" section of U{http://www.graphviz.org/doc/info/shapes.html}. c O s t |||d�S )N)r? r@ rA )r= )r r? r@ rA r r r �fakeElementMaker� s z TableMakerTests.fakeElementMakerc C s. ddl m} d| _d| _tj|| jd�| _d S )Nr )� tableMaker� input labelzthe port)Z_E)r, rN � inputLabel�port� functools�partialrM )r rN r r r r- � s �zTableMakerTests.setUpc s� � fdd�}g dgfD ]v}� j � j|� jd�}� �t|j�d� |jd }t||�}� �t|�d� � �|d jd� � �t|t �� jg� qdS ) z� The table returned by L{tableMaker} always contains the input symbol label in its first row, and that row contains one cell with a port attribute set to the provided port. c s t | � o| j�d�� jkS )NrQ )rH rA �getrQ rK r r r �hasPort� s �z3TableMakerTests.test_inputLabelRow.<locals>.hasPortzan output label�rQ r r �tdN) rN rP rQ Z assertGreater�lenr@ rB r7 r? rH )r rU Z outputLabels�table� inputLabelRowZportCandidatesr r r �test_inputLabelRow� s � �z"TableMakerTests.test_inputLabelRowc C s> | j dd| jd�}| �t|j�d� |j\}| �d|j� dS )z� L{tableMaker} does not add a colspan attribute to the input label's cell or a second row if there no output labels. rO r rV r �colspanN)rN rQ r7 rX r@ Z assertNotInrA )r rY rZ r r r �test_noOutputLabels� s z#TableMakerTests.test_noOutputLabelsc C sh | j | jd| jd�}| �t|j�d� |j\}}dd� }| �tt||��d� | �t|t�ddg� d S ) z� L{tableMaker} adds a colspan attribute to the input label's cell equal to the number of output labels and a second row that contains the output labels. )�output label 1�output label 2rV r c S s$ t | � o"| jdko"| j�d�dkS )NrW r\ r/ )rH r? rA rT rK r r r �hasCorrectColspan� s ��z@TableMakerTests.test_withOutputLabels.<locals>.hasCorrectColspanr r^ r_ N)rN rP rQ r7 rX r@ rB rH )r rY ZinputRowZ outputRowr` r r r �test_withOutputLabels� s � ��z%TableMakerTests.test_withOutputLabelsN) r r! r" r< rM r- r[ r] ra r r r r rL � s rL z!Graphviz tools are not installed.c @ s e Zd ZdZdd� ZdS )�IntegrationTestsz[ Tests which make sure Graphviz can understand the output produced by Automat. c C sF t jdt jt jd�}|�d�t� �� ��d��\}}| �|j d� dS )z8 L{graphviz} emits valid graphviz data. r )r �stdout� zutf-8r N) r �Popen�PIPEZ communicate�joinr) � asDigraph�encoder7 � returncode)r �pr �errr r r �test_validGraphviz� s �� z#IntegrationTests.test_validGraphvizN)r r! r" r< rm r r r r rb � s rb c @ s e Zd ZdZdd� ZdS )� SpotCheckszj Tests to make sure that the output contains salient features of the machine being generated. c C sD d� t� �� �}| �d|� | �d|� | �d|� | �d|� dS )z� The output of L{graphviz} should contain the names of the states, inputs, outputs in the state machine. rd r r r r N)rg r) rh ZassertIn)r Zgvoutr r r � test_containsMachineFeatures� s z'SpotChecks.test_containsMachineFeaturesN)r r! r" r< ro r r r r rn � s rn c @ s e Zd ZdZdd� Zdd� ZdS )�RecordsDigraphActionsz/ Records calls made to L{FakeDigraph}. c C s | � � d S �N)�resetr r r r �__init__ s z RecordsDigraphActions.__init__c C s g | _ g | _d S rq )� renderCalls� saveCallsr r r r rr s zRecordsDigraphActions.resetN)r r! r" r< rs rr r r r r rp � s rp c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � FakeDigraphzZ A fake L{graphviz.Digraph}. Instantiate it with a L{RecordsDigraphActions}. c C s || _ d S rq )� _recorder)r Zrecorderr r r rs s zFakeDigraph.__init__c K s | j j�|� d S rq )rw rt �append�r �kwargsr r r �render s zFakeDigraph.renderc K s | j j�|� d S rq )rw ru rx ry r r r �save s zFakeDigraph.saveN)r r! r" r< rs r{ r| r r r r rv s rv c @ s e Zd ZdZdd� Zdd� ZdS )�FakeMethodicalMachinezL A fake L{MethodicalMachine}. Instantiate it with a L{FakeDigraph} c C s || _ d S rq �Z_digraph)r Zdigraphr r r rs s z FakeMethodicalMachine.__init__c C s | j S rq r~ r r r r rh " s z FakeMethodicalMachine.asDigraphN)r r! r" r< rs rh r r r r r} s r} zTwisted is not installed.c @ s^ e Zd Zdd� Zdd� Zdd� Zddd �Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dS )�VisualizeToolTestsc C s2 t � | _t| j�| _d| _dg| _g | _d| _d S )Nz tool-testZignoredz fake.fqpn)rp �digraphRecorderrv � fakeDigraph� fakeProgname� fakeSysPath�collectedOutput�fakeFQPNr r r r r- + s zVisualizeToolTests.setUpc G s | j �d�|�� d S )N� )r� rx rg )r �argsr r r � collectPrints4 s z VisualizeToolTests.collectPrintsc c s |t | j�fV d S rq )r} r� )r Zfqpnr r r �fakeFindMachines7 s z#VisualizeToolTests.fakeFindMachinesNc C s> ddl m} ||p| j|p | jg|p(| j|p0| j|p8| jd�S )Nr )�tool)Z _prognameZ_argvZ_syspathZ _findMachinesZ_print)r, r� r� r� r� r� r� )r Zprogname�argvZsyspathZ findMachines�printr� r r r r� : s �zVisualizeToolTests.toolc C s& | j | jgd� | �| jd d� dS )z� L{tool} adds '' to sys.path to ensure L{automat._discover.findMachines} searches the current directory. �r� r rd N)r� r� r7 r� r r r r �test_checksCurrentDirectoryH s z.VisualizeToolTests.test_checksCurrentDirectoryc C s@ | j | jdgd� | �| j� | j | jdgd� | �| j� dS )z6 Passing -q/--quiet hides all output. z--quietr� z-qN)r� r� � assertFalser� r r r r �test_quietHidesOutputQ s z(VisualizeToolTests.test_quietHidesOutputc C s� dD ]�}| j �� g | _| j| j|dgd� | �tdd� | jD ��� | �t| j j �d� | j j \}| �d� | j�|d � | �| j j � qd S ) ze Passing an empty string for --image-directory/-i disables rendering images. )�--image-directoryz-ird r� c s s | ]}d |kV qdS ��imageNr �rC �liner r r � <genexpr>d s �z6VisualizeToolTests.test_onlySaveDot.<locals>.<genexpr>r �{}.dot�filenameN) r� rr r� r� r� r� �anyr7 rX ru �formatrt �r �argr r r r �test_onlySaveDotZ s � �z#VisualizeToolTests.test_onlySaveDotc C s� dD ]�}| j �� g | _| j| j|dgd� | �tdd� | jD ��� | �t| j j �d� | j j \}| �d� | j�|d � | � |d � | �| j j � qd S ) zc Passing an empty string for --dot-directory/-d disables saving dot files. )�--dot-directoryz-drd r� c s s | ]}d |kV qdS �r Nr r� r r r r� x s �z8VisualizeToolTests.test_saveOnlyImage.<locals>.<genexpr>r r� r� �cleanupN) r� rr r� r� r� r� r� r7 rX rt r� � assertTrueru r� r r r �test_saveOnlyImagen s � �z%VisualizeToolTests.test_saveOnlyImagec C s� d}d}| j | jd|d|gd� | �tdd� | jD ��� | �tdd� | jD ��� | �t| jj�d � | jj\}| �|d |� | �|d � | �t| jj �d � | jj \}| �|d |� d S ) z� Passing different directories to --image-directory and --dot-directory writes images and dot files to those directories. r� r r� r� r� c s s | ]}d |kV qdS r� r r� r r r r� � s �zQVisualizeToolTests.test_saveDotAndImagesInDifferentDirectories.<locals>.<genexpr>c s s | ]}d |kV qdS r� r r� r r r r� � s �r � directoryr� N) r� r� r� r� r� r7 rX r� rt ru )r ZimageDirectoryZ dotDirectory� renderCallZsaveCallr r r �+test_saveDotAndImagesInDifferentDirectories� s* � � � z>VisualizeToolTests.test_saveDotAndImagesInDifferentDirectoriesc C s� d}| j | jd|d|gd� | �tdd� | jD ��� | �t| jj�d� | jj\}| �|d |� | � |d � | � t| jj �� d S ) z� Passing the same directory to --image-directory and --dot-directory writes images and dot files to that one directory. Z imagesAndDotr� r� r� c s s | ]}d |kV qdS )z image and dotNr r� r r r r� � s �zJVisualizeToolTests.test_saveDotAndImagesInSameDirectory.<locals>.<genexpr>r r� r� N) r� r� r� r� r� r7 rX r� rt r� ru )r r� r� r r r �$test_saveDotAndImagesInSameDirectory� s � � z7VisualizeToolTests.test_saveDotAndImagesInSameDirectory)NNNNN) r r! r" r- r� r� r� r� r� r� r� r� r� r r r r r & s � r ) Z __future__r rR r r Zunittestr r r> Z _methodicalr Z test_discoverr r r r) r* �sr( r= rB rH rL rb rn rp rv r} r r r r r �<module> s< ) N
Save File
Cancel