Add more test cases to mockDataSource