U [e5@sddlmZddlZddlZddlZddlmZmZddlZddl m Z ddl m Z dd Z d d Zd d Zee dGdddeZejGdddeZddZddZee dGdddeZee dee dGdddeZee dGdddeZGdddeZGd d!d!eZGd"d#d#eZee dee dee d$Gd%d&d&eZdS)')print_functionN)TestCaseskipIf)MethodicalMachine)isTwistedInstalledcCs,z tdWntk r"YdSXdSdS)z2 Is the graphviz Python module installed? graphvizFTN) __import__ ImportErrorr r ^/var/www/html/services/stratfitenv/lib/python3.8/site-packages/automat/_test/test_visualize.pyisGraphvizModuleInstalleds  rc Cs>t\}}t|ztjd|dd WSt|XdS)z+ Are the graphviz tools installed? dotT)stdinshellN)ospipeclose subprocesscall)rwr r r isGraphvizInstalleds   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 stateNr selfr r r begin.sz)sampleMachine..SampleObject.begincSsdS)z end stateNr rr r r end1sz'sampleMachine..SampleObject.endcSsdS)z sample inputNr rr r r go4sz&sampleMachine..SampleObject.gocSsdS)z sample outputNr rr r r out7sz'sampleMachine..SampleObject.outN) __name__ __module__ __qualname__staterrinputroutputr Zuponr mmr r SampleObject-s     r))robjectr)r)sor r'r sampleMachine(s r,z!Graphviz module is not installed.c@s0eZdZdZddZddZddZdd Zd S) ElementMakerTestszL L{elementMaker} generates HTML representing the specified element. cCsddlm}||_dS)Nr) elementMaker) _visualizer.)rr.r r r setUpFs zElementMakerTests.setUpc Cs"d}|||jddddddS)zK L{elementMaker} orders HTML attributes lexicographically. z
div213bacN assertEqualr.rexpectedr r r test_sortsAttrsJsz!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-
r1z a " quoterza stringr5Nr9r;r r r test_quotesAttrsUsz"ElementMakerTests.test_quotesAttrscCsd}|||ddS)zN L{elementMaker} should render an element with no attributes. z
r1Nr9r;r r r test_noAttrsbszElementMakerTests.test_noAttrsN)r!r"r#__doc__r0r=r>r?r r r r r-@s   r-c@s(eZdZdZeZeZeZdS) HTMLElementz2Holds an HTML element, as created by elementMaker.N) r!r"r#r@attrZibnamechildren attributesr r r r rAjsrAcs.|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 predicater r |s z findElements..)isLeafrD)elementrKr rJr rFrs rFcCs t|t S)z2 This HTML element is actually leaf node. ) isinstancerArNr r r rMsrMc@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)rCrDrE)rA)rrCrDrEr r r fakeElementMakersz TableMakerTests.fakeElementMakercCs.ddlm}d|_d|_tj||jd|_dS)Nr) tableMaker input labelzthe port)Z_E)r/rS inputLabelport functoolspartialrR)rrSr r r r0s  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)NrV)rMrEgetrVrPrr r hasPorts z3TableMakerTests.test_inputLabelRow..hasPortzan output labelrVrrtdN) rSrUrVZ assertGreaterlenrDrFr:rCrM)rrZZ outputLabelstable inputLabelRowZportCandidatesr rr test_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. rTr r[rcolspanN)rSrVr:r]rDZ assertNotInrE)rr^r_r r r test_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 2r[rcSs$t| o"|jdko"|jddkS)Nr\rar2)rMrCrErYrPr r r hasCorrectColspans  z@TableMakerTests.test_withOutputLabels..hasCorrectColspanrrcrdN)rSrUrVr:r]rDrFrM)rr^ZinputRowZ outputRowrer r r test_withOutputLabelss  z%TableMakerTests.test_withOutputLabelsN) r!r"r#r@rRr0r`rbrfr r r r rQs  rQz!Graphviz tools are not installed.c@seZdZdZddZdS)IntegrationTestsz[ Tests which make sure Graphviz can understand the output produced by Automat. cCsFtjdtjtjd}|dtd\}}||j ddS)z8 L{graphviz} emits valid graphviz data. r)rstdoutzutf-8rN) rPopenPIPE communicatejoinr, asDigraphencoder: returncode)rpr errr r r test_validGraphvizs  z#IntegrationTests.test_validGraphvizN)r!r"r#r@rsr r r r rgsrgc@seZdZdZddZdS) 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} should contain the names of the states, inputs, outputs in the state machine. rirrrr N)rmr,rnZassertIn)rZgvoutr r r test_containsMachineFeaturess    z'SpotChecks.test_containsMachineFeaturesN)r!r"r#r@rur r r r rtsrtc@s eZdZdZddZddZdS)RecordsDigraphActionsz/ Records calls made to L{FakeDigraph}. cCs |dSN)resetrr r r __init__szRecordsDigraphActions.__init__cCsg|_g|_dSrw) renderCalls saveCallsrr r r rxszRecordsDigraphActions.resetN)r!r"r#r@ryrxr r r r rvsrvc@s(eZdZdZddZddZddZdS) FakeDigraphzZ A fake L{graphviz.Digraph}. Instantiate it with a L{RecordsDigraphActions}. cCs ||_dSrw) _recorder)rZrecorderr r r ryszFakeDigraph.__init__cKs|jj|dSrw)r}rzappendrkwargsr r r renderszFakeDigraph.rendercKs|jj|dSrw)r}r{r~rr r r saveszFakeDigraph.saveN)r!r"r#r@ryrrr r r r r| sr|c@s eZdZdZddZddZdS)FakeMethodicalMachinezL A fake L{MethodicalMachine}. Instantiate it with a L{FakeDigraph} cCs ||_dSrwZ_digraph)rZdigraphr r r ryszFakeMethodicalMachine.__init__cCs|jSrwrrr r r rn"szFakeMethodicalMachine.asDigraphN)r!r"r#r@ryrnr r r r rsrzTwisted is not installed.c@s^eZdZddZddZddZddd Zd d Zd d ZddZ ddZ ddZ ddZ dS)VisualizeToolTestscCs2t|_t|j|_d|_dg|_g|_d|_dS)Nz tool-testZignoredz fake.fqpn)rvdigraphRecorderr| fakeDigraph fakeProgname fakeSysPathcollectedOutputfakeFQPNrr r r r0+s  zVisualizeToolTests.setUpcGs|jd|dS)N )rr~rm)rargsr r r collectPrints4sz VisualizeToolTests.collectPrintsccs|t|jfVdSrw)rr)rZfqpnr r r fakeFindMachines7sz#VisualizeToolTests.fakeFindMachinesNcCs>ddlm}||p|j|p |jg|p(|j|p0|j|p8|jdS)Nr)tool)Z _prognameZ_argvZ_syspathZ _findMachinesZ_print)r/rrrrrr)rZprognameargvZsyspathZ findMachinesprintrr r r r:s  zVisualizeToolTests.toolcCs&|j|jgd||jdddS)z L{tool} adds '' to sys.path to ensure L{automat._discover.findMachines} searches the current directory. rrriN)rrr:rrr r r test_checksCurrentDirectoryHsz.VisualizeToolTests.test_checksCurrentDirectorycCs@|j|jdgd||j|j|jdgd||jdS)z6 Passing -q/--quiet hides all output. z--quietrz-qN)rr assertFalserrr r r test_quietHidesOutputQs 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-irircss|]}d|kVqdSimageNr rGliner r r dsz6VisualizeToolTests.test_onlySaveDot..r{}.dotfilenameN) rrxrrrranyr:r]r{formatrzrargrr r r test_onlySaveDotZs    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-drircss|]}d|kVqdSrNr rr r r rxsz8VisualizeToolTests.test_saveOnlyImage..rrrcleanupN) rrxrrrrrr:r]rzr assertTruer{rr r r test_saveOnlyImagens    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|kVqdSrr rr r r rszQVisualizeToolTests.test_saveDotAndImagesInDifferentDirectories..css|]}d|kVqdSrr rr r r rsr directoryrN) rrrrrr:r]rrzr{)rZimageDirectoryZ dotDirectory renderCallZsaveCallr r r +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 dotNr rr r r rszJVisualizeToolTests.test_saveDotAndImagesInSameDirectory..rrrN) rrrrrr:r]rrzrr{)rrrr r r $test_saveDotAndImagesInSameDirectorys   z7VisualizeToolTests.test_saveDotAndImagesInSameDirectory)NNNNN) r!r"r#r0rrrrrrrrrr r r r r&s    r) __future__rrWrrZunittestrrrBZ _methodicalrZ test_discoverrrrr,r-sr*rArFrMrQrgrtrvr|rrr r r r s<      ) N