3 SrSSKJr SSKJr SSKJr SSKJr "SS\5r"SS \5r "S S \5r "S S \5r g)z6Helper functions for testing calls to the Rewrite API.)absolute_import)print_function)division)unicode_literalsc\rSrSrSrg)RewriteHaltExceptionN)__name__ __module__ __qualname____firstlineno____static_attributes__r -platform/gsutil/gslib/tests/rewrite_helper.pyrrsrrc$\rSrSrSrSrSrSrg)HaltingRewriteCallbackHandlerzETest callback handler for intentionally stopping a rewrite operation.cXlgN _halt_at_byte)self halt_at_bytes r__init__&HaltingRewriteCallbackHandler.__init__s%rc8XR:a [S5eg)z=Forcibly exits if the operation has passed the halting point.zArtificially halting rewriteN)rrrtotal_bytes_rewrittenunused_total_sizes rcall"HaltingRewriteCallbackHandler.call"s 2 22 !? @@3rrNr r r r__doc__rr!rr rrrrsM&Arrc$\rSrSrSrSrSrSrg)"EnsureRewriteResumeCallbackHandler(z?Test callback handler for ensuring a rewrite operation resumed.cXlgr_required_byter required_bytes rr+EnsureRewriteResumeCallbackHandler.__init__+s'rc^XR::a[SU<SUR<S35eg)z9Exits if the total bytes rewritten is less than expected.zRewrite did not resume; z bytes written, but ( bytes should have already been written.N)r*rrs rr!'EnsureRewriteResumeCallbackHandler.call/s2 3 33  !$"5"5 7 884rr)Nr#r rrr&r&(sG(8rr&c$\rSrSrSrSrSrSrg)#EnsureRewriteRestartCallbackHandler8zATest callback handler for ensuring a rewrite operation restarted.cXlSUlg)NF)r*_got_restart_bytesr+s rr,EnsureRewriteRestartCallbackHandler.__init__;s'#DrcUR(d5XR::aSUlg[SU<SUR<S35eg)zr<sH=&%'9 AF A 8 8 :&:r