Add more tests

[tsalo]

I just let Cursor write the tests. Will need to review them.

[codecov-commenter]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 37.98%. Comparing base (7a5bd44) to head (7c223d4).

Additional details and impacted files

@@            Coverage Diff             @@
##             main      #23      +/-   ##
==========================================
+ Coverage   34.10%   37.98%   +3.88%     
==========================================
  Files          13       13              
  Lines         953      953              
  Branches      134      134              
==========================================
+ Hits          325      362      +37     
+ Misses        599      566      -33     
+ Partials       29       25       -4     

☔ 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.