SNOW-1762538 add application detection for Jupyter notebook and Snowbook

Author: sfc-gh-mmishchenko

src/snowflake/connector/connection.py

@@ -2294,3 +2294,5 @@ def _detect_application() -> None | str:
             for jpmod in ("ipykernel", "jupyter_core", "jupyter_client")
         ):
             return "jupyter_notebook"
+        if "snowbooks" in sys.modules:
+            return "snowflake_notebook"

test/unit/test_connection.py

@@ -194,12 +194,23 @@ def test_partner_env_var(mock_post_requests):
 
 
 @pytest.mark.skipolddriver
-def test_imported_module(mock_post_requests):
-    with patch.dict(sys.modules, {"streamlit": "foo"}):
-        assert fake_connector().application == "streamlit"
+@pytest.mark.parametrize(
+    "sys_modules,application",
+    [
+        ({"streamlit": None}, "streamlit"),
+        (
+            {"ipykernel": None, "jupyter_core": None, "jupyter_client": None},
+            "jupyter_notebook",
+        ),
+        ({"snowbooks": None}, "snowflake_notebook"),
+    ],
+)
+def test_imported_module(mock_post_requests, sys_modules, application):
+    with patch.dict(sys.modules, sys_modules):
+        assert fake_connector().application == application
 
     assert (
-        mock_post_requests["data"]["CLIENT_ENVIRONMENT"]["APPLICATION"] == "streamlit"
+        mock_post_requests["data"]["CLIENT_ENVIRONMENT"]["APPLICATION"] == application
     )