o >h'9@sddlmZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z mZddlmZdd lmZd d lmZd d ZddZddZGdddeZGdddZd3ddZe 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)4) annotationsN) dataclass)Protocol)TestCaseskipIf)TypeMachineBuilderpep614)MethodicalMachine) TypeMachine)isTwistedInstalledcCs$ztdWdStyYdSw)z2 Is the graphviz Python module installed? graphvizFT) __import__ ImportErrorrr|/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/automat/_test/test_visualize.pyisGraphvizModuleInstalleds   rc CsBt\}}t|ztjd|dd Wt|St|w)z+ Are the graphviz tools installed? dotT)stdinshell)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)initialcSdS)z initial stateNrselfrrrbegin0z)sampleMachine..SampleObject.begincSr )z end stateNrr!rrrend4r$z'sampleMachine..SampleObject.endcSr )z sample inputNrr!rrrgo8r$z&sampleMachine..SampleObject.gocSr )z sample outputNrr!rrrout<r$z'sampleMachine..SampleObject.outN) __name__ __module__ __qualname__stater#r%inputr&outputr'uponrmmrr SampleObject/s     r1)r objectr&)r1sorr/r sampleMachine)s r4c@seZdZdddZdS)SamplereturnNonecCdSNrr!rrrr&Hr$z Sample.goNr6r7)r(r)r*r&rrrrr5Gsr5c@s eZdZdS)CoreN)r(r)r*rrrrr;Is r;r6TypeMachine[Sample, Core]cCsttt}|d}ddd }|d |}|d }|tj|d |tj|d t|tj|ddd}| S)zA Create a sample L{TypeMachine} with some sample states. r#protor5corer;r6intcSr )Nr)r=r>rrrbuilditSz"sampleTypeMachine..builditdatar%Nsampler7cSr8r9r)rDr>rrrr'[rBzsampleTypeMachine..out)r=r5r>r;r6r?)rDr5r>r;r6r7) rr5r;r+r.r&toreturnsrbuild)builderr#rArCr%r'rrrsampleTypeMachineLs      rIz!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) _visualizerK)r"rKrrrsetUphs  zElementMakerTests.setUpc C"d}|||jddddddS)zK L{elementMaker} orders HTML attributes lexicographically. z
div213bacN assertEqualrKr"expectedrrrtest_sortsAttrsmsz!ElementMakerTests.test_sortsAttrsc CrN)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-
rOz a " quoter za stringrSNrWrYrrrtest_quotesAttrstsz"ElementMakerTests.test_quotesAttrscCsd}|||ddS)zN L{elementMaker} should render an element with no attributes. z
rONrWrYrrr test_noAttrsszElementMakerTests.test_noAttrsN)r(r)r*__doc__rMr[r\r]rrrrrJas  rJc@s*eZdZUdZded<ded<ded<dS) HTMLElementz2Holds an HTML element, as created by elementMaker.strnamezlist[HTMLElement]childrenzdict[str, str] attributesN)r(r)r*r^__annotations__rrrrr_s  r_cs.|r|gSt|r gSfdd|jDS)zk Recursively collect all elements in an L{HTMLElement} tree that match the optional predicate. cs g|] }t|D]}|q qSr) findElements).0childresult predicaterr sz findElements..)isLeafrb)elementrjrrirres recCs t|t S)z2 This HTML element is actually leaf node. ) isinstancer_rmrrrrls rlc@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)rarbrc)r_)r"rarbrcrrrfakeElementMakersz TableMakerTests.fakeElementMakercCs.ddlm}d|_d|_tj||jd|_dS)Nr ) tableMaker input labelzthe port)_E)rLrr inputLabelport functoolspartialrq)r"rrrrrrMs zTableMakerTests.setUpcsfdd}gdgfD];}jj|jd}t|jd|jd}t||}t|d|djdt|t jgq 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. cst| o |jdjkS)Nrv)rlrcgetrvror!rrhasPortsz3TableMakerTests.test_inputLabelRow..hasPortzan output labelrvrr tdN) rrrurv assertGreaterlenrbrerXrarl)r"rz outputLabelstable inputLabelRowportCandidatesrr!rtest_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. rsrr{r colspanN)rrrvrXr~rb assertNotInrc)r"rrrrrtest_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{r cSs$t| o|jdko|jddkS)Nr|rrP)rlrarcryrorrrhasCorrectColspans z@TableMakerTests.test_withOutputLabels..hasCorrectColspanr rrN)rrrurvrXr~rbrerl)r"rinputRow outputRowrrrrtest_withOutputLabelss  z%TableMakerTests.test_withOutputLabelsN) r(r)r*r^rqrMrrrrrrrrps rpz!Graphviz tools are not installed.c@seZdZdZdddZdS)IntegrationTestsz[ Tests which make sure Graphviz can understand the output produced by Automat. r6r7cCsNt}d|d}tjdtjtjd}||\}}||j ddS)z8 C{graphviz} emits valid graphviz data. zutf-8r)rstdoutrN) r4 asDigraphjoinencoderPopenPIPE communicaterX returncode)r"digraphtextpr'errrrrtest_validGraphvizs z#IntegrationTests.test_validGraphvizNr:)r(r)r*r^rrrrrrsrc@ 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)rr4rassertInr"gvoutrrrtest_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)rrIrrrrrr test_containsTypeMachineFeaturess     z+SpotChecks.test_containsTypeMachineFeaturesN)r(r)r*r^rrrrrrr s rc@r)RecordsDigraphActionsz/ Records calls made to L{FakeDigraph}. cCs |dSr9)resetr!rrr__init__/s zRecordsDigraphActions.__init__cCsg|_g|_dSr9) renderCalls saveCallsr!rrrr2s zRecordsDigraphActions.resetN)r(r)r*r^rrrrrrr* rc@s(eZdZdZddZddZddZdS) FakeDigraphzZ A fake L{graphviz.Digraph}. Instantiate it with a L{RecordsDigraphActions}. cC ||_dSr9) _recorder)r"recorderrrrr= zFakeDigraph.__init__cK|jj|dSr9)rrappendr"kwargsrrrrender@zFakeDigraph.rendercKrr9)rrrrrrrsaveCrzFakeDigraph.saveN)r(r)r*r^rrrrrrrr7s  rc@r)FakeMethodicalMachinezL A fake L{MethodicalMachine}. Instantiate it with a L{FakeDigraph} cCrr9_digraph)r"rrrrrLrzFakeMethodicalMachine.__init__cCs|jSr9rr!rrrrOszFakeMethodicalMachine.asDigraphN)r(r)r*r^rrrrrrrGrrc@s`eZdZddZddZddZ ddd Zd d Zd d ZddZ ddZ ddZ ddZ dS)VisualizeToolTestscCs2t|_t|j|_d|_dg|_g|_d|_dS)Nz tool-testignoredz fake.fqpn)rdigraphRecorderr fakeDigraph fakeProgname fakeSysPathcollectedOutputfakeFQPNr!rrrrMWs   zVisualizeToolTests.setUpcGs|jd|dS)N )rrr)r"argsrrr collectPrints`sz VisualizeToolTests.collectPrintsccs|t|jfVdSr9)rr)r"fqpnrrrfakeFindMachinescsz#VisualizeToolTests.fakeFindMachinesNcCs>ddlm}||p |j|p|jg|p|j|p|j|p|jdS)Nr )tool) _progname_argv_syspath _findMachines_print)rLrrrrrr)r"prognameargvsyspath findMachinesprintrrrrrfs  zVisualizeToolTests.toolcCs&|j|jgd||jdddS)z L{tool} adds '' to sys.path to ensure L{automat._discover.findMachines} searches the current directory. rrrN)rrrXrr!rrrtest_checksCurrentDirectoryssz.VisualizeToolTests.test_checksCurrentDirectorycCs@|j|jdgd||j|j|jdgd||jdS)z6 Passing -q/--quiet hides all output. z--quietrz-qN)rr assertFalserr!rrrtest_quietHidesOutput|s z(VisualizeToolTests.test_quietHidesOutputcCsdD]C}|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-irrcs|]}d|vVqdSimageNrrflinerrr z6VisualizeToolTests.test_onlySaveDot..r {}.dotfilenameN) rrrrrranyrXr~rformatrr"argrrrrtest_onlySaveDots  z#VisualizeToolTests.test_onlySaveDotcCsdD]J}|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-drrcsrrNrrrrrrrz8VisualizeToolTests.test_saveOnlyImage..r rrcleanupN) rrrrrrrrXr~rr 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. rrrrrcsrrrrrrrrrzQVisualizeToolTests.test_saveDotAndImagesInDifferentDirectories..csrrrrrrrrrr directoryrN) rrrrrrXr~rrr)r"imageDirectory dotDirectory renderCallsaveCallrrr+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. imagesAndDotrrrcsr)z image and dotNrrrrrrrzJVisualizeToolTests.test_saveDotAndImagesInSameDirectory..r rrN) rrrrrrXr~rrrr)r"rrrrr$test_saveDotAndImagesInSameDirectorys  z7VisualizeToolTests.test_saveDotAndImagesInSameDirectory)NNNNN) r(r)r*rMrrrrrrrrrrrrrrSs      r)r6r<)' __future__rrwrr dataclassesrtypingrunittestrrautomatrr _methodicalr _typedr test_discoverr rrr4r5r;rIrJr2r_rerlrprrrrrrrrrrsR           $   L