def get_gcs_connection(): ''' Checks if there is a GCS connection and, if not, creates one :return: True once the connection is established ''' #TODO returning True doesn't seem right here but gets the job done # Does not account for no connectivity, api issues... if storage.get_default_connection() and storage.get_default_bucket(): return True else: connect = storage.get_connection().from_service_account_json( SRV_ACCT_CREDS) bucket = storage.bucket.Bucket(name=BUCKET_NAME) storage.set_defaults(bucket, PROJECT_ID, connect) return True
import six import tempfile import time import unittest2 from gcloud import exceptions from gcloud import storage from gcloud import _helpers from gcloud.storage._helpers import _base64_md5hash HTTP = httplib2.Http() SHARED_BUCKETS = {} _helpers._PROJECT_ENV_VAR_NAME = 'GCLOUD_TESTS_PROJECT_ID' storage.set_defaults() def setUpModule(): if 'test_bucket' not in SHARED_BUCKETS: # %d rounds milliseconds to nearest integer. bucket_name = 'new%d' % (1000 * time.time(),) # In the **very** rare case the bucket name is reserved, this # fails with a ConnectionError. SHARED_BUCKETS['test_bucket'] = storage.create_bucket(bucket_name) def tearDownModule(): for bucket in SHARED_BUCKETS.values(): bucket.delete(force=True)
def _callFUT(self, bucket=None, project=None, connection=None): from gcloud.storage import set_defaults return set_defaults(bucket=bucket, project=project, connection=connection)
def _callFUT(self, bucket=None, project=None, connection=None): from gcloud.storage import set_defaults return set_defaults(bucket=bucket, project=project, connection=connection)
def initialize_gcloud(): datastore.set_defaults() storage.set_defaults()
import httplib2 import tempfile import time import unittest2 from gcloud import exceptions from gcloud import storage from gcloud.storage._helpers import _base64_md5hash from gcloud.storage import _implicit_environ HTTP = httplib2.Http() SHARED_BUCKETS = {} storage._PROJECT_ENV_VAR_NAME = 'GCLOUD_TESTS_PROJECT_ID' storage.set_defaults() CONNECTION = _implicit_environ.CONNECTION def setUpModule(): if 'test_bucket' not in SHARED_BUCKETS: # %d rounds milliseconds to nearest integer. bucket_name = 'new%d' % (1000 * time.time(), ) # In the **very** rare case the bucket name is reserved, this # fails with a ConnectionError. SHARED_BUCKETS['test_bucket'] = CONNECTION.create_bucket(bucket_name) def tearDownModule(): for bucket in SHARED_BUCKETS.values():