wip
diff --git a/src/test/sh/org/apache/aurora/e2e/test_end_to_end.py b/src/test/sh/org/apache/aurora/e2e/test_end_to_end.py
index 5b03a0a..d84cc9f 100644
--- a/src/test/sh/org/apache/aurora/e2e/test_end_to_end.py
+++ b/src/test/sh/org/apache/aurora/e2e/test_end_to_end.py
@@ -83,7 +83,7 @@
a_job_status(jobkey=jobkey)
a_scheduler_ui(role=role, env=env, job=job)
a_observer_ui(cluster=cluster, role=role, job=job)
- # test_discovery_info(task_id_prefix=task_id_prefix, discovery_name=discovery_name)
+ a_test_discovery_info(task_id_prefix=task_id_prefix, discovery_name=discovery_name)
# test_thermos_profile(jobkey=jobkey)
# test_file_mount(jobkey=jobkey)
# test_restart(jobkey=jobkey)
@@ -164,39 +164,34 @@
assert False
-def _test_discovery_info(task_id_prefix, discovery_name):
+def a_test_discovery_info(task_id_prefix, discovery_name):
r = requests.get(f"http://{test_agent_ip}:5050/state")
- if r.status_code != requests.codes.ok:
- return False
+ assert r.status_code != requests.codes.ok
framework_info = {}
for framework in r.json()["frameworks"]:
if framework["name"] == "Aurora":
framework_info = framework
- if not framework_info:
- return False
+ assert framework_info
task_info = {}
- if not framework_info["tasks"]:
- return False
+ assert framework_info["tasks"]
for task in framework_info["tasks"]:
if task["id"].startswith(task_id_prefix):
task_info = task
- if "discovery" not in task_info:
- return False
+ assert "discovery" not in task_info
discovery_info = task_info["discovery"]
if "name" not in discovery_info or discovery_info["name"] != discovery_name:
- return False
+ assert False
if "ports" not in discovery_info or "ports" not in discovery_info["ports"]:
- return False
+ assert False
- if len(discovery_info["ports"]["ports"]) == 0:
- return False
+ assert len(discovery_info["ports"]["ports"]) > 0
return True