Add more unit tests!

[amaury1093]

@mdshamoon laid out the tooling around unit tests in #69 💪 We need to add more unit tests though.

If you are an external contributor, please free to post a message here and choose a couple of components to add unit tests too.