def test_async_create_with_bad_request_factory(self): pool = DeferredResource._create_thread_pool(1) try: with self.assertRaises(self.BadFactoryError): run_deferred(DeferredResource._create_async(self.bad_factory, _pool=pool), print_traceback=False) self.assertFalse(pool.threads) finally: pool.stop()
def test_async_create_with_bad_request_factory(self): pool = DeferredResource._create_thread_pool(1) try: with self.assertRaises(self.BadFactoryError): run_deferred( DeferredResource._create_async( self.bad_factory, _pool=pool), print_traceback=False ) self.assertFalse(pool.threads) finally: pool.stop()
def test_async_create(self): res = fake_resource() def_res = run_deferred(DeferredResource._create_async(lambda: res)) try: self.assertTrue(isinstance(def_res, DeferredResource)) self.assertEqual(def_res._resource, res) finally: def_res.stop()
def start(self): self.resource = yield DeferredResource.build( 'pubsub', 'v1', credentials=self.credentials) self.resource.start() # Check to see if the topic exists. Anything that's not a 200 means it # doesn't exist or is inaccessable. res = yield self.resource.api.projects.topics.get(topic=self.topic) log.msg('PubSub client for topic %s started: %s' % (self.topic, res))
def create_resource(self, credentials=None): self.resource = fake_resource() self.deferred_resource = DeferredResource( resource=self.resource, credentials=credentials, retry_wait_seconds=0.1, # Run tests fast. ) with self.deferred_resource: yield
def test_async_create(self): res = fake_resource() def_res = run_deferred( DeferredResource._create_async(lambda: res) ) try: self.assertTrue(isinstance(def_res, DeferredResource)) self.assertEqual(def_res._resource, res) finally: def_res.stop()
def _loadResource(self): """Loads and instantiates a cloud endpoints resource to CBE master push.""" # Construct our DeferredResource. service = yield DeferredResource.build( 'master_push', 'v0', credentials=auth.create_credentials_for_master(self.activeMaster), discoveryServiceUrl=self.discoveryUrlTemplate, verbose=self.verbose, log_prefix='CBEStatusPush') defer.returnValue(service)
def create_buildbucket_service( master, hostname=None, verbose=None): """Asynchronously creates buildbucket API resource. Returns: A DeferredResource as Deferred. """ hostname = hostname or get_default_buildbucket_hostname(master) return DeferredResource.build( 'buildbucket', 'v1', credentials=auth.create_credentials_for_master(master), max_concurrent_requests=10, discoveryServiceUrl=buildbucket_api_discovery_url(hostname), verbose=verbose or False, log_prefix=common.LOG_PREFIX, )