Add missing tests from #1372

[rly]

Add missing tests from #1372 and change a check into an assertion to clarify the expectation.

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 92.90%. Comparing base (42f7311) to head (b9cd125).
⚠️ Report is 1 commits behind head on dev.

Additional details and impacted files

@@            Coverage Diff             @@
##              dev    #1397      +/-   ##
==========================================
+ Coverage   92.86%   92.90%   +0.04%     
==========================================
  Files          41       41              
  Lines        9919     9919              
  Branches     2027     2026       -1     
==========================================
+ Hits         9211     9215       +4     
+ Misses        433      432       -1     
+ Partials      275      272       -3     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.