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_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)
def test_create_job(self):
project_uuid = uuid.uuid4().hex
obj = JobConfig(project=project_uuid, config={})
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')
