3 dZddlmZddlmZddlmZddlmZGddeZGdd eZ Gd d eZ Gd d eZ y)z6Helper functions for testing calls to the Rewrite API.)absolute_import)print_function)division)unicode_literalsc eZdZy)RewriteHaltExceptionN)__name__ __module__ __qualname__-platform/gsutil/gslib/tests/rewrite_helper.pyrrsr rceZdZdZdZdZy)HaltingRewriteCallbackHandlerzETest callback handler for intentionally stopping a rewrite operation.c||_yN) _halt_at_byte)self halt_at_bytes r__init__z&HaltingRewriteCallbackHandler.__init__s %Dr c8||jk\r tdy)z=Forcibly exits if the operation has passed the halting point.zArtificially halting rewriteN)rrrtotal_bytes_rewrittenunused_total_sizes rcallz"HaltingRewriteCallbackHandler.call"s! 2 22 !? @@3r Nr r r __doc__rrr r rrrsM&Ar rceZdZdZdZdZy)"EnsureRewriteResumeCallbackHandlerz?Test callback handler for ensuring a rewrite operation resumed.c||_yr)_required_byter required_bytes rrz+EnsureRewriteResumeCallbackHandler.__init__+s 'Dr cZ||jkrtd|d|jdy)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 rrz'EnsureRewriteResumeCallbackHandler.call/s4 3 33  !$"5"5 7 884r Nrr r rrr(sG(8r rceZdZdZdZdZy)#EnsureRewriteRestartCallbackHandlerzATest callback handler for ensuring a rewrite operation restarted.c ||_d|_y)NF)r!_got_restart_bytesr"s rrz,EnsureRewriteRestartCallbackHandler.__init__;s'D#Dr c|js3||jkrd|_ytd|d|jdy)zr.sH=&%'9 AF A 8 8 :&:r