Chore: Add assertions to fix mypy errors for trino and errors for latest duckdb (#5724)

Author: themisvaltinos

sqlmesh/core/config/connection.py

@@ -2013,7 +2013,17 @@ def _static_connection_kwargs(self) -> t.Dict[str, t.Any]:
             OAuth2Authentication,
         )
 
+        auth: t.Optional[
+            t.Union[
+                BasicAuthentication,
+                KerberosAuthentication,
+                OAuth2Authentication,
+                JWTAuthentication,
+                CertificateAuthentication,
+            ]
+        ] = None
         if self.method.is_basic or self.method.is_ldap:
+            assert self.password is not None  # for mypy since validator already checks this
             auth = BasicAuthentication(self.user, self.password)
         elif self.method.is_kerberos:
             if self.keytab:
@@ -2032,11 +2042,12 @@ def _static_connection_kwargs(self) -> t.Dict[str, t.Any]:
         elif self.method.is_oauth:
             auth = OAuth2Authentication()
         elif self.method.is_jwt:
+            assert self.jwt_token is not None
             auth = JWTAuthentication(self.jwt_token)
         elif self.method.is_certificate:
+            assert self.client_certificate is not None
+            assert self.client_private_key is not None
             auth = CertificateAuthentication(self.client_certificate, self.client_private_key)
-        else:
-            auth = None
 
         return {
             "auth": auth,

tests/core/integration/test_aux_commands.py

@@ -287,20 +287,20 @@ def test_destroy(copy_to_temp_path):
 
     # Validate tables have been deleted as well
     with pytest.raises(
-        Exception, match=r"Catalog Error: Table with name model_two does not exist!"
+        Exception, match=r"Catalog Error: Table with name.*model_two.*does not exist"
     ):
         context.fetchdf("SELECT * FROM db_1.first_schema.model_two")
     with pytest.raises(
-        Exception, match=r"Catalog Error: Table with name model_one does not exist!"
+        Exception, match=r"Catalog Error: Table with name.*model_one.*does not exist"
     ):
         context.fetchdf("SELECT * FROM db_1.first_schema.model_one")
 
     with pytest.raises(
-        Exception, match=r"Catalog Error: Table with name model_two does not exist!"
+        Exception, match=r"Catalog Error: Table with name.*model_two.*does not exist"
     ):
         context.engine_adapters["second"].fetchdf("SELECT * FROM db_2.second_schema.model_two")
     with pytest.raises(
-        Exception, match=r"Catalog Error: Table with name model_one does not exist!"
+        Exception, match=r"Catalog Error: Table with name.*model_one.*does not exist"
     ):
         context.engine_adapters["second"].fetchdf("SELECT * FROM db_2.second_schema.model_one")