Tests: write unit tests

[joshjung]

This task should not be too difficult, but we need to write Jest unit tests to get to full coverage.