add unit tests.
Merge pull request #21 from hansode/feature-shunit2