[Misc] Refactor unit tests

[zsh-eng]

Currently, there are bugs in some of the unit tests which actually call the database.

Unit tests should not be calling the database directly.
Add the appropriate mocks and extract out common logic for tests.