add unit test - 2 cases