Added timeout for flaky test

[ajaysubra]

Description

This is inline with the other existing tests and the suggestion from the failure message.

Check List

• [ ] Are you changing anything with the public API?
• [ ] Have you tested this change on real device?
• [ ] Are your changes backwards compatible with previous SDK Versions?
• [ ] Have you added unit test coverage for your changes?
• [ ] Have you verified that your changes are compatible with all the operating system version this SDK currently supports?

Manual Test Plan

1.

Supporting Materials

[ajaysubra]

Thanks for merging this @evan-masseau