Started adding test cases for database handler