3 Ndq@s$dZddlZddlZddlZddlZddlZddlZddlZddlZddl j Z ddl m Z m Z mZmZddlmZmZedk rddlmZmZmZddlmZddlmZddlmZdd lmZdd lm Z m!Z!m"Z"d Z#Gd d d e Z$Gddde Z%Gddde Z&eGdddej'Z(dS)z! Tests for the trait documenter. N)Bool HasTraitsIntProperty)sphinxrequires_sphinx)ClassDocumenter INSTANCEATTROptions)DocumenterBridge)path) SphinxTestApp)LoggingReporter)_get_definition_tokenstrait_definitionTraitDocumenterzextensions = ['sphinx.ext.autodoc'] # The suffix of source filenames. source_suffix = '.rst' autodoc_mock_imports = [ 'dummy' ] c@seZdZdZedddZdS) MyTestClassz Class-level docstring. *zd First line The answer to Life, the Universe, and Everything. )descN)__name__ __module__ __qualname____doc__rbarrr[/var/www/html/virt/lib64/python3.6/site-packages/traits/util/tests/test_trait_documenter.pyr6src@seZdZeeddZdS)Fakeu ミスあり)labelN)rrrrrtest_attributerrrrrEsrc@s*eZdZdZeZeZdZeddZ dS) FindTheTraitszB Class for testing the can_document_member functionality. icCsdS)z6 I'm a regular property, not a trait. Nr)selfrrr not_a_traitYszFindTheTraits.not_a_traitN) rrrrran_int another_int magic_numberpropertyr!rrrrrKs rc@sdeZdZdZddZddZddZdd Zd d Zd d Z ddZ e j ddZ e j ddZdS)TestTraitDocumenterz! Tests for the trait documenter. cCs(d|_tj|j}tj|j}||_dS)Nzp depth_interval = Property(Tuple(Float, Float), depends_on="_depth_interval") )sourceioStringIOtokenizegenerate_tokensreadlinetokens)r string_ior-rrrsetUpds  zTestTraitDocumenter.setUpcCsFtjd}tj|}tj|j}t|}tj|}|j |j |dS)Nz depth_interval = Property(Tuple(Float, Float), depends_on="_depth_interval") ) textwrapdedentr(r)r*r+r,r untokenize assertEqualrstrip)r srcr.r-Zdefinition_tokensstringrrrtest_get_definition_tokensms   z.TestTraitDocumenter.test_get_definition_tokensc Cs\tj}t|dd}d|_t|_tjd|jdWdQRX|jt |j j j j ddS)Ntestz rzFtraits.util.trait_documenter.ClassLevelDocumenter.add_directive_header)mockZ MagicMockrZ object_namerparentpatchZadd_directive_headerr3len directiveresultappendZ mock_calls)r Zmocked_directive documenterrrr test_add_line~s z!TestTraitDocumenter.test_add_linec Csx|j&}t|td}|jdd|j}WdQRXx&|D]}|jjdr:Pq:W|jd|jd||j d|dS)Nz.MyTestClass.barT)Z all_membersz :annotation:z+Didn't find the expected trait :annotation:z First line ) create_directiverrgenerater@lstrip startswithZfailZassertInZ assertNotIn)r r?rBr@itemrrrtest_abbreviated_annotationss      z0TestTraitDocumenter.test_abbreviated_annotationscCsttdd}|j|ddS)Nr)cls trait_nameu$Property(Bool, label="ミスあり"))rrr3)r Z definitionrrr test_successful_trait_definitions z4TestTraitDocumenter.test_successful_trait_definitionc Cs&|jtttddWdQRXdS)Nr!)rKrL) assertRaises ValueErrorrr)r rrrtest_failed_trait_definitions z0TestTraitDocumenter.test_failed_trait_definitionc Cs|j|}t|td}|j|j|jtjtdd||jtjtdd||j tjtdd||j tjtdd|WdQRXdS)Nz.FindTheTraitsr"Tr#r$r!) rErr parse_nameZ import_object assertTruerZcan_document_memberr Z assertFalse)r r?Zclass_documenterrrrtest_can_document_members"  z,TestTraitDocumenter.test_can_document_memberccs|j}tjj|d}t|ddd}|jtWdQRXtt|d}||jj _ d|jj j d<i}t j }d |jj_||d <t|j td td f|VWdQRXdS) aW Helper function to create a a "directive" suitable for instantiating the TraitDocumenter with, along with resources to support that directive, and clean up the resources afterwards. Returns ------- contextmanager A context manager that returns a DocumenterBridge instance. zconf.pywzutf-8)encodingN)srcdirdummyZdocnamestater9r:)tmpdirosr joinopenwriteCONF_PYr ZbuilderenvappZ temp_datar;ZMockdocumentsettingsZ tab_widthr rr )r rZZ conf_filefrakwdsrYrrrrEs   z$TestTraitDocumenter.create_directivec cs$tj}z |VWdtj|XdS)z Helper function to create a temporary directory. Returns ------- contextmanager Context manager that returns the path to a temporary directory. N)tempfilemkdtempshutilrmtree)r rZrrrrZs  zTestTraitDocumenter.tmpdirN)rrrrr/r7rCrJrMrPrS contextlibcontextmanagerrErZrrrrr&`s "r&))rrjr(r[rhrfr0r*ZunittestZ unittest.mockr;Z traits.apirrrrZ$traits.testing.optional_dependenciesrrZsphinx.ext.autodocrr r Zsphinx.ext.autodoc.directiver Zsphinx.testing.pathr Zsphinx.testing.utilr Zsphinx.util.docutilsrZtraits.util.trait_documenterrrrr_rrrZTestCaser&rrrr s0