def cleanup():
from integration.config import make_code_bucket
bucket = make_code_bucket()
files = list(bucket.list_files('*.tgz'))
_log().debug('Cleaning up {}'.format(files))
for path in files:
_log().debug('Removing {}'.format(path))
bucket.remove(path)
def _make_deployment(self, stage, **kwargs):
from foundations import Job, JobSourceBundle, BucketJobDeployment
from uuid import uuid4
from integration.config import make_code_bucket, make_result_bucket
job = Job(stage, **kwargs)
job_name = str(uuid4())
source_bundle = JobSourceBundle.for_deployment()
return BucketJobDeployment(job_name, job, source_bundle,
make_code_bucket(), make_result_bucket())
def test_can_deploy_job_remotely(self):
from foundations.global_state import foundations_context
from os.path import isfile
from integration.config import make_code_bucket
from foundations_gcp import GCPBucket
def method():
pass
stage = foundations_context.pipeline().stage(method)
deployment = self._make_deployment(stage)
deployment.deploy()
bucket = make_code_bucket()
code_path = '{}.tgz'.format(deployment.job_name())
self.assertTrue(bucket.exists(code_path))