U ‰ª/eàã@sädZddlZddlZddlZddlmZddlmZddlm Z ddl Z ddl Z ddl m Z ddlmZddlmZmZmZmZGd d „d ƒZGd d „d eƒZGd d„de ƒZGdd„dejƒZdd„Zdd„Zedkràe ¡dS)z* Tests for zmq shell / display publisher. éN)ÚQueue)ÚThread)Ú MagicMock)ÚSession)ÚInt)ÚInteractiveShellÚ KernelMagicsÚZMQDisplayPublisherÚZMQInteractiveShellc@seZdZdZdZdd„ZdS)ÚNoReturnDisplayHookz¢ A dummy DisplayHook which allows us to monitor the number of times an object is called, but which does *not* return a message when it is called. rcCs|jd7_dS)Né)Ú call_count©ÚselfÚobj©rúB/tmp/pip-unpacked-wheel-g43y689f/ipykernel/tests/test_zmq_shell.pyÚ__call__#szNoReturnDisplayHook.__call__N)Ú__name__Ú __module__Ú __qualname__Ú__doc__r rrrrrr sr cs eZdZdZ‡fdd„Z‡ZS)ÚReturnDisplayHookz’ A dummy DisplayHook with the same counting ability as its base class, but which also returns the same message when it is called. cstƒ |¡|S©N)Úsuperrr©Ú __class__rrr.s zReturnDisplayHook.__call__)rrrrrÚ __classcell__rrrrr'srcs(eZdZdZedƒZ‡fdd„Z‡ZS)ÚCounterSessionz This is a simple subclass to allow us to count the calls made to the session object by the display publisher. rcs |jd7_tƒj||ŽdS)zu A trivial override to just augment the existing call with an increment to the send counter. r N)Ú send_countrÚsend)rÚargsÚkwargsrrrr <szCounterSession.send)rrrrrrr rrrrrr3src@sPeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dS)ÚZMQDisplayPublisherTestsz6 Tests the ZMQDisplayPublisher in zmqshell.py cCs8t ¡|_|j tj¡|_tƒ|_t|j|jd|_dS)N)ÚsessionÚ pub_socket) ÚzmqÚContextÚcontextÚsocketÚPUBrr$r Údisp_pub©rrrrÚsetUpJs zZMQDisplayPublisherTests.setUpcCs"|j ¡|j ¡|j ¡dS)z¿ We need to close the socket in order to proceed with the tests. TODO - There is still an open file handler to '/dev/null', presumably created by zmq. N)r+Z clear_outputr)Úcloser(Ztermr,rrrÚtearDownQs  z!ZMQDisplayPublisherTests.tearDowncCs(|jj|jkst‚|jj|jks$t‚dS)z Since there's no explicit constructor, here we confirm that keyword args get assigned correctly, and override the defaults. N)r+r$ÚAssertionErrorr%r)r,rrrÚtest_display_publisher_creation\sz8ZMQDisplayPublisherTests.test_display_publisher_creationcsxˆjjgkst‚dd„}ˆj |¡ˆjj|gks6t‚tƒ‰‡‡fdd„}t|d}| ¡ˆjdd}|gkstt‚dS) z„ Confirms that the thread_local attribute is correctly initialised with an empty list for the display hooks cSs|Srr)ÚmsgrrrÚhooklsz>ZMQDisplayPublisherTests.test_thread_local_hooks..hookcsˆ ˆjj¡dSr)Úputr+Ú_hooksr©ÚqrrrÚset_thread_hookstszJZMQDisplayPublisherTests.test_thread_local_hooks..set_thread_hooks)Útargeté )ÚtimeoutN)r+r5r0Ú register_hookrrÚstartÚget)rr3r8ÚtZ thread_hooksrr6rÚtest_thread_local_hookses   z0ZMQDisplayPublisherTests.test_thread_local_hookscCs:tdd}|jjdkst‚|j |¡|jjdks6t‚dS)zc Publish should prepare the message and eventually call `send` by default. r ©ÚarN)Údictr$rr0r+Úpublish)rÚdatarrrÚ test_publish|s  z%ZMQDisplayPublisherTests.test_publishcCshtdd}tƒ}|j |¡|jdks*t‚|jjdks:t‚|j |¡|jdksTt‚|jjdksdt‚dS)zé If a hook is installed, and on calling the object it does *not* return a message, then we assume that the message has been consumed, and should not be processed (`sent`) in the normal manner. r rArN) rCr r+r<r r0r$rrD©rrEr3rrrÚtest_display_hook_halts_send†s   z5ZMQDisplayPublisherTests.test_display_hook_halts_sendcCshtdd}tƒ}|j |¡|jdks*t‚|jjdks:t‚|j |¡|jdksTt‚|jjdksdt‚dS)zã If a hook is installed and on calling the object it returns a new message, then we assume that this is just a message transformation, and the message should be sent in the usual manner. r rArN) rCrr+r<r r0r$rrDrGrrrÚ#test_display_hook_return_calls_send™s   zs(