test: Add unit test