\dldZddlZddlmZddlmZGddZdZdZGd d eZ dS) zt Tests for miscellaneous behaviors of the top-level L{twisted} package (ie, for the code in C{twisted/__init__.py}. N) ModuleType)TestCasec*eZdZdZdZdZdZdZdS)SetAsideModulez L{SetAsideModule} is a context manager for temporarily removing a module from C{sys.modules}. @ivar name: The name of the module to remove. c||_dSN)name)selfr s 9lib/python3.11/site-packages/twisted/test/test_twisted.py__init__zSetAsideModule.__init__s  cfdttjD}|D]}tj|=|S)z Find the given module and all of its hierarchically inferior modules in C{sys.modules}, remove them from it, and return whatever was found. cli|]0\}}|jks|jdz-||1S).)r startswith).0 moduleNamemoduler s r z,SetAsideModule._unimport.."sO   $Vdi'':+@+@S+Q+Q' '''r )listsysmodulesitems)r r rs` r _unimportzSetAsideModule._unimportsd     (,S[->->-@-@(A(A     " "D D!!r cD||j|_dSr)rr r)r s r __enter__zSetAsideModule.__enter__+s~~di00 r c||jtj|jdSr)rr rrupdate)r excTypeexcValue tracebacks r __exit__zSetAsideModule.__exit__.s4 ty!!! 4<(((((r N)__name__ __module__ __qualname____doc__r rrr"r r rrsZ   111)))))r rcji}td||tj|dS)a Take a mapping defining a package and turn it into real C{ModuleType} instances in C{sys.modules}. Consider these example:: a = {"foo": "bar"} b = {"twisted": {"__version__": "42.6"}} c = {"twisted": {"plugin": {"getPlugins": stub}}} C{_install(a)} will place an item into C{sys.modules} with C{"foo"} as the key and C{"bar" as the value. C{_install(b)} will place an item into C{sys.modules} with C{"twisted"} as the key. The value will be a new module object. The module will have a C{"__version__"} attribute with C{"42.6"} as the value. C{_install(c)} will place an item into C{sys.modules} with C{"twisted"} as the key. The value will be a new module object with a C{"plugin"} attribute. An item will also be placed into C{sys.modules} with the key C{"twisted.plugin"} which refers to that module object. That module will have an attribute C{"getPlugins"} with a value of C{stub}. @param modules: A mapping from names to definitions of modules. The names are native strings like C{"twisted"} or C{"unittest"}. Values may be arbitrary objects. Any value which is not a dictionary will be added to C{sys.modules} unmodified. Any dictionary value indicates the value is a new module and its items define the attributes of that module. The definition of this structure is recursive, so a value in the dictionary may be a dictionary to trigger another level of processing. @return: L{None} N) _makePackagesrrr)rresults r _installr+3s8DF$(((Kvr ci}t|D]\}}|Yt|tr>t |}|jt||||||<Z|||<`t|trYt |jdz|z}|jt||||||jdz|z<|||<|||<|S)a Construct module objects (for either modules or packages). @param parent: L{None} or a module object which is the Python package containing all of the modules being created by this function call. Its name will be prepended to the name of all created modules. @param attributes: A mapping giving the attributes of the particular module object this call is creating. @param result: A mapping which is populated with all created module names. This is suitable for use in updating C{sys.modules}. @return: A mapping of all of the attributes created by this call. This is suitable for populating the dictionary of C{parent}. @see: L{_install}. Nr) rr isinstancedictr__dict__rr)r#)parent attributesr*attrsr valuers r r)r)Zs & Ej..0011$$ u >%&& %#D))&&}VUF'K'KLLL%t $t %&& $#FOc$9D$@AA&&}VUF'K'KLLL7=v,t34$d #d Lr c$eZdZdZdZdZdZdS)MakePackagesTestszh Tests for L{_makePackages}, a helper for populating C{sys.modules} with fictional modules. ci}tdtd|||tddS)z A non-C{dict} value in the attributes dictionary passed to L{_makePackages} is preserved unchanged in the return value. Nreactor)r7)r)r. assertEqualr rs r test_nonModulez MakePackagesTests.test_nonModulesM dD333W=== $y"9"9"9:::::r c`i}tdttd|||t||dt|d|dj|d|djdS)z A C{dict} value in the attributes dictionary passed to L{_makePackages} is turned into a L{ModuleType} instance with attributes populated from the items of that C{dict} value. N123versiontwistedr@)r)r.assertIsInstancerr8r#r>r9s r test_moduleWithAttributez*MakePackagesTests.test_moduleWithAttributes dDe)<)<)<===wGGG gt,,, gi0*=== GI$6$?@@@  2 :;;;;;r c i}tdtttd|||t||dt|d|dj||djt|d|djj|d|djjdS)z Processing of the attributes dictionary is recursive, so a C{dict} value it contains may itself contain a C{dict} value to the same effect. N321r=)webr?r@z twisted.web)r)r.rArr8r#rEr>r9s r test_packageWithModulez(MakePackagesTests.test_packageWithModules dD$u2E2E2E)F)F)FGGGQQQ gt,,, gi0*=== GI$6$?@@@ gi04jAAA  (:(>(GHHH  2 6 >?????r N)r#r$r%r&r:rBrFr'r r r5r5sP ;;; < < < @ @ @ @ @r r5) r&rtypesrtwisted.trial.unittestrrr+r)r5r'r r rIs  ++++++ ))))))))B$$$N$$$N(@(@(@(@(@(@(@(@(@(@r