def test_update_job(self): job = JobConfig(config={}).to_dict() httpretty.register_uri(httpretty.PATCH, BaseApiHandler.build_url( self.api_config.base_url, '/', 'username', 'project_name', 'jobs', 1), body=json.dumps(job), content_type='application/json', status=200) # Schema response result = self.api_handler.update_job('username', 'project_name', 1, {'name': 'new'}) assert result.to_dict() == job # Raw response self.set_raw_response() result = self.api_handler.update_job('username', 'project_name', 1, {'name': 'new'}) assert result == job # Async self.assert_async_call(api_handler_call=lambda: self.api_handler. update_job('username', 'project_name', 1, {'name': 'new'}, background=True), method='patch')
def test_resume_job_with_config(self): job = JobConfig(config={}).to_dict() config = {'config': {'logging': {'level': 'error'}}} httpretty.register_uri(httpretty.POST, BaseApiHandler.build_url( self.api_config.base_url, '/', 'username', 'project_name', 'jobs', 1, 'resume'), body=json.dumps(job), content_type='application/json', status=200) # Schema response result = self.api_handler.resume('username', 'project_name', 1, config) assert result.to_dict() == job # Raw response self.set_raw_response() result = self.api_handler.resume('username', 'project_name', 1, config) assert result == job # Async self.assert_async_call( api_handler_call=lambda: self.api_handler.resume( 'username', 'project_name', 1, config, background=True), method='post')
def test_get_bookmarked_jobs(self): project_uuid = uuid.uuid4().hex obj_uuid = uuid.uuid4().hex objs = [{ 'content_object': JobConfig(config={}, uuid=obj_uuid, project=project_uuid).to_dict() } for _ in range(10)] httpretty.register_uri(httpretty.GET, BaseApiHandler.build_url( self.api_config.base_url, '/bookmarks', 'user', 'jobs'), body=json.dumps({ 'results': objs, 'count': 10, 'next': None }), content_type='application/json', status=200) # Schema response result = self.api_handler.jobs('user') assert len(result['results']) == 10 assert isinstance(result['results'][0], JobConfig) # Raw response self.set_raw_response() result = self.api_handler.jobs('user') assert len(result['results']) == 10 assert isinstance(result['results'][0], Mapping)
def test_create_job(self): project_uuid = uuid.uuid4().hex obj = JobConfig(project=project_uuid) httpretty.register_uri( httpretty.POST, BaseApiHandler.build_url( self.api_config.base_url, '/', 'user', 'project', 'jobs'), body=json.dumps(obj.to_dict()), content_type='application/json', status=200) # Schema response result = self.api_handler.create_job('user', 'project', obj) assert result.to_dict() == obj.to_dict() # Raw response self.set_raw_response() result = self.api_handler.create_job('user', 'project', obj) assert result == obj.to_dict() # Async self.assert_async_call( api_handler_call=lambda: self.api_handler.create_job( 'user', 'project', obj, background=True), method='post') # Test create experiment with dict httpretty.register_uri( httpretty.POST, BaseApiHandler.build_url( self.api_config.base_url, '/', 'user', 'project', 'jobs'), body=json.dumps(obj.to_dict()), content_type='application/json', status=200) # Schema response self.set_schema_response() result = self.api_handler.create_job('user', 'project', obj.to_dict()) assert result.to_dict() == obj.to_dict() # Raw response self.set_raw_response() result = self.api_handler.create_job('user', 'project', obj.to_dict()) assert result == obj.to_dict() # Async self.assert_async_call( api_handler_call=lambda: self.api_handler.create_job( 'user', 'project', obj.to_dict(), background=True), method='post')
def test_get_job(self): job = JobConfig(config={}).to_dict() httpretty.register_uri(httpretty.GET, BaseApiHandler.build_url( self.api_config.base_url, '/', 'username', 'project_name', 'jobs', 1), body=json.dumps(job), content_type='application/json', status=200) # Schema response result = self.api_handler.get_job('username', 'project_name', 1) assert result.to_dict() == job # Raw response self.set_raw_response() result = self.api_handler.get_job('username', 'project_name', 1) assert result == job
def test_list_builds(self): project_uuid = uuid.uuid4().hex xp_uuid = uuid.uuid4().hex xps = [ JobConfig(config={}, uuid=xp_uuid, project=project_uuid).to_dict() for _ in range(10) ] httpretty.register_uri(httpretty.GET, BaseApiHandler.build_url( self.api_config.base_url, '/', 'user', 'project', 'builds'), body=json.dumps({ 'results': xps, 'count': 10, 'next': None }), content_type='application/json', status=200) # Schema response response = self.api_handler.list_builds('user', 'project') assert len(response['results']) == 10 assert isinstance(response['results'][0], JobConfig) # Raw response self.set_raw_response() response = self.api_handler.list_builds('user', 'project') assert len(response['results']) == 10 assert isinstance(response['results'][0], Mapping) # Pagination httpretty.register_uri( httpretty.GET, BaseApiHandler.build_url(self.api_config.base_url, '/', 'user', 'project', 'builds') + '?offset=2', body=json.dumps({ 'results': xps, 'count': 10, 'next': None }), content_type='application/json', status=200) # Schema response self.set_schema_response() response = self.api_handler.list_builds('user', 'project', page=2) assert len(response['results']) == 10 assert isinstance(response['results'][0], JobConfig) # Raw response self.set_raw_response() response = self.api_handler.list_builds('user', 'project', page=2) assert len(response['results']) == 10 assert isinstance(response['results'][0], Mapping) # Query, Sort httpretty.register_uri( httpretty.GET, BaseApiHandler.build_url(self.api_config.base_url, '/', 'username', 'project_name', 'builds') + '?query=started_at:>=2010-10-10,sort=created_at', body=json.dumps({ 'results': xps, 'count': 10, 'next': None }), content_type='application/json', status=200) # Schema response self.set_schema_response() response = self.api_handler.list_builds( 'user', 'project', query='started_at:>=2010-10-10', sort='created_at') assert len(response['results']) == 10 assert isinstance(response['results'][0], JobConfig) # Raw response self.set_raw_response() response = self.api_handler.list_builds( 'user', 'project', query='started_at:>=2010-10-10', sort='created_at') assert len(response['results']) == 10 assert isinstance(response['results'][0], Mapping)