Updated test unit test cases