add unit test for diff scenario