Tests: Update test_database to update for new behavior.