
                             d dl mZ d dlmZmZ d dlmZ d dlmZ d dl	m
Z
 d dlZd dlmZ de
d	ej                  fd
Zde
d	ej                  fdZd	efdZy)    )	MagicMock)AbortedGoogleAPICallError)Any)	ErrorInfo)StatusN)
rpc_status	status_pbreturnc                    t        j                  |       }t        t        j                        }|j
                  |j
                  _        |j                  |j                  _        |j                  |j                  _        |S )N)spec)	r	   	to_statusr   grpcCalldetailsreturn_valuecodetrailing_metadata)r
   status	mock_calls      Dlib/third_party/google/cloud/pubsublite/testing/test_reset_signal.py	make_callr      s\    !!),Ftyy)I%+^^I""(++INN/5/G/GI,    c                 >    t        |       }d |j                  _        |S )N)r   r   r   )r
   r   s     r   make_call_without_metadatar   !   s     )$I/3I,r   c                      t               } | j                  t        dd             t        d| g      }t	        dt        |            S )NRESETzpubsublite.googleapis.com)reasondomain
   )r   r    )response)r   Packr   r   r   r   )anyr
   s     r   make_reset_signalr%   (   s=    
%CHHYg.IJKB.I2	) 455r   )unittest.mockr   google.api_core.exceptionsr   r    cloudsdk.google.protobuf.any_pb2r   google.rpc.error_details_pb2r   google.rpc.status_pb2r   r   grpc_statusr	   r   r   r   r%    r   r   <module>r-      sW    $ B 0 2 (  " DII & TYY 6- 6r   