(chore): unit tests

[markharding]

[jberrong]

Hi Mark,

Do you happen to have any acceptance criteria/lists on tests needed? Would be glad to contribute to this.

Thanks.
-Jonathan

[markharding]

Hi Jonathan,

That would be part of this task. Want to take it on?

Mark