def _check_kedroviz_running(context):
"""
Check that a service is running and responding appropriately
Args:
context (behave.runner.Context): Test context
"""
data_json = json.loads(download_url("http://localhost:4141/api/nodes.json"))
try:
assert context.result.poll() is None
assert "predict" in data_json["nodes"][0]["full_name"]
finally:
context.result.terminate()
def _check_kedroviz_running(context):
"""
Check that a service is running and responding appropriately
Args:
context (behave.runner.Context): Test context
"""
data_json = json.loads(download_url("http://localhost:4141/api/main"))
try:
assert context.result.poll() is None
assert ("example_iris_data" == sorted(
data_json["nodes"], key=lambda i: i["full_name"])[0]["full_name"])
finally:
context.result.terminate()
def _check_service_up(context):
"""
Check that a service is running and responding appropriately
Args:
context (behave.runner.Context): Test context
"""
data_json = json.loads(
download_url("http://localhost:4141/logs/nodes.json"))
try:
assert context.result.poll() is None
assert "example_iris_data" in data_json[0]["inputs"]
finally:
context.result.terminate()
def _check_service_up(context):
"""
Check that a service is running and responding appropriately
Args:
context (behave.runner.Context): Test context
"""
data_json = json.loads(
download_url("http://localhost:4141/api/nodes.json"))
try:
assert context.result.poll() is None
assert (
data_json["snapshots"][0]["nodes"][0]["full_name"] ==
"predict([example_model,example_test_x]) -> [example_predictions]")
finally:
context.result.terminate()
def _check_service_up(context: behave.runner.Context, url: str, string: str):
"""
Check that a service is running and responding appropriately
Args:
context: Test context.
url: Url that is to be read.
string: The string to be checked.
"""
data = download_url(url)
try:
assert context.result.poll() is None
assert string in data
finally:
if "docker" in context.result.args:
kill_docker_containers(context.project_name)
else:
context.result.terminate()