def test_register_agent_raises_error(patch_post, cloud_api):
response = {"data": {"register_agent": {"id": None}}}
patch_post(response)
with set_temporary_config({"cloud.auth_token": "secret_token", "backend": "cloud"}):
client = Client()
with pytest.raises(ValueError):
client.register_agent(agent_type="type", name="name", labels=["1", "2"])
def test_register_agent(patch_post, cloud_api):
response = {"data": {"register_agent": {"id": "ID"}}}
patch_post(response)
with set_temporary_config({"cloud.auth_token": "secret_token"}):
client = Client()
agent_id = client.register_agent(agent_type="type",
name="name",
labels=["1", "2"])
assert agent_id == "ID"
