a ze@sFddlZddlZddlZddlZGdddejZedkrBedS)Nc@s<eZdZddZddZddZddZd d Zd d Zd S) TestResponsecCs0ttjtj|_|jd|_ddd|_dS)Nrbzwww.python.orgclose)ZHost Connection)socketAF_INET SOCK_STREAMsockmakefilefp test_headersselfrM/opt/bitninja-python-dojo/embedded/lib/python3.9/test/test_urllib_response.pysetUp s zTestResponse.setUpcsZtj|j|tjfdd}||jj|| |jj| t |dS)Ncs(}Wdn1s0YdSNr)Zspamaddbaserrfsz!TestResponse.test_with..f) urllibresponserr ZassertIsInstancetempfile_TemporaryFileWrapperZ assertFalseclosed assertTrueZ assertRaises ValueError)rrrrr test_withs zTestResponse.test_withcsDdfdd}tj|j|}|||jj|dS)NFcsddS)NTrrZclosehook_calledrr closehook sz1TestResponse.test_addclosehook..closehook)rrZ addclosehookr rrr)rrrrrtest_addclosehooks  zTestResponse.test_addclosehookcCs8tj|j|j}|||j||j|jdSr)rrZaddinfor r assertEqualinfoheaders)rr"rrr test_addinfo*szTestResponse.test_addinfocCsd}d}tj|j|j||}|||j||||||||j |j||j |||j |dS)Nzhttp://www.python.org) rrZ addinfourlr r r!r"geturlZgetcoder#urlstatus)rr'codeZinfourlrrrtest_addinfourl/szTestResponse.test_addinfourlcCs|jdSr)r rr rrrtearDown;szTestResponse.tearDownN) __name__ __module__ __qualname__rrr r$r*r+rrrrrs    r__main__) rrZurllib.responserZunittestZTestCaserr,mainrrrrs 6