Updated unit tests with missing test cases