0x1949 Team - FAZEMRX - MANAGER
Edit File: test_visualize.cpython-310.pyc
o q+WZ�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$ zt d� W dS ty Y dS w )z2 Is the graphviz Python module installed? �graphvizFT)� __import__� ImportError� r r �>/usr/lib/python3/dist-packages/automat/_test/test_visualize.py�isGraphvizModuleInstalled s ��r c C sB t �� \} }t �|� ztjd| dd� W t �| � S t �| � w )z+ Are the graphviz tools installed? �dotT)�stdin�shell)�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 � dS )z initial stateNr ��selfr r r �begin. � z)sampleMachine.<locals>.SampleObject.beginc S r )z end stateNr r r r r �end1 r z'sampleMachine.<locals>.SampleObject.endc S r )z sample inputNr r r r r �go4 r z&sampleMachine.<locals>.SampleObject.goc S r )z sample outputNr r r r r �out7 r z'sampleMachine.<locals>.SampleObject.outN) �__name__� __module__� __qualname__�stater r �inputr! �outputr" �uponr ��mmr r � SampleObject- s r, )r �objectr! )r, �sor 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)� _visualizer1 )r r1 r r r �setUpF s zElementMakerTests.setUpc C �" 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�� assertEqualr1 �r �expectedr r r �test_sortsAttrsJ s ��z!ElementMakerTests.test_sortsAttrsc C r4 )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>r5 z a " quoter za stringr9 Nr= r? 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>r5 Nr= r? r r r � test_noAttrsb s z ElementMakerTests.test_noAttrsN)r# r$ r% �__doc__r3 rA rB rC r r r r r0 @ s r0 c @ s( e Zd ZdZe�� Ze�� Ze�� ZdS )� HTMLElementz2Holds an HTML element, as created by elementMaker.N) r# r$ r% rD �attr�ib�name�children� attributesr r r r rE j s rE c s. � | �r| gS t | �r g S � fdd�| jD �S )zk Recursively collect all elements in an L{HTMLElement} tree that match the optional predicate. c s g | ] }t |� �D ]}|�q qS r )� findElements)�.0�child�result�� predicater r � <listcomp>| s � �z findElements.<locals>.<listcomp>)�isLeafrI )�elementrP r rO r rK r s �rK c C s t | t� S )z2 This HTML element is actually leaf node. )� isinstancerE �rS r r r rR � s rR 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)rH rI rJ )rE )r rH rI rJ r r r �fakeElementMaker� s z TableMakerTests.fakeElementMakerc C s. ddl m} d| _d| _tj|| jd�| _d S )Nr )� tableMaker� input labelzthe port)�_E)r2 rX � inputLabel�port� functools�partialrW )r rX r r r r3 � s �zTableMakerTests.setUpc s� � fdd�}g dgfD ];}� j � j|� jd�}� �t|j�d� |jd }t||�}� �t|�d� � �|d jd� � �t|t �� jg� q dS ) 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 )Nr\ )rR rJ �getr\ rU r r r �hasPort� s �z3TableMakerTests.test_inputLabelRow.<locals>.hasPortzan output label�r\ r r �tdN) rX r[ r\ � assertGreater�lenrI rK r> rH rR )r r` � outputLabels�table� inputLabelRow�portCandidatesr 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. rY r ra r �colspanN)rX r\ r> rd rI � assertNotInrJ )r rf rg 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 2ra r c S s$ t | � o| jdko| j�d�dkS )Nrb rj r6 )rR rH rJ r_ rU r r r �hasCorrectColspan� s ��z@TableMakerTests.test_withOutputLabels.<locals>.hasCorrectColspanr rm rn N)rX r[ r\ r> rd rI rK rR )r rf �inputRow� outputRowro r r r �test_withOutputLabels� s � � �z%TableMakerTests.test_withOutputLabelsN) r# r$ r% rD rW r3 ri rl rr r r r r rV � s rV z!Graphviz tools are not installed.c @ � 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�PIPE� communicate�joinr/ � asDigraph�encoder> � returncode)r �pr" �errr r r �test_validGraphviz� s ��z#IntegrationTests.test_validGraphvizN)r# r$ r% rD r� r r r r rt � s rt c @ rs )� 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. rv r r r! r" N)rz r/ r{ �assertIn)r �gvoutr r r � test_containsMachineFeatures� s z'SpotChecks.test_containsMachineFeaturesN)r# r$ r% rD r� r r r r r� � s r� c @ � 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 r� )� renderCalls� saveCallsr r r r r� s zRecordsDigraphActions.resetN)r# r$ r% rD r� r� r r r r r� � � r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � FakeDigraphzZ A fake L{graphviz.Digraph}. Instantiate it with a L{RecordsDigraphActions}. c C � || _ d S r� )� _recorder)r �recorderr r r r� � zFakeDigraph.__init__c K � | j j�|� d S r� )r� r� �append�r �kwargsr r r �render � zFakeDigraph.renderc K r� r� )r� r� r� r� r r r �save r� zFakeDigraph.saveN)r# r$ r% rD r� r� r� r r r r r� s r� c @ r� )�FakeMethodicalMachinezL A fake L{MethodicalMachine}. Instantiate it with a L{FakeDigraph} c C r� r� ��_digraph)r �digraphr r r r� r� z FakeMethodicalMachine.__init__c C s | j S r� r� r r r r r{ " s z FakeMethodicalMachine.asDigraphN)r# r$ r% rD r� r{ r r r r r� r� r� zTwisted is not installed.c @ sh e Zd Zdd� Zdd� Zdd� Z ddd �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-test�ignoredz fake.fqpn)r� �digraphRecorderr� � fakeDigraph� fakeProgname� fakeSysPath�collectedOutput�fakeFQPNr r r r r3 + s zVisualizeToolTests.setUpc G s | j �d�|�� d S )N� )r� r� rz )r �argsr r r � collectPrints4 s z VisualizeToolTests.collectPrintsc c s � |t | j�fV d S r� )r� r� )r �fqpnr r r �fakeFindMachines7 s �z#VisualizeToolTests.fakeFindMachinesNc C s> ddl m} ||p | j|p| jg|p| j|p| j|p | jd�S )Nr )�tool)� _progname�_argv�_syspath� _findMachines�_print)r2 r� r� r� r� r� r� )r �progname�argv�syspath� 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 rv N)r� r� r> 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 ]C}| 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-irv r� c s � � | ]}d |v V qdS ��imageNr �rL �liner r r � <genexpr>d � � �z6VisualizeToolTests.test_onlySaveDot.<locals>.<genexpr>r �{}.dot�filenameN) r� r� r� r� r� r� �anyr> rd r� �formatr� �r �argr r r r �test_onlySaveDotZ s � ��z#VisualizeToolTests.test_onlySaveDotc C s� dD ]J}| 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-drv r� c s r� �r Nr r� r r r r� x r� z8VisualizeToolTests.test_saveOnlyImage.<locals>.<genexpr>r r� r� �cleanupN) r� r� r� r� r� r� r� r> rd r� r� � assertTruer� 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 r� r� r r� r r r r� � r� zQVisualizeToolTests.test_saveDotAndImagesInDifferentDirectories.<locals>.<genexpr>c s r� r� r r� r r r r� � r� r � directoryr� N) r� r� r� r� r� r> rd r� r� r� )r �imageDirectory� dotDirectory� renderCall�saveCallr 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. � imagesAndDotr� r� r� c s r� )z image and dotNr r� r r r r� � r� zJVisualizeToolTests.test_saveDotAndImagesInSameDirectory.<locals>.<genexpr>r r� r� N) r� r� r� r� r� r> rd r� r� r� r� )r r� r� r r r �$test_saveDotAndImagesInSameDirectory� s � � z7VisualizeToolTests.test_saveDotAndImagesInSameDirectory)NNNNN) r# r$ r% r3 r� r� r� r� r� r� r� r� r� r r r r r� & s � r� ) � __future__r r] r r �unittestr r rF � _methodicalr � test_discoverr r r r/ r0 �sr- rE rK rR rV rt r� r� r� r� r� r r r r �<module> s> ) N