update(tests): add more unit tests