Add unittests

When we open-sourced this tool, the tests are omitted because it'll introduce another dependency to gtest. But it's worth the effort to add the test to this package too to track regression.