Add test cases for result set