Flaky test fix

[AishwaryaKalloli]

Split of original CL: # 900

Motivation and Context

Depending on time for the test was resulting in flaky behavior, so made the testing method independent of time.

How Has This Been Tested?

Test passes.

Breaking Changes

No

Types of changes

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to change)
• Documentation update

Checklist

• I have read the MCP Documentation
• My code follows the repository's style guidelines
• New and existing tests pass locally
• I have added appropriate error handling
• I have added or updated documentation as needed

[ihrpr]

Thank you! Really appreciate the additional effort to split into a separate PR.