Improve test coverage

[philipch07]

Description

Second part of improving test coverage, aiming for >= 80%

Reference issue

Fixes #193 (didn't mean to close it with the previous pr)

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 82.78%. Comparing base (cc629db) to head (6959fc1).
⚠️ Report is 1 commits behind head on master.

Additional details and impacted files

@@            Coverage Diff             @@
##           master     #235      +/-   ##
==========================================
+ Coverage   74.95%   82.78%   +7.83%     
==========================================
  Files          12       12              
  Lines        1685     1685              
==========================================
+ Hits         1263     1395     +132     
+ Misses        336      221     -115     
+ Partials       86       69      -17     

Flag	Coverage Δ
go	82.78% <ø> (+7.83%)	⬆️

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

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