Add tests for the database connection wrapper