feat(tests): add more unit tests