Added unit test
Merge pull request #23 from cdap-guides/feature/unit-test

Added unit test