Description of changes:
Instead of waiting a fixed 5 seconds for container to restart (which sometimes could take longer, leading to a failed test), the test now polls the container until its restarted.
Testing done:
Passed all unit and e2e tests
E2E tests passed on 5/5 attempts
[X] I've reviewed the guidance in CONTRIBUTING.md
License Acceptance
By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.
Issue #, if available:
Description of changes: Instead of waiting a fixed 5 seconds for container to restart (which sometimes could take longer, leading to a failed test), the test now polls the container until its restarted.
Testing done: Passed all unit and e2e tests E2E tests passed on 5/5 attempts
License Acceptance
By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.