ó ÊdYN2c@sddlZddljjZddlZddlZddl Zddl Z ddl Z ddl Z ddlZddlmZddlmZddlZdZdefd„ƒYZdefd„ƒYZd efd „ƒYZd efd „ƒYZd „Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z!d„Z"d„Z#d„Z$d„Z%d„Z&d„Z'd„Z(d„Z)d„Z*d„Z+d„Z,d „Z-d!„Z.d"„Z/d#„Z0d$„Z1d%„Z2d&„Z3d'„Z4d(„Z5d)„Z6dS(*iÿÿÿÿN(t get_rules(t exceptionsstests/data/bad_credentials.jsontMockServiceAccountCredentialscBseZed„ƒZRS(cCs ||fS(N((tclstfiletscopes((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytfrom_json_keyfile_name(s(t__name__t __module__t classmethodR(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR'stMockCredsIdentitycBseZd„ZRS(cCs|S(N((tselftarg((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt authorize.s(RRR (((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR -stMockCredsRaiseOauth2cBseZd„ZRS(cCstjjƒ‚dS(N(t oauth2clienttclienttHttpAccessTokenRefreshError(R R ((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR 3s(RRR (((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR2st MockDrivecBs\eZd„Zd„Zd d d d„Zd„Zd d d„Zd„Zddd„Z RS( cCsMtt|ƒjƒd|_d|_d|_d|_d|_d|_ dS(N( tsuperRt__init__tNonetdefaulttfile_idt mime_typetqtorder_byttype(R ((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR8s     cCs|S(N((R ((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytfilesAscCs||_||_||_|S(N(RRtfields(R RtorderByR((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytlistDs   cCs|S(N((R ((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytexecuteJscCs||_||_|S(N(RR(R tfileIdtmimeType((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytexportMs  cCs||_||_|S(N(RR(R RR((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytgetRs  sutf-8tstrictcCs|S(N((R tencodingterrors((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytdecodeWsN( RRRRRRR R#R$R((((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR7s   cCstƒS(N(R (t json_file((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytmock_google_credentials]scCs|jtjƒƒS(N(R thttplib2tHttp(tcreds((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytmock_authorize_connectionascCsT|jtjdtƒd}tj}||ƒ}|tjf}||k}|s>tjd|fd||fƒidt j ƒks”tj tƒr£tj tƒndd6tj |ƒd6tj |ƒd6d t j ƒksëtj |ƒrútj |ƒnd d 6tj |ƒd 6}di|d6}t tj|ƒƒ‚nd}}}}dS(s*The google credentials are called properlytServiceAccountCredentialss some files==sX%(py5)s {%(py5)s = %(py2)s {%(py2)s = %(py0)s.google_credentials }(%(py3)s) } == %(py8)sRtpy0tpy8tpy2Rtpy3tpy5tsassert %(py10)stpy10N(s==(sX%(py5)s {%(py5)s = %(py2)s {%(py2)s = %(py0)s.google_credentials }(%(py3)s) } == %(py8)ssassert %(py10)s(tsetattrRtservice_accountRRtgoogle_credentialstSCOPESt @pytest_art_call_reprcomparet @py_builtinstlocalst_should_repr_global_namet _safereprtAssertionErrort_format_explanationR(t monkeypatchRt @py_assert1t @py_assert4t @py_assert7t @py_assert6t @py_format9t @py_format11((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_google_credentials_successgs  ÃcCs*tjtjƒtjdƒWdQXdS(s,A non-existant json file raises an exceptions no_such.jsonN(tpytesttraisestextNoGoogleCredentialsFileErrorRR9(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt&test_google_credentials_file_existanceqscCs\dtfd„ƒY}|jtjd|ƒd}tjtjƒtj |ƒWdQXdS(s<Bad permissions on the json credentials file raises an errort"MockServiceAccountCredentialsRaisecBseZed„ƒZRS(cSsttjdƒ‚dS(Nsbogus permission error(tIOErrorterrnotEPERM(RRR((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyRzs(RRR R(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyRPysR/s some fileN( tobjectR7RR8RKRLRMtInsufficientPermissionsErrorRR9(RCRPR((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt(test_google_credentials_file_permissionswscCsYdtfd„ƒY}|jtjd|ƒd}tjtƒtj|ƒWdQXdS(s6"Other" ioerror on json credentials file raises itselfRPcBseZed„ƒZRS(cSsttjdƒ‚dS(Nsbogus disk full error(RQRRtENOSPC(RRR((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyRŠs(RRR R(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyRP‰sR/s some fileN( RTR7RR8RKRLRQRR9(RCRPR((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt%test_google_credentials_other_ioerror‡scCs-tjtjƒtjtjƒWdQXdS(s#A non-json file raises an exceptionN(RKRLRMtNotAServiceAccountErrorRR9tlibtesttTEST_CONFIG_PATH(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt test_google_credentials_not_json—scCs*tjtjƒtjtƒWdQXdS(s#A bad json file raises an exceptionN(RKRLRMtInvalidGoogleCredentialsErrorRR9tINVALID_JSON_PATH(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt test_google_credentials_bad_jsonscCs©tj}tƒ}||ƒ}tj}t||ƒ}| rddi tj|ƒd6dtj ƒkputj tƒr‡tjtƒndd6tj|ƒd6dtj ƒkp¼tj tƒrÎtjtƒndd6d tj ƒkpótj tƒrtjtƒnd d 6tj|ƒd 6d tj ƒkp:tj tƒrLtjtƒnd d 6tj|ƒd6tj|ƒd6}t tj |ƒƒ‚nt }}}}}dS(s.Authorizing a connection returns a http objectR5s²assert %(py13)s {%(py13)s = %(py0)s(%(py8)s {%(py8)s = %(py3)s {%(py3)s = %(py1)s.authorize_connection }(%(py6)s {%(py6)s = %(py4)s() }) }, %(py11)s {%(py11)s = %(py9)s.Http }) }R1R+tpy9tpy11t isinstanceR0Rtpy1R3R tpy4tpy6tpy13N(Rtauthorize_connectionR R+R,RbR;R@R=R>R?RARBR(t @py_assert2t @py_assert5RFt @py_assert10t @py_assert12t @py_format14((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt!test_authorize_connection_success¥s ÿ;cCs-tjtjƒtjtƒƒWdQXdS(s1Failing to authorize a connection raises an errorN(RKRLRMtGoogleAuthorizationErrorRRgR(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_authorize_connection_fail«sc Cs9dd„}|jtjd|ƒtjtƒtƒ\}}}tj}||k}|s't j d|fd||fƒidt j ƒkst j |ƒr¬t j|ƒndd6dt j ƒksÔt j tƒrãt jtƒndd6t j|ƒd 6}di|d 6}tt j|ƒƒ‚nd}}tj}||k}|st j d|fd||fƒidt j ƒksŽt j |ƒrt j|ƒndd6dt j ƒksÅt j tƒrÔt jtƒndd6t j|ƒd 6}di|d 6}tt j|ƒƒ‚nd}}tj}t||ƒ} | s+d didt j ƒkskt j tƒrzt jtƒndd6dt j ƒks¢t j |ƒr±t j|ƒndd6dt j ƒksÙt j tƒrèt jtƒndd6t j|ƒd 6t j| ƒd 6}tt j|ƒƒ‚nd}} dS(s=The Google API build function is called with the correct argscSs |||fS(N((tapitversionthttp((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt mock_buildµstbuilds==s1%(py0)s == %(py4)s {%(py4)s = %(py2)s.DRIVE_API }RpR0RR2RdR5sassert %(py6)sRes5%(py0)s == %(py4)s {%(py4)s = %(py2)s.DRIVE_VERSION }RqsOassert %(py6)s {%(py6)s = %(py0)s(%(py1)s, %(py4)s {%(py4)s = %(py2)s.Http }) }RbRrRcR+N(s==(s1%(py0)s == %(py4)s {%(py4)s = %(py2)s.DRIVE_API }sassert %(py6)s(s==(s5%(py0)s == %(py4)s {%(py4)s = %(py2)s.DRIVE_VERSION }sassert %(py6)s(RR7tgoogleapiclientt discoveryRt build_driveR R.t DRIVE_APIR;R<R=R>R?R@RARBt DRIVE_VERSIONR+R,Rb( RCRsRpRqRrt @py_assert3RDt @py_format5t @py_format7Ri((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_build_drive_success³s0   £  £  ÓcCs0tjtjƒtjtƒtƒWdQXdS(s?The Google API build function raises an error when oauth2 failsN(RKRLRMtGoogleDriveConnectionErrorRRwRR.(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_build_drive_fail_oauth2ÁscCsRdd„}|jtjd|ƒtjtjƒtj t ƒt ƒWdQXdS(s@The Google API build function raises an error when the api failscSstjjƒ‚dS(N(RuR'tError(RpRqRr((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyRsÊsRtN( RR7RuRvRKRLRMR~RRwR R.(RCRs((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt%test_build_drive_fail_googleapiclientÈs cCsØtƒ}d}tj||ƒ|j}|dk }|stjd|fd|dfƒidtjƒks|tj |ƒr‹tj |ƒndd6tj |ƒd6dtjƒksÃtj dƒrÒtj dƒndd6}di|d 6}t tj |ƒƒ‚nd}}|j }|dk }|s÷tjd|fd|dfƒidtjƒksmtj |ƒr|tj |ƒndd6tj |ƒd6dtjƒks´tj dƒrÃtj dƒndd6}di|d 6}t tj |ƒƒ‚nd}}|j}|dk }|sètjd|fd|dfƒidtjƒks^tj |ƒrmtj |ƒndd6tj |ƒd6dtjƒks¥tj dƒr´tj dƒndd6}di|d 6}t tj |ƒƒ‚nd}}|j}|dk }|sÙtjd|fd|dfƒidtjƒksOtj |ƒr^tj |ƒndd6tj |ƒd6dtjƒks–tj dƒr¥tj dƒndd6}di|d 6}t tj |ƒƒ‚nd}}|j}|dk }|sÊtjd|fd|dfƒidtjƒks@tj |ƒrOtj |ƒndd6tj |ƒd6dtjƒks‡tj dƒr–tj dƒndd6}di|d 6}t tj |ƒƒ‚nd}}dS( s;We call the google drive object with the expected argumentss some namesis nots-%(py2)s {%(py2)s = %(py0)s.q } is not %(py4)st mock_driveR0R2RRdR5sassert %(py6)sRes4%(py2)s {%(py2)s = %(py0)s.order_by } is not %(py4)ss2%(py2)s {%(py2)s = %(py0)s.fields } is not %(py4)ss0%(py2)s {%(py2)s = %(py0)s.type } is not %(py4)ss3%(py2)s {%(py2)s = %(py0)s.default } is not %(py4)sN(sis not(s-%(py2)s {%(py2)s = %(py0)s.q } is not %(py4)ssassert %(py6)s(sis not(s4%(py2)s {%(py2)s = %(py0)s.order_by } is not %(py4)ssassert %(py6)s(sis not(s2%(py2)s {%(py2)s = %(py0)s.fields } is not %(py4)ssassert %(py6)s(sis not(s0%(py2)s {%(py2)s = %(py0)s.type } is not %(py4)ssassert %(py6)s(sis not(s3%(py2)s {%(py2)s = %(py0)s.default } is not %(py4)ssassert %(py6)s(RRtget_drive_file_infoRRR;R<R=R>R?R@RARBRRRR(R‚tnameRDRzR{R|((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt test_get_drive_file_info_successÕsL  £  £  £  £  £cCsFdtfd„ƒY}tjtjƒtj|ƒdƒWdQXdS(s[We get an exception when google drive api raises one although the supplied name is goodtMockDriveRaiseClientcBseZd„ZRS(cSstjjƒ‚dS(N(RuR'R€(R RR((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR$æs(RRR$(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR†åss some nameN(RRKRLRMtGoogleListErrorRRƒ(R†((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt,test_get_drive_file_info_fail_drive_goodnameâscs~dtjjfd„ƒY‰dtf‡fd†ƒY}|jtjdˆƒtjtjƒt j |ƒdƒWdQXdS(sVWe get an exception when google drive api raises one when the supplied name is badt MockHttpErrorcBseZd„ZRS(cSsidd6|_dS(Nt400tstatus(tresp(R ((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyRñs(RRR(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR‰ðsR†cseZ‡fd†ZRS(csˆƒ}|‚dS(N((R RRterr(R‰(sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR$õs (RRR$((R‰(sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR†ôst HttpErrors some nameN( RuR'R€RR7RKRLRMtInvalidGoogleListCallErrorRRƒ(RCR†((R‰sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt+test_get_drive_file_info_fail_drive_badnameís cCsFdtfd„ƒY}tjtjƒtj|ƒdƒWdQXdS(s5We get an exception when google oauth2 api raises onetMockDriveRaiseOauth2cBseZd„ZRS(cSstjjƒ‚dS(N(RRR€(R RR((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR$s(RRR$(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR‘ss some nameN(RRKRLRMR‡RRƒ(R‘((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt$test_get_drive_file_info_fail_oauth2ÿsc Cs+tƒ}d}d}tj|||ƒ|j}||k}|stjd|fd||fƒidtjƒks…tj|ƒr”tj |ƒndd6tj |ƒd6dtjƒksÌtj|ƒrÛtj |ƒndd 6}di|d 6}t tj |ƒƒ‚nd}}|j }tj}||k}|stjd|fd||fƒidtjƒkstj|ƒrŽtj |ƒndd6tj |ƒd6dtjƒksÆtjtƒrÕtj tƒndd 6tj |ƒd 6}di|d6}t tj |ƒƒ‚nd}}}dS(s-The google drive object is called as expectedssome ids some names==s/%(py2)s {%(py2)s = %(py0)s.file_id } == %(py4)sR‚R0R2tidRdR5sassert %(py6)sResS%(py2)s {%(py2)s = %(py0)s.mime_type } == %(py6)s {%(py6)s = %(py4)s.DST_MIMETYPE }Rsassert %(py8)sR1N(s==(s/%(py2)s {%(py2)s = %(py0)s.file_id } == %(py4)ssassert %(py6)s(s==(sS%(py2)s {%(py2)s = %(py0)s.mime_type } == %(py6)s {%(py6)s = %(py4)s.DST_MIMETYPE }sassert %(py8)s(RRtdownload_sheetRR;R<R=R>R?R@RARBRRt DST_MIMETYPE( R‚R“R„RDRzR{R|RiRH((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_download_sheet_success s&  £  ³cCsIdtfd„ƒY}tjtjƒtj|ƒddƒWdQXdS(s4We get an exception when google drive api raises oneR†cBseZd„ZRS(cSstjjƒ‚dS(N(RuR'R€(R ((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR s(RRR (((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR†sssome ids some nameN(RRKRLRMtGoogleDownloadErrorRR”(R†((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_download_sheet_fail_drivescCsIdtfd„ƒY}tjtjƒtj|ƒddƒWdQXdS(s5We get an exception when google oauth2 api raises oneR‘cBseZd„ZRS(cSstjjƒ‚dS(N(RRR€(R ((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR $s(RRR (((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR‘#sssome ids some nameN(RRKRLRMR—RR”(R‘((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_download_sheet_fail_oauth2!scCsIdtfd„ƒY}tjtjƒtj|ƒddƒWdQXdS(s6We get an exception when google returns no spreadsheett MockDriveNonecBseZd„ZRS(cSsdS(N(R(R ((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR /s(RRR (((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyRš.sssome ids some nameN(RRKRLRMtNoSheetDownloadedErrorRR”(Rš((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt test_download_sheet_fail_content,scs*dtfd„ƒY‰d‡fd†}|jtdtƒ|jtjd|ƒtjdtdƒ\}}dg}||k}|st j d|fd||fƒid t j ƒksÉt j |ƒrØt j|ƒnd d 6t j|ƒd 6}di|d6}tt j|ƒƒ‚nd}}dS(s5We get something back when the download is successfult MockDriveDatacBs)eZddd„Zd„Zd„ZRS(sutf-8R%cSsdS(NsMockDriveData() ((R R&R'((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR(<scSsidd6gS(Nssome idR“((R RR((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR$?scSs|S(N((R tvalue((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytsplitBs(RRR(R$RŸ(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR;s csˆƒS(N((RpRqRr(R(sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyRsEsR9Rts creds.jsons some namesMockDriveData() s==s%(py0)s == %(py3)strulesR0R3R5sassert %(py5)sR4N(s==(s%(py0)s == %(py3)ssassert %(py5)s(RRR7RR*RuRvtrules_from_googleR.R;R<R=R>R?R@RARB(RCRstmetainfoR RhRDt @py_format4t @py_format6((RsT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_rules_from_google_success9s   lcsdtfd„ƒY‰d‡fd†}|jtdtƒ|jtjd|ƒtjt j ƒtj dt dƒWdQXdS( s?We get an exception when the spreadsheet is not found by googletMockDriveNoSheetcBseZd„ZRS(cSsgS(N((R RR((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR$Ts(RRR$(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyR¦SscsˆƒS(N((RpRqRr(R¦(sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyRsWsR9Rts creds.jsons some nameN( RRR7RR*RuRvRKRLRMtNoSuchSpreadsheetErrorR¡R.(RCRs((R¦sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_rules_from_google_failQs cCs_d„}|jtdtƒ|jtd|ƒtjtjƒtjdddƒWdQXdS(s1We get an exception when httplib2 raises an errorcSstjdƒ‚dS(Nstest exception(R+t HttpLib2Error(R-R ((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytmock_build_driveesR9Rws creds.jsons some nameN( R7RR*RKRLRMR©R¡R(RCRª((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyt test_rules_from_google_httperrorcs   cCsÀtjdƒ\}}|dd!}ddg}||k}|s®tjd|fd||fƒitj|ƒd6tj|ƒd 6}di|d 6}ttj|ƒƒ‚nd }}}d S(s,The first 2 lines of a rule file can be readssample_rules.csviisline 1 sline 2 s==s%(py1)s == %(py4)sRcRdR5sassert %(py6)sReN(s==(s%(py1)s == %(py4)ssassert %(py6)s(Rtrules_from_fileR;R<R@RARBR(R¢tdatat @py_assert0RzRhR{R|((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_rules_from_file_firstrs EcCs·tjdƒ\}}|d}d}||k}|s¥tjd |fd ||fƒitj|ƒd6tj|ƒd6}di|d 6}ttj|ƒƒ‚nd }}}d S(s(The last line of a rule file can be readssample_rules.csviÿÿÿÿs last line s==s%(py1)s == %(py4)sRcRdR5sassert %(py6)sReN(s==(s%(py1)s == %(py4)ssassert %(py6)s(RR¬R;R<R@RARBR(R¢R­R®RzRhR{R|((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_rules_from_file_lastxs  EcCs*tjtjƒtjdƒWdQXdS(s,A non-existant rule file raises an exceptions no_such.csvN(RKRLRMtNoRulesFileErrorRR¬(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_rule_file_existance~sc Csbd„}|jtd|ƒd}tj}itjd6|d6}||ƒ}||k}|sLtjd|fd||fƒidtjƒks¢tjtƒr±tj tƒndd 6d tjƒksÙtj|ƒrètj |ƒnd d 6tj |ƒd 6tj |ƒd 6tj |ƒd6}di|d6}t tj |ƒƒ‚nd}}}}dS(sReturn rules from a filecSs|S(N((tpath((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytmock_rules_from_fileˆsR¬s some/patht rules_sourcet rules_paths==sS%(py6)s {%(py6)s = %(py2)s {%(py2)s = %(py0)s.get_rule_file }(%(py4)s) } == %(py8)sRR0R³R1R2RdReR5sassert %(py10)sR6N(s==(sS%(py6)s {%(py6)s = %(py2)s {%(py2)s = %(py0)s.get_rule_file }(%(py4)s) } == %(py8)ssassert %(py10)s( R7Rt get_rule_filet FILE_SOURCER;R<R=R>R?R@RARBR( RCR´R³RDRzRiRFRHRI((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_get_rule_file_file†s   Ãc Cs^d„}|jtd|ƒd}d}tj}itjd6|d6|d6}||ƒ}|tj|f}||k}|sDtjd|fd||fƒid tjƒksÁtj tƒrÐtj tƒnd d 6tj |ƒd 6tj |ƒd 6tj |ƒd6tj |ƒd6} di| d6} t tj | ƒƒ‚nd}}}}}dS(sReturn rules from googlecSs |||fS(N((R-R R„((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytmock_rules_from_google•sR¡s some.jsons some nameRµR9tspreadsheet_names==sS%(py6)s {%(py6)s = %(py2)s {%(py2)s = %(py0)s.get_rule_file }(%(py4)s) } == %(py9)sRR0R`R2RdReR5sassert %(py11)sRaN(s==(sS%(py6)s {%(py6)s = %(py2)s {%(py2)s = %(py0)s.get_rule_file }(%(py4)s) } == %(py9)ssassert %(py11)s(R7RR·t GOOGLE_SOURCERgR;R<R=R>R?R@RARBR( RCRºR-R„RDRzRit @py_assert8RFt @py_format10t @py_format12((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pyttest_get_rule_file_google“s    œ(7t __builtin__R=t_pytest.assertion.rewritet assertiontrewriteR;RRtgoogleapiclient.discoveryRutgoogleapiclient.errorsR+toauth2client.clientRtoauth2client.service_accountRKtenforcerRRRMRZR^RTRR RRR*R.RJRORVRXR\R_RmRoR}RRR…RˆRR’R–R˜R™RœR¥R¨R«R¯R°R²R¹RÀ(((sT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_get_rules.pytsZ         &