def _instrument_iid_service(self, app=None, status=200, payload=_DEFAULT_RESPONSE): if not app: app = firebase_admin.get_app() fcm_service = messaging._get_messaging_service(app) recorder = [] fcm_service._client.session.mount( 'https://iid.googleapis.com', testutils.MockAdapter(payload, status, recorder)) return fcm_service, recorder
def _instrument_messaging_service(self, app, status=200, payload=_DEFAULT_RESPONSE): fcm_service = messaging._get_messaging_service(app) recorder = [] fcm_service._client.session.mount( 'https://fcm.googleapis.com', MockAdapter(payload, status, recorder)) return fcm_service, recorder
def setUp(self): self.testbed = testbed.Testbed() self.testbed.activate() self.testbed.init_app_identity_stub() self.testbed.init_memcache_stub() self.testbed.setup_env(app_id='your-app-id', overwrite=True) # Stub the FCM admin module try: self._app = firebase_admin.get_app('tbans') except ValueError: self._app = firebase_admin.initialize_app(name='tbans') messaging._get_messaging_service(self._app).send = self._stub_send
def setup(self): app = firebase_admin.get_app() self.fcm_service = messaging._get_messaging_service(app) self.recorder = []