Getting test cases for DB working correctly