U ucT@sNddlZddlmZddlZddlmZmZddlmZGdddej Z dS)N) replace_loop)patchMock)ApplicationRunnerc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)TestApplicationRunnerzA Test the autobahn.asyncio.wamp.ApplicationRunner class. cOsBz |jWntk r$|j}YnX|j}|||f||dS)N)assertRaisesRegexAttributeErrorassertRaisesRegexp)self exceptionerrorargskwfrqc:\users\psant\downloads\hotstarbackend\school\venv38\Lib\site-packages\autobahn\asyncio\test\test_wamp_runner.py_assertRaisesRegex(s   z(TestApplicationRunner._assertRaisesRegexc Cs~ttj}tjtd|dNtttfd|_i}tdd|d}|d|||j j ddW5QRXW5QRXd S) z Ensure that loop.create_connection is called with the exact SSL context object that is passed (as ssl) to the __init__ method of ApplicationRunner. get_event_loopZ return_valuews://127.0.0.1:8080/wsrealmssl_unused_rN rrrobjectasynciorun_until_completerrunassertIscreate_connection call_args)r looprrunnerrrrtest_explicit_SSLContext1s  z.TestApplicationRunner.test_explicit_SSLContextc Csvttb}tjtd|dFtttfd|_tdd}|d|d|j j ddW5QRXW5QRXd S) z Ensure that loop.create_connection is called with ssl=False if no ssl argument is passed to the __init__ method of ApplicationRunner and the websocket URL starts with "ws:". rrrrrFrrNrr r#r$rrr test_omitted_SSLContext_insecure@s    z6TestApplicationRunner.test_omitted_SSLContext_insecurec Csxttd}tjtd|dHtttfd|_tdd}||j| d|j j ddW5QRXW5QRXdS) z Ensure that loop.create_connection is called with ssl=True if no ssl argument is passed to the __init__ method of ApplicationRunner and the websocket URL starts with "wss:". rrzwss://127.0.0.1:8080/wssrTrrN) rrrrrrrrfailr r!r"r&rrrtest_omitted_SSLContext_secureMs    z4TestApplicationRunner.test_omitted_SSLContext_securec CsRtt>}tttfd|_tdddd}d}|t||jdW5QRXdS) zw ApplicationRunner must raise an exception if given an ssl value of True but only a "ws:" URL. rws://127.0.0.1:8080/wssrTr^ssl argument value passed to ApplicationRunner conflicts with the "ws:" prefix of the url argument\. Did you mean to use "wss:"\?$rN)rrrrr Exceptionr)r r#r$r rrr"test_conflict_SSL_True_with_ws_urlZs z8TestApplicationRunner.test_conflict_SSL_True_with_ws_urlc Csddl}z |jWntk r*d}Yn X|}tt>}tttfd|_tdd|d}d}|t||j d W5QRXdS) z ApplicationRunner must raise an exception if given an ssl value that is an instance of SSLContext, but only a "ws:" URL. rNTrr*rrr+r) rcreate_default_contextrrrrrrr,r)r rcontextr#r$r rrr$test_conflict_SSLContext_with_ws_urlhs   z:TestApplicationRunner.test_conflict_SSLContext_with_ws_urlN) __name__ __module__ __qualname____doc__rr%r'r)r-r0rrrrr$s   r) unittestZtxaio.testutilrrZ unittest.mockrrZautobahn.asyncio.wamprTestCaserrrrrs