Skip to content

Commit

Permalink
Edge case where node count is 1
Browse files Browse the repository at this point in the history
  • Loading branch information
juliannguyen4 committed Jun 19, 2024
1 parent cd37dc4 commit 0c57d19
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions test/metrics/test_node_close_listener.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,9 +123,11 @@ def get_first_node(node_info: dict):
time.sleep(10)

assert node_close_called is True
print(f"{node_close_called} is true. Passed")
finally:
c.close()
finally:
# Cleanup
subprocess.run(["aerolab", "cluster", "stop"], check=True)
subprocess.run(["aerolab", "cluster", "destroy", "--force"], check=True)
print("Cleaning up...")
if NODE_COUNT > 1:
subprocess.run(["aerolab", "cluster", "stop"], check=True)
subprocess.run(["aerolab", "cluster", "destroy", "--force"], check=True)

0 comments on commit 0c57d19

Please sign in to comment.