test: move wait statement to shared code

[JComins000]

Ticket

Description

Test Plan

Checklist

• Changes have been manually QA'd
• New features have been approved by the corresponding PM
• User-facing API changes have the "User-facing API Change" label
• Release notes have been added as a separate file under docs/release-notes/
  See Release Note for details.
• Licenses have been included for new code which was copied and/or modified from any external code

[netlify]

✅ Deploy Preview for determined-ui ready!

Name	Link
🔨 Latest commit	9db49a3
🔍 Latest deploy log	https://app.netlify.com/sites/determined-ui/deploys/669ffc64c571890008316f30
😎 Deploy Preview	https://deploy-preview-9701--determined-ui.netlify.app
📱 Preview on mobile	Toggle QR Code... Use your smartphone camera to open QR code link.

---

To edit notification comments on pull requests, go to your Netlify site configuration.

[djanicekpach]

lgtm. Approved assuming tests pass.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 48.99%. Comparing base (8776e35) to head (9db49a3).

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #9701      +/-   ##
==========================================
- Coverage   53.55%   48.99%   -4.57%     
==========================================
  Files        1255      932     -323     
  Lines      152711   123481   -29230     
  Branches     3293     3294       +1     
==========================================
- Hits        81787    60495   -21292     
+ Misses      70774    62836    -7938     
  Partials      150      150              

Flag	Coverage Δ
harness	?
web	52.05% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

see 323 files with indirect coverage changes