chore: complete unit tests