feature: add more tests.