U j'9@sddlmZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z mZddlmZdd lmZd d lmZd d ZddZddZGdddeZGdddZddddZe e de e dGddde ZeGdddeZd d!Zd"d#Ze e de e dGd$d%d%e Z e e de e d&e e dGd'd(d(e Z!e e de e dGd)d*d*e Z"Gd+d,d,eZ#Gd-d.d.eZ$Gd/d0d0eZ%e e de e d&e e dGd1d2d2e Z&dS)3) annotationsN) dataclass)Protocol)TestCaseskipIf)TypeMachineBuilderpep614)MethodicalMachine) TypeMachine)isTwistedInstalledcCs,z tdWntk r"YdSXdSdS)z2 Is the graphviz Python module installed? ZgraphvizFTN) __import__ ImportErrorrr@/tmp/pip-unpacked-wheel-09g3rnyj/automat/_test/test_visualize.pyisGraphvizModuleInstalleds  rc Cs>t\}}t|ztjd|dd WSt|XdS)z+ Are the graphviz tools installed? dotT)stdinshellN)ospipeclose subprocesscall)rwrrrisGraphvizInstalleds   rcs,tGfdddt}|}|S)zG Create a sample L{MethodicalMachine} with some sample states. cs`eZdZjddddZddZddZd d Z e eee gd S) z#sampleMachine..SampleObjectT)initialcSsdS)z initial stateNrselfrrrbegin0sz)sampleMachine..SampleObject.begincSsdS)z end stateNrrrrrend4sz'sampleMachine..SampleObject.endcSsdS)z sample inputNrrrrrgo8sz&sampleMachine..SampleObject.gocSsdS)z sample outputNrrrrrout<sz'sampleMachine..SampleObject.outN) __name__ __module__ __qualname__stater!r"inputr#outputr$uponrmmrr SampleObject/s     r.)r objectr#)r.sorr,r sampleMachine)s r1c@seZdZddddZdS)SampleNonereturncCsdSNrrrrrr#Hz Sample.goN)r%r&r'r#rrrrr2Gsr2c@s eZdZdS)CoreN)r%r&r'rrrrr8Isr8zTypeMachine[Sample, Core]r4cCsttt}|d}dddddd}|d|}|d }|tj|d |tj|d t|tj|ddd d d d}| S)zA Create a sample L{TypeMachine} with some sample states. r!r2r8int)protocorer5cSsdS)Nr)r:r;rrrbuilditSsz"sampleTypeMachine..builditdatar"Nr3)sampler;r5cSsdSr6r)r?r;rrrr$[szsampleTypeMachine..out) rr2r8r(r+r#toZreturnsrbuild)Zbuilderr!r=r>r"r$rrrsampleTypeMachineLs    rBz!Graphviz module is not installed.zTwisted is not installed.c@s0eZdZdZddZddZddZdd Zd S) ElementMakerTestszL L{elementMaker} generates HTML representing the specified element. cCsddlm}||_dS)Nr ) elementMaker) _visualizerD)r rDrrrsetUphs zElementMakerTests.setUpc Cs"d}|||jddddddS)zK L{elementMaker} orders HTML attributes lexicographically. z
div213bacN assertEqualrDr expectedrrrtest_sortsAttrsmsz!ElementMakerTests.test_sortsAttrsc Cs"d}|||jddddddS)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-
rGz a " quoter za stringrKNrOrQrrrtest_quotesAttrsts z"ElementMakerTests.test_quotesAttrscCsd}|||ddS)zN L{elementMaker} should render an element with no attributes. z
rGNrOrQrrr test_noAttrsszElementMakerTests.test_noAttrsN)r%r&r'__doc__rFrSrTrUrrrrrCas  rCc@s*eZdZUdZded<ded<ded<dS) HTMLElementz2Holds an HTML element, as created by elementMaker.strnamezlist[HTMLElement]childrenzdict[str, str] attributesN)r%r&r'rV__annotations__rrrrrWs rWcs.|r|gSt|rgSfdd|jDS)zk Recursively collect all elements in an L{HTMLElement} tree that match the optional predicate. cs g|]}t|D]}|qqSr) findElements).0childresult predicaterr s z findElements..)isLeafrZ)elementrbrrarr]s r]cCs t|t S)z2 This HTML element is actually leaf node. ) isinstancerWrerrrrdsrdc@s8eZdZdZddZddZddZdd 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}. cOst|||dS)N)rYrZr[)rW)r rYrZr[rrrfakeElementMakersz TableMakerTests.fakeElementMakercCs.ddlm}d|_d|_tj||jd|_dS)Nr ) tableMaker input labelzthe port)Z_E)rErj inputLabelport functoolspartialri)r rjrrrrFs zTableMakerTests.setUpcsfdd}gdgfD]v}jj|jd}t|jd|jd}t||}t|d|djdt|t jgqdS) 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. cst| o|jdjkS)Nrm)rdr[getrmrgrrrhasPortsz3TableMakerTests.test_inputLabelRow..hasPortzan output labelrmrr tdN) rjrlrmZ assertGreaterlenrZr]rPrYrd)r rqZ outputLabelstable inputLabelRowZportCandidatesrrrtest_inputLabelRows   z"TableMakerTests.test_inputLabelRowcCs>|jdd|jd}|t|jd|j\}|d|jdS)z L{tableMaker} does not add a colspan attribute to the input label's cell or a second row if there no output labels. rkrrrr colspanN)rjrmrPrtrZZ assertNotInr[)r rurvrrrtest_noOutputLabelssz#TableMakerTests.test_noOutputLabelscCsh|j|jd|jd}|t|jd|j\}}dd}|tt||d|t|tddgd 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 1output label 2rrr cSs$t| o"|jdko"|jddkS)NrsrxrH)rdrYr[rprgrrrhasCorrectColspans  z@TableMakerTests.test_withOutputLabels..hasCorrectColspanr rzr{N)rjrlrmrPrtrZr]rd)r ruZinputRowZ outputRowr|rrrtest_withOutputLabelss z%TableMakerTests.test_withOutputLabelsN) r%r&r'rVrirFrwryr}rrrrrhs  rhz!Graphviz tools are not installed.c@seZdZdZddddZdS)IntegrationTestsz[ Tests which make sure Graphviz can understand the output produced by Automat. r3r4cCsNt}d|d}tjdtjtjd}||\}}||j ddS)z8 C{graphviz} emits valid graphviz data. zutf-8r)rstdoutrN) r1 asDigraphjoinencoderPopenPIPE communicaterP returncode)r digraphtextpr$errrrrtest_validGraphvizs  z#IntegrationTests.test_validGraphvizN)r%r&r'rVrrrrrr~sr~c@s eZdZdZddZddZdS) SpotCheckszj Tests to make sure that the output contains salient features of the machine being generated. cCsDdt}|d||d||d||d|dS)z The output of L{graphviz.Digraph} should contain the names of the states, inputs, outputs in the state machine. rr!r"r#r$N)rr1rassertInr Zgvoutrrrtest_containsMachineFeaturess    z'SpotChecks.test_containsMachineFeaturescCsPdt}|d||d||d||d||d|dS)z The output of L{graphviz.Digraph} should contain the names of the states, inputs, outputs in the state machine. rr!r"r#z data:builditr$N)rrBrrrrrr test_containsTypeMachineFeaturess     z+SpotChecks.test_containsTypeMachineFeaturesN)r%r&r'rVrrrrrrr s rc@s eZdZdZddZddZdS)RecordsDigraphActionsz/ Records calls made to L{FakeDigraph}. cCs |dSr6)resetrrrr__init__/szRecordsDigraphActions.__init__cCsg|_g|_dSr6) renderCalls saveCallsrrrrr2szRecordsDigraphActions.resetN)r%r&r'rVrrrrrrr*src@s(eZdZdZddZddZddZdS) FakeDigraphzZ A fake L{graphviz.Digraph}. Instantiate it with a L{RecordsDigraphActions}. cCs ||_dSr6) _recorder)r Zrecorderrrrr=szFakeDigraph.__init__cKs|jj|dSr6)rrappendr kwargsrrrrender@szFakeDigraph.rendercKs|jj|dSr6)rrrrrrrsaveCszFakeDigraph.saveN)r%r&r'rVrrrrrrrr7src@s eZdZdZddZddZdS)FakeMethodicalMachinezL A fake L{MethodicalMachine}. Instantiate it with a L{FakeDigraph} cCs ||_dSr6Z_digraph)r rrrrrLszFakeMethodicalMachine.__init__cCs|jSr6rrrrrrOszFakeMethodicalMachine.asDigraphN)r%r&r'rVrrrrrrrGsrc@s^eZdZddZddZddZddd Zd d Zd d ZddZ ddZ ddZ ddZ dS)VisualizeToolTestscCs2t|_t|j|_d|_dg|_g|_d|_dS)Nz tool-testZignoredz fake.fqpn)rdigraphRecorderr fakeDigraph fakeProgname fakeSysPathcollectedOutputfakeFQPNrrrrrFWs  zVisualizeToolTests.setUpcGs|jd|dS)N )rrr)r argsrrr collectPrints`sz VisualizeToolTests.collectPrintsccs|t|jfVdSr6)rr)r ZfqpnrrrfakeFindMachinescsz#VisualizeToolTests.fakeFindMachinesNcCs>ddlm}||p|j|p |jg|p(|j|p0|j|p8|jdS)Nr )tool)Z _prognameZ_argvZ_syspathZ _findMachinesZ_print)rErrrrrr)r ZprognameargvZsyspathZ findMachinesprintrrrrrfs  zVisualizeToolTests.toolcCs&|j|jgd||jdddS)z L{tool} adds '' to sys.path to ensure L{automat._discover.findMachines} searches the current directory. rrrN)rrrPrrrrrtest_checksCurrentDirectoryssz.VisualizeToolTests.test_checksCurrentDirectorycCs@|j|jdgd||j|j|jdgd||jdS)z6 Passing -q/--quiet hides all output. z--quietrz-qN)rr assertFalserrrrrtest_quietHidesOutput|s z(VisualizeToolTests.test_quietHidesOutputcCsdD]}|jg|_|j|j|dgd|tdd|jD|t|jj d|jj \}|d |j|d||jj qd S) ze Passing an empty string for --image-directory/-i disables rendering images. )--image-directoryz-irrcss|]}d|kVqdSimageNrr^linerrr sz6VisualizeToolTests.test_onlySaveDot..r {}.dotfilenameN) rrrrrranyrPrtrformatrr argrrrrtest_onlySaveDots  z#VisualizeToolTests.test_onlySaveDotcCsdD]}|jg|_|j|j|dgd|tdd|jD|t|jj d|jj \}|d |j|d| |d ||jj qd S) zc Passing an empty string for --dot-directory/-d disables saving dot files. )--dot-directoryz-drrcss|]}d|kVqdSrNrrrrrrsz8VisualizeToolTests.test_saveOnlyImage..r rrcleanupN) rrrrrrrrPrtrr assertTruerrrrrtest_saveOnlyImages  z%VisualizeToolTests.test_saveOnlyImagecCsd}d}|j|jd|d|gd|tdd|jD|tdd|jD|t|jjd |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. rrrrrcss|]}d|kVqdSrrrrrrrszQVisualizeToolTests.test_saveDotAndImagesInDifferentDirectories..css|]}d|kVqdSrrrrrrrsr directoryrN) rrrrrrPrtrrr)r ZimageDirectoryZ dotDirectory renderCallZsaveCallrrr+test_saveDotAndImagesInDifferentDirectoriess&   z>VisualizeToolTests.test_saveDotAndImagesInDifferentDirectoriescCsd}|j|jd|d|gd|tdd|jD|t|jjd|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 imagesAndDotrrrcss|]}d|kVqdS)z image and dotNrrrrrrszJVisualizeToolTests.test_saveDotAndImagesInSameDirectory..r rrN) rrrrrrPrtrrrr)r rrrrr$test_saveDotAndImagesInSameDirectorys  z7VisualizeToolTests.test_saveDotAndImagesInSameDirectory)NNNNN) r%r&r'rFrrrrrrrrrrrrrrSs    r)' __future__rrnrrZ dataclassesrtypingrZunittestrrZautomatrrZ _methodicalr Z_typedr Z test_discoverr rrr1r2r8rBrCr/rWr]rdrhr~rrrrrrrrrsP          $  L