Add unit test cases for enhancing line coverage