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,
)