Tech debt: Add extensive unit test coverage of existing components

[TheIllusionistMirage]

There is a lack of proper unit test coverage since the initial focus was to get a prototype up and running quickly. Extensive unit test coverage is needed to ensure health of the code base.