Enhance Snowflake connection fixture with error handling and improved assertions

Author: khavnekar-y

tests/testing_snowflake_connection.py

@@ -2,7 +2,9 @@
 import snowflake.connector
 import os
 from dotenv import load_dotenv
-load_dotenv()
+
+load_dotenv()  # Load environment variables from a .env file (if used locally)
+
 @pytest.fixture
 def snowflake_connection():
     """Fixture to establish a connection to Snowflake."""
@@ -17,24 +19,28 @@ def snowflake_connection():
             schema=os.getenv("SNOWFLAKE_SCHEMA")
         )
         yield conn
+    except Exception as e:
+        print(f"Error connecting to Snowflake: {e}")  # Print error for debugging
+        yield None  # Yield None if connection fails
     finally:
-        conn.close()
+        if conn is not None:
+            conn.close()  # Close the connection only if it was created
 
 def test_snowflake_connection(snowflake_connection):
     """Test if Snowflake connection is successful."""
-    assert snowflake_connection is not None
-    assert snowflake_connection.is_closed() == False
+    assert snowflake_connection is not None, "Connection should not be None"
+    assert snowflake_connection.is_closed() == False, "Connection should be open"
 
 def test_snowflake_query_execution(snowflake_connection):
     """Test if a simple query executes successfully."""
     cursor = snowflake_connection.cursor()
     try:
         cursor.execute("SELECT CURRENT_VERSION()")
         result = cursor.fetchone()
-        assert result is not None
-        assert isinstance(result[0], str)
+        assert result is not None, "Query result should not be None"
+        assert isinstance(result[0], str), "Result should be a string"
     finally:
         cursor.close()
 
 if __name__ == "__main__":
-    pytest.main()
+    pytest.main()