Add tests for Snowflake connection and query execution

Author: khavnekar-y

tests/testing_snowflake_connection.py

@@ -3,33 +3,40 @@
 import os
 from dotenv import load_dotenv
 
-load_dotenv()  # Load environment variables from a .env file (if used locally)
-
+load_dotenv()
 @pytest.fixture
 def snowflake_connection():
     """Fixture to establish a connection to Snowflake."""
     conn = None
     try:
-        # Print the connection parameters for debugging
-        print("Connecting to Snowflake with the following parameters:")
-        print(f"User: {os.getenv('SNOWFLAKE_USER')}")
-        print(f"Account: {os.getenv('SNOWFLAKE_ACCOUNT')}")
-        print(f"Warehouse: {os.getenv('SNOWFLAKE_WAREHOUSE')}")
-        print(f"Database: {os.getenv('SNOWFLAKE_DATABASE')}")
-        print(f"Schema: {os.getenv('SNOWFLAKE_SCHEMA')}")
-
         conn = snowflake.connector.connect(
             user=os.getenv("SNOWFLAKE_USER"),
-            password=os.getenv("SNOWFLAKE_PASSWORD"),
             account=os.getenv("SNOWFLAKE_ACCOUNT"),
+            password=os.getenv("SNOWFLAKE_PASSWORD"),
             warehouse=os.getenv("SNOWFLAKE_WAREHOUSE"),
             database=os.getenv("SNOWFLAKE_DATABASE"),
             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:
-        if conn is not None:
-            conn.close()  # Close the connection only if it was created
+        if conn:
+            conn.close()
+
+def test_snowflake_connection(snowflake_connection):
+    """Test if Snowflake connection is successful."""
+    assert snowflake_connection is not None
+    assert snowflake_connection.is_closed() == False
+
+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)
+    finally:
+        cursor.close()
+
+if __name__ == "__main__":
+    pytest.main()