o iT@sNddlZddlmZddlZddlmZmZddlmZGdddej Z dS)N) replace_loop)patchMock)ApplicationRunnerc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)TestApplicationRunnerzA Test the autobahn.asyncio.wamp.ApplicationRunner class. cOsFz|jWn ty|j}Ynw|j}|||g|Ri|dS)N)assertRaisesRegexAttributeErrorZassertRaisesRegexp)self exceptionerrorargskwfrlC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\autobahn/asyncio/test/test_wamp_runner.py_assertRaisesRegex(s   z(TestApplicationRunner._assertRaisesRegexc CsttM}tjtd|d,tttfd|_i}tdd|d}|d|||j j ddWd n1s=wYWd d SWd d S1sUwYd 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_loop 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 CsttI}tjtd|d(tttfd|_tdd}|d|d|j j ddWd n1s9wYWd d SWd d S1sQwYd 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 CsttJ}tjtd|d)tttfd|_tdd}||j| d|j j ddWdn1s:wYWddSWddS1sRwYdS) 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) rrrrrrrrZfailr r!r"r&rrrtest_omitted_SSLContext_secureMs   "z4TestApplicationRunner.test_omitted_SSLContext_securecCshtt%}tttfd|_tdddd}d}|t||jdWddS1s-wYdS) 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_urlcCsddl}z|jWn tyd}Ynw|}tt%}tttfd|_tdd|d}d}|t||j d WddS1sFwYdS) 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,r/rrrrr$s   r) ZunittestZtxaio.testutilrrZ unittest.mockrrZautobahn.asyncio.wamprZTestCaserrrrrs