def test_client_wait_for_build_retry(fixture_build_details_json):
iibc = IIBClient("fake-host.test",
poll_interval=1,
retries=10,
backoff_factor=0)
with pytest.raises(requests.exceptions.RequestException,
match=".*Max retries exceeded*."):
iibc.wait_for_build(
IIBBuildDetailsModel.from_dict(fixture_build_details_json))
def test_iib_error_response():
with requests_mock.Mocker() as m:
m.register_uri(
"GET",
"/api/v1/builds",
status_code=400,
json={"error": "An ugly error has occurred!"},
)
iibc = IIBClient("fake-host")
with pytest.raises(IIBException):
iibc.get_builds()
def test_http_error_response():
with requests_mock.Mocker() as m:
m.register_uri(
"GET",
"/api/v1/builds",
status_code=400,
text="An ugly HTTP error has occurred!",
)
iibc = IIBClient("fake-host")
with pytest.raises(HTTPError):
iibc.get_builds()
def test_iib_client_failure(fixture_build_details_json):
error_msg = ("Either both or neither of overwrite-from-index "
"and overwrite-from-index-token should be specified.")
with requests_mock.Mocker() as m:
m.register_uri(
"POST",
"/api/v1/builds/add",
status_code=200,
json=fixture_build_details_json,
)
m.register_uri(
"POST",
"/api/v1/builds/rm",
status_code=200,
json=fixture_build_details_json,
)
iibc = IIBClient("fake-host")
with pytest.raises(ValueError, match=error_msg):
iibc.remove_operators(
"index-image",
["operator1"],
[],
binary_image="binary",
overwrite_from_index=True,
)
with pytest.raises(ValueError, match=error_msg):
iibc.remove_operators(
"index-image",
["operator1"],
[],
binary_image="binary",
overwrite_from_index_token="str",
)
with pytest.raises(ValueError, match=error_msg):
iibc.add_bundles(
"index-image",
["bundles-map"],
[],
binary_image="binary",
cnr_token="cnr",
organization="org",
overwrite_from_index=True,
)
with pytest.raises(ValueError, match=error_msg):
iibc.add_bundles(
"index-image",
["bundles-map"],
[],
binary_image="binary",
cnr_token="cnr",
organization="org",
overwrite_from_index_token="str",
)
def test_client_wait_for_build_timeout(fixture_build_details_json):
# set wait timeout for 2 seconds
iibc = IIBClient("fake-host.test",
poll_interval=1,
wait_for_build_timeout=2)
with requests_mock.Mocker() as m:
m.register_uri(
"GET",
"/api/v1/builds/{}".format(fixture_build_details_json["id"]),
status_code=200,
json=fixture_build_details_json,
)
with pytest.raises(IIBException, match="Timeout*."):
iibc.wait_for_build(
IIBBuildDetailsModel.from_dict(fixture_build_details_json))
def test_client_wait_for_build(fixture_build_details_json):
iibc = IIBClient("fake-host", poll_interval=1)
bdetails_finished = copy.copy(fixture_build_details_json)
bdetails_finished["state"] = "complete"
with requests_mock.Mocker() as m:
m.register_uri(
"GET",
"/api/v1/builds/1",
[
{"json": fixture_build_details_json, "status_code": 200},
{"json": bdetails_finished, "status_code": 200},
],
)
iibc.wait_for_build(IIBBuildDetailsModel.from_dict(fixture_build_details_json))
bdetails_finished["state"] = "failed"
with requests_mock.Mocker() as m:
m.register_uri(
"GET",
"/api/v1/builds/1",
[
{"json": fixture_build_details_json, "status_code": 200},
{"json": bdetails_finished, "status_code": 200},
],
)
iibc.wait_for_build(IIBBuildDetailsModel.from_dict(fixture_build_details_json))
def test_iibbuilddetails_pager(
fixture_builds_page1_json,
fixture_builds_page2_json,
fixture_build_details_json,
fixture_build_details_json2,
):
with requests_mock.Mocker() as m:
m.register_uri("GET",
"/api/v1/builds",
status_code=200,
json=fixture_builds_page1_json)
m.register_uri(
"GET",
"/api/v1/builds?page=2",
status_code=200,
json=fixture_builds_page2_json,
)
m.register_uri(
"GET",
"/api/v1/builds?page=1",
status_code=200,
json=fixture_builds_page1_json,
)
iibc = IIBClient("fake-host")
pager = iibc.get_builds()
assert pager.items() == [
IIBBuildDetailsModel.from_dict(
fixture_builds_page1_json["items"][0])
]
pager.next()
assert pager.items() == [
IIBBuildDetailsModel.from_dict(
fixture_builds_page2_json["items"][0])
]
pager.prev()
assert pager.items() == [
IIBBuildDetailsModel.from_dict(
fixture_builds_page1_json["items"][0])
]
def test_iib_client_no_binary_image(fixture_build_details_json3):
with requests_mock.Mocker() as m:
m.register_uri(
"POST",
"/api/v1/builds/add",
status_code=200,
json=fixture_build_details_json3,
)
m.register_uri(
"POST",
"/api/v1/builds/rm",
status_code=200,
json=fixture_build_details_json3,
)
iibc = IIBClient("fake-host")
assert iibc.add_bundles(
"index-image", ["bundles-map"],
[]) == IIBBuildDetailsModel.from_dict(fixture_build_details_json3)
assert (iibc.add_bundles(
"index-image",
["bundles-map"],
[],
cnr_token="cnr",
organization="org",
overwrite_from_index=True,
overwrite_from_index_token="str",
) == IIBBuildDetailsModel.from_dict(fixture_build_details_json3))
assert (iibc.add_bundles("index-image", ["bundles-map"], [],
raw=True) == fixture_build_details_json3)
assert (iibc.remove_operators(
"index-image",
["operator1"],
[],
overwrite_from_index=True,
overwrite_from_index_token="str",
) == IIBBuildDetailsModel.from_dict(fixture_build_details_json3))
assert (iibc.remove_operators("index-image", ["operator1"], [],
raw=True) == fixture_build_details_json3)
def test_rebuild_index():
iibc = IIBClient("fake-host")
iibc.rebuild_index("some index")
def test_health():
iibc = IIBClient("fake-host")
iibc.health()
def test_client_auth():
auth = IIBBasicAuth("foo", "bar")
iibc = IIBClient("fake-host", auth=auth)
assert iibc.iib_session.session.headers["auth"] == ("foo", "bar")
def test_iib_client(fixture_build_details_json, fixture_builds_page1_json):
with requests_mock.Mocker() as m:
m.register_uri(
"POST",
"/api/v1/builds/add",
status_code=200,
json=fixture_build_details_json,
)
m.register_uri(
"POST",
"/api/v1/builds/rm",
status_code=200,
json=fixture_build_details_json,
)
m.register_uri(
"GET", "/api/v1/builds", status_code=200, json=fixture_builds_page1_json
)
m.register_uri(
"GET", "/api/v1/builds/1", status_code=200, json=fixture_build_details_json
)
iibc = IIBClient("fake-host")
assert iibc.add_bundles(
"index-image", "binary", ["bundles-map"], []
) == IIBBuildDetailsModel.from_dict(fixture_build_details_json)
assert iibc.add_bundles(
"index-image",
"binary",
["bundles-map"],
[],
cnr_token="cnr",
organization="org",
) == IIBBuildDetailsModel.from_dict(fixture_build_details_json)
assert (
iibc.add_bundles("index-image", "binary", ["bundles-map"], [], raw=True)
== fixture_build_details_json
)
assert iibc.remove_operators(
"index-image", "binary", ["operator1"], []
) == IIBBuildDetailsModel.from_dict(fixture_build_details_json)
assert (
iibc.remove_operators("index-image", "binary", ["operator1"], [], raw=True)
== fixture_build_details_json
)
assert iibc.get_build(1) == IIBBuildDetailsModel.from_dict(
fixture_build_details_json
)
assert iibc.get_build(1, raw=True) == fixture_build_details_json
assert iibc.get_builds() == IIBBuildDetailsPager.from_dict(
iibc, fixture_builds_page1_json
)
assert iibc.get_builds(raw=True) == fixture_builds_page1_json