~b6 ddlmZddlZddlZddlZddlmZmZddlZddl m Z ddl m Z dZ d Zd Zee d ee d Gd deZej"GddeZdZdZee d ee d GddeZee d ee dee d GddeZee d ee d GddeZGddeZGddeZGddeZee d ee dee d Gd d!eZy)")print_functionN)TestCaseskipIf)MethodicalMachine)isTwistedInstalledc: tdy#t$rYywxYw)z2 Is the graphviz Python module installed? graphvizTF) __import__ ImportError>/usr/lib/python3/dist-packages/automat/_test/test_visualize.pyisGraphvizModuleInstalledrs(: s  ctj\}}tj| tjd|d tj|S#tj|wxYw)z+ Are the graphviz tools installed? dotT)stdinshell)ospipeclose subprocesscall)rws risGraphvizInstalledrsM 779DAqHHQK??5>>   s AA3cltGfddt}|}|jS)zG Create a sample L{MethodicalMachine} with some sample states. ceZdZWjddZWjdZWj dZWjdZ ejeee gy)#sampleMachine..SampleObjectT)initialcy)z initial stateNrselfs rbeginz)sampleMachine..SampleObject.begin.rcy)z end stateNrr#s rendz'sampleMachine..SampleObject.end1r&rcy)z sample inputNrr#s rgoz&sampleMachine..SampleObject.go4r&rcy)z sample outputNrr#s routz'sampleMachine..SampleObject.out7r&rN) __name__ __module__ __qualname__stater%r(inputr*outputr,upon)mmsr SampleObjectr -sz $                 2sSE"rr5)robjectr*)r5sor4s @r sampleMachiner8(s0  B #v # BEEG Irz!Graphviz module is not installed.zTwisted is not installed.c(eZdZdZdZdZdZdZy)ElementMakerTestszL L{elementMaker} generates HTML representing the specified element. cddlm}||_y)Nr) elementMaker) _visualizer<)r$r<s rsetUpzElementMakerTests.setUpGs -(rc Rd}|j||jddddy)zK L{elementMaker} orders HTML attributes lexicographically. z
div213bacN assertEqualr<r$expecteds rtest_sortsAttrsz!ElementMakerTests.test_sortsAttrsKs74 **5-0-0-0+2 3rc Rd}|j||jddddy)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-
r@z a " quoterza stringrDNrHrJs rtest_quotesAttrsz"ElementMakerTests.test_quotesAttrsVs8 D **5-8-.-7+9 :rcJd}|j||jdy)zN L{elementMaker} should render an element with no attributes. z
r@NrHrJs r test_noAttrszElementMakerTests.test_noAttrscs$# 4#4#4U#;rLrNrPrrrr:r:@s) 3 :=rr:cveZdZdZej Zej Zej Zy) HTMLElementz2Holds an HTML element, as created by elementMaker.N) r-r.r/rQattribnamechildren attributesrrrrSrSks,< 4779DtwwyHJrrSc||r|gSt|rgS|jDcgc]}t||D]}|c}}Scc}}w)zk Recursively collect all elements in an L{HTMLElement} tree that match the optional predicate. )isLeafrW findElements)element predicatechildresults rr[r[ssa y  !)) :&ui8 :  :F :: :sAc$t|t S)z2 This HTML element is actually leaf node. ) isinstancerSr\s rrZrZs';/ //rc.eZdZdZdZdZdZdZdZy)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}. ct|||S)N)rVrWrX)rS)r$rVrWrXs rfakeElementMakerz TableMakerTests.fakeElementMakersxJOOrcxddlm}d|_d|_t j ||j |_y)Nr) tableMaker input labelzthe port)_E)r=rh inputLabelport functoolspartialrf)r$rhs rr>zTableMakerTests.setUps1+' #++J/3/D/DFrcfd}gdgfD]}jj|j}jt |j d|j d}t ||}jt |dj|djdjt |tjgy)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. cpt| xr(|jjdjk(S)Nrl)rZrXgetrl)r\r$s rhasPortz3TableMakerTests.test_inputLabelRow..hasPorts6w'D**..v6$))C Erzan output labelrlrrtdN) rhrkrl assertGreaterlenrWr[rIrVrZ)r$rr outputLabelstable inputLabelRowportCandidatess` rtest_inputLabelRowz"TableMakerTests.test_inputLabelRows E "3!45 0LOODOO\)-$4E   s5>>2A 6!NN1-M)%9N   S0! 4   ^A.33T :   \-@"oo. 0 0rc|jdd|j}|jt|jd|j\}|j d|j y)z L{tableMaker} does not add a colspan attribute to the input label's cell or a second row if there no output labels. rirrsrcolspanN)rhrlrIrvrW assertNotInrX)r$rxrys rtest_noOutputLabelsz#TableMakerTests.test_noOutputLabelssW  r B U^^,a0 >> M$<$<=rcP|j|jd|j}|jt |j d|j \}}d}|jt t ||d|jt |tddgy) 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 2rsrc|t| xr/|jdk(xr|jjddk(S)Nrtr}rA)rZrVrXrqrbs rhasCorrectColspanz@TableMakerTests.test_withOutputLabels..hasCorrectColspansCw'A ,A**..y9S@ BrrrrN)rhrkrlrIrvrWr[rZ)r$rxinputRow outputRowrs rtest_withOutputLabelsz%TableMakerTests.test_withOutputLabelss 2D%)YY 0 U^^,a0#nn) B \(4EFG  i8;K;K;M NrN) r-r.r/rQrfr>r{rrrrrrdrds#PF00>Nrrdz!Graphviz tools are not installed.ceZdZdZdZy)IntegrationTestsz[ Tests which make sure Graphviz can understand the output produced by Automat. c4tjdtjtj}|jdj t j jd\}}|j|jdy)z8 L{graphviz} emits valid graphviz data. r)rstdoutzutf-8rN) rPopenPIPE communicatejoinr8 asDigraphencoderI returncode)r$pr,errs rtest_validGraphvizz#IntegrationTests.test_validGraphvizsh   U*//$.OO 5==)B)B)D!E"(&/3S q)rN)r-r.r/rQrrrrrrs  *rrceZdZdZdZy) SpotCheckszj Tests to make sure that the output contains salient features of the machine being generated. cdjtj}|jd||jd||jd||jd|y)z The output of L{graphviz} should contain the names of the states, inputs, outputs in the state machine. rr%r(r*r,N)rr8rassertIn)r$gvouts rtest_containsMachineFeaturesz'SpotChecks.test_containsMachineFeaturessU  1134 gu% eU# dE" eU#rN)r-r.r/rQrrrrrrs  $rrceZdZdZdZdZy)RecordsDigraphActionsz/ Records calls made to L{FakeDigraph}. c$|jyN)resetr#s r__init__zRecordsDigraphActions.__init__s  rc g|_g|_yr) renderCalls saveCallsr#s rrzRecordsDigraphActions.reset srN)r-r.r/rQrrrrrrrsrrc"eZdZdZdZdZdZy) FakeDigraphzZ A fake L{graphviz.Digraph}. Instantiate it with a L{RecordsDigraphActions}. c||_yr) _recorder)r$recorders rrzFakeDigraph.__init__s !rc N|jjj|yr)rrappendr$kwargss rrenderzFakeDigraph.renders ""))&1rc N|jjj|yr)rrrrs rsavezFakeDigraph.saves   ''/rN)r-r.r/rQrrrrrrrrs "20rrceZdZdZdZdZy)FakeMethodicalMachinezL A fake L{MethodicalMachine}. Instantiate it with a L{FakeDigraph} c||_yr_digraph)r$digraphs rrzFakeMethodicalMachine.__init__#s  rc|jSrrr#s rrzFakeMethodicalMachine.asDigraph&s }}rN)r-r.r/rQrrrrrrrs rrcTeZdZdZdZdZ d dZdZdZdZ d Z d Z d Z y) VisualizeToolTestsct|_t|j|_d|_dg|_g|_d|_y)Nz tool-testignoredz fake.fqpn)rdigraphRecorderr fakeDigraph fakeProgname fakeSysPathcollectedOutputfakeFQPNr#s rr>zVisualizeToolTests.setUp/sA46&t';';<'%;!# rcX|jjdj|y)N )rrr)r$argss r collectPrintsz VisualizeToolTests.collectPrints8s ##CHHTN3rc#>K|t|jfywr)rr)r$fqpns rfakeFindMachinesz#VisualizeToolTests.fakeFindMachines;s)$*:*:;;;sNcddlm}||xs |j|xs |jg|xs |j|xs |j |xs |j S)Nr)tool) _progname_argv_syspath _findMachines_print)r=rrrrrr)r$prognameargvsyspath findMachinesprintrs rrzVisualizeToolTests.tool>sZ &3$"3"3)4==/0 0 0&?$*?*?.D.. 0 0rc||j|jg|j|jddy)z L{tool} adds '' to sys.path to ensure L{automat._discover.findMachines} searches the current directory. rrrN)rrrIrr#s rtest_checksCurrentDirectoryz.VisualizeToolTests.test_checksCurrentDirectoryLs3   ' ))!,b1rc|j|jdg|j|j|j|jdg|j|jy)z6 Passing -q/--quiet hides all output. z--quietrz-qN)rr assertFalserr#s rtest_quietHidesOutputz(VisualizeToolTests.test_quietHidesOutputUsZ  y1 2 --.  t, - --.rc dD]}|jjg|_|j|j|dg|j t d|jD|jt|jjd|jj\}|jdj|j|d|j |jj y) ze Passing an empty string for --image-directory/-i disables rendering images. )--image-directoryz-irrc3$K|]}d|v ywimageNr.0lines r z6VisualizeToolTests.test_onlySaveDot..hs!C%)")D!Cr{}.dotfilenameN) rrrrrranyrIrvrformatrr$argrs rtest_onlySaveDotz#VisualizeToolTests.test_onlySaveDot^s / ?C  & & (#%D IIDMM33I 4   S!C-1-A-A!CC D   S!5!5!?!?@! D**44GT   X__T]];!*- /   T11== > ?rcHdD]}|jjg|_|j|j|dg|j t d|jD|jt|jjd|jj\}|jdj|j|d|j|d|j |jjy ) zc Passing an empty string for --dot-directory/-d disables saving dot files. )--dot-directoryz-drrc3$K|]}d|v ywrNrrs rrz8VisualizeToolTests.test_saveOnlyImage..|s!C%)"'$!CrrrrcleanupN) rrrrrrrrIrvrr assertTruerrs rtest_saveOnlyImagez%VisualizeToolTests.test_saveOnlyImagers - =C  & & (#%D IIDMM33I 4   S!C-1-A-A!CC D   S!5!5!A!ABA F**66GT   X__T]];!*- / OODO ,   T11;; < =rcd}d}|j|jd|d|g|jtd|jD|jtd|jD|j t |jjd|jj\}|j |d ||j|d |j t |jjd|jj\}|j |d |y ) z Passing different directories to --image-directory and --dot-directory writes images and dot files to those directories. rrrrrc3$K|]}d|v ywrrrs rrzQVisualizeToolTests.test_saveDotAndImagesInDifferentDirectories..s> $$tO>rc3$K|]}d|v ywrrrs rrzQVisualizeToolTests.test_saveDotAndImagesInDifferentDirectories..s> $"TM>rr directoryrN) rrrrrrIrvrrr)r$imageDirectory dotDirectory renderCallsaveCalls r+test_saveDotAndImagesInDifferentDirectoriesz>VisualizeToolTests.test_saveDotAndImagesInDifferentDirectoriess$ !   +^)<9  : >(,(<(<>> ? >(,(<(<>> ? T11==>B,,88  K0.A  9-. T11;;.s> $,t3>rrrrN) rrrrrrIrvrrrr)r$rrs r$test_saveDotAndImagesInSameDirectoryz7VisualizeToolTests.test_saveDotAndImagesInSameDirectorys #   +Y)96  7 >(,(<(<>> ? T11==>B,,88  K0)< I./ T11;;<=r)NNNNN) r-r.r/r>rrrrrrrrrrrrrr*sE $4< 02/?(=*>2>rr) __future__rrmrrunittestrrrT _methodicalr test_discoverr rrr8r:sr6rSr[rZrdrrrrrrrrrrs% % +-  0 % '')LM   "=>&=&=?N&=R& :0 % '')LM   "=>KNhKN?NKN\ % '')LM  !!#FG   "=>*x*?HN*" % '')LM   "=>$$?N$$ F  0& 0 F  % '')LM  !!#FG   "=>E>E>?HNE>r