def test_job_factory_skip(mock_job_init): service_config = Mock() job_config = {'cloud': None} mock_job_init.return_value = None job_factory = BaseJobFactory( service_name='test', job_types={'gce': GCETestJob}, can_skip=True ) job = job_factory.create_job(job_config, service_config) assert isinstance(job, NoOpJob)
def main(): """ mash - publish service application entry point """ try: logging.basicConfig() log = logging.getLogger('MashService') log.setLevel(logging.DEBUG) service_name = 'publish' # Create job factory job_factory = BaseJobFactory(service_name=service_name, job_types={ 'azure': AzurePublishJob, 'ec2': EC2PublishJob, 'gce': NoOpJob, 'oci': NoOpJob, 'aliyun': AliyunPublishJob }) config = BaseConfig() # run service, enter main loop ListenerService(service_exchange=service_name, config=config, custom_args={ 'job_factory': job_factory, 'thread_pool_count': config.get_publish_thread_pool_count() }) except MashException as e: # known exception log.error('{0}: {1}'.format(type(e).__name__, format(e))) traceback.print_exc() sys.exit(1) except KeyboardInterrupt: sys.exit(0) except SystemExit: # user exception, program aborted by user sys.exit(0) except Exception as e: # exception we did no expect, show python backtrace log.error('Unexpected error: {0}'.format(e)) traceback.print_exc() sys.exit(1)
def main(): """ mash - replicate service application entry point """ try: logging.basicConfig() log = logging.getLogger('MashService') log.setLevel(logging.INFO) service_name = 'replicate' # Create job factory job_factory = BaseJobFactory( service_name=service_name, job_types={ 'azure': NoOpJob, 'ec2': EC2ReplicateJob, 'gce': NoOpJob, 'oci': NoOpJob, 'aliyun': AliyunReplicateJob } ) # run service, enter main loop ListenerService( service_exchange=service_name, config=BaseConfig(), custom_args={ 'job_factory': job_factory } ) except MashException as e: # known exception log.error('%s: %s', type(e).__name__, format(e)) traceback.print_exc() sys.exit(1) except KeyboardInterrupt: sys.exit(0) except SystemExit as e: # user exception, program aborted by user sys.exit(e) except Exception as e: # exception we did no expect, show python backtrace log.error('Unexpected error: {0}'.format(e)) traceback.print_exc() sys.exit(1)
def main(): """ mash - raw image upload service application entry point """ try: logging.basicConfig() log = logging.getLogger('MashService') log.setLevel(logging.DEBUG) service_name = 'raw_image_upload' # Create job factory job_factory = BaseJobFactory( service_name=service_name, job_types={ 'azure_sas': AzureSASUploadJob, 's3bucket': S3BucketUploadJob }, job_type_key='raw_image_upload_type', can_skip=True ) # run service, enter main loop ListenerService( service_exchange=service_name, config=UploadConfig(), custom_args={ 'job_factory': job_factory } ) except MashException as e: # known exception log.error('{0}: {1}'.format(type(e).__name__, format(e))) traceback.print_exc() sys.exit(1) except KeyboardInterrupt: sys.exit(0) except SystemExit: # user exception, program aborted by user sys.exit(0) except Exception as e: # exception we did no expect, show python backtrace log.error('Unexpected error: {0}'.format(e)) traceback.print_exc() sys.exit(1)