Update unit test.
Merge pull request #7 from zpbx/dev-test

Update unit test.