[Unit Tests]: Add Unit tests to code

[cwisniew]

Describe the problem

There is a lot -- well ok most of -- the code base that is not covered by unit tests. This is a placeholder that can group together unit test tasks instead of having to create and fill in a new issue for each unit test (trying to reduce friction for anyone wanting to add tests quickly)

The improvement you'd like to see

More unit tests.

Expected Benefits

More tests mean less chance of breaking stuff when we make changes.mean

Additional Context

If you want to add a test just comment here on what you want to add and I will add it to the list