Adding basic unit test for refactoring work