test: Add UnitTest