Clean up test schema in connector tests

Author: anusudarsan

testing/trino-testing/src/main/java/io/trino/testing/BaseConnectorSmokeTest.java

@@ -375,11 +375,15 @@ public void testCreateSchema()
             return;
         }
 
-        assertUpdate(createSchemaSql(schemaName));
-        assertThat(query("SHOW SCHEMAS"))
-                .skippingTypesCheck()
-                .containsAll(format("VALUES '%s', '%s'", getSession().getSchema().orElseThrow(), schemaName));
-        assertUpdate("DROP SCHEMA " + schemaName);
+        try {
+            assertUpdate(createSchemaSql(schemaName));
+            assertThat(query("SHOW SCHEMAS"))
+                    .skippingTypesCheck()
+                    .containsAll(format("VALUES '%s', '%s'", getSession().getSchema().orElseThrow(), schemaName));
+        }
+        finally {
+            assertUpdate("DROP SCHEMA IF EXISTS " + schemaName);
+        }
     }
 
     @Test

testing/trino-testing/src/main/java/io/trino/testing/BaseConnectorTest.java

@@ -3527,11 +3527,15 @@ public void testCreateSchemaWithNonLowercaseOwnerName()
                 .setIdentity(Identity.ofUser("ADMIN"))
                 .build();
         String schemaName = "test_schema_create_uppercase_owner_name_" + randomNameSuffix();
-        assertUpdate(newSession, createSchemaSql(schemaName));
-        assertThat(query(newSession, "SHOW SCHEMAS"))
-                .skippingTypesCheck()
-                .containsAll(format("VALUES '%s'", schemaName));
-        assertUpdate(newSession, "DROP SCHEMA " + schemaName);
+        try {
+            assertUpdate(newSession, createSchemaSql(schemaName));
+            assertThat(query(newSession, "SHOW SCHEMAS"))
+                    .skippingTypesCheck()
+                    .containsAll(format("VALUES '%s'", schemaName));
+        }
+        finally {
+            assertUpdate(newSession, "DROP SCHEMA IF EXISTS " + schemaName);
+        }
     }
 
     @Test