chore(v2): clean up mock tests

## Summary

What:

The mock tests have some unit tests that aren't grouped (the last 3 or something). They should be put into a group for better test result output. There is a `TODO` in `mock_test.ts` line 283.

Why:

Because the test output is ugly IMHO and we should make it somewhat pretty 💀 

## Acceptance Criteria

No changes in behavior. Just move the tests.
