fix sql-facade

Author: sfc-gh-pczajka

RELEASE-NOTES.md

@@ -22,6 +22,7 @@
 * Added `--encryption` flag to `snow stage create` command defining the type of encryption for all files on the stage.
 
 ## Fixes and improvements
+* Fix `use` commands error if current database is not set.
 
 
 # v3.8.3

src/snowflake/cli/_plugins/nativeapp/sf_sql_facade.py

@@ -159,7 +159,10 @@ def _use_object_optional(self, object_type: UseObjectType, name: str | None):
             )
 
         try:
-            prev_obj = to_identifier(current_obj_result_row[0])
+            if current_obj_result_row[0]:
+                prev_obj = to_identifier(current_obj_result_row[0])
+            else:
+                prev_obj = None
         except IndexError:
             prev_obj = None
 

tests/nativeapp/test_sf_sql_facade.py

@@ -841,6 +841,24 @@ def test_use_role_same_id(mock_execute_query, mock_cursor, new_role, current_rol
     assert mock_execute_query.mock_calls == expected
 
 
+def test_use_role_current_role_empty(mock_execute_query, mock_cursor):
+    side_effects, expected = mock_execute_helper(
+        [
+            (
+                mock_cursor([(None,)], []),
+                mock.call("select current_role()"),
+            ),
+            (None, mock.call('use role "test role"')),
+        ]
+    )
+    mock_execute_query.side_effect = side_effects
+
+    with sql_facade._use_role_optional("test role"):  # noqa: SLF001
+        pass
+
+    assert mock_execute_query.mock_calls == expected
+
+
 @pytest.mark.parametrize(
     "old_db, expected_old_db",
     [("old_db", "old_db"), ("old db", '"old db"')],
@@ -888,6 +906,24 @@ def test_use_db_same_id(mock_execute_query, mock_cursor, new_db, current_db):
     assert mock_execute_query.mock_calls == expected
 
 
+def test_use_db_current_db_empty(mock_execute_query, mock_cursor):
+    side_effects, expected = mock_execute_helper(
+        [
+            (
+                mock_cursor([(None,)], []),
+                mock.call("select current_database()"),
+            ),
+            (None, mock.call('use database "new db"')),
+        ]
+    )
+    mock_execute_query.side_effect = side_effects
+
+    with sql_facade._use_database_optional("new db"):  # noqa: SLF001
+        pass
+
+    assert mock_execute_query.mock_calls == expected
+
+
 @pytest.mark.parametrize(
     "old_schema, expected_old_schema",
     [("old_schema", "old_schema"), ("old schema", '"old schema"')],
@@ -941,6 +977,24 @@ def test_use_schema_same_id(
     assert mock_execute_query.mock_calls == expected
 
 
+def test_use_schema_current_schema_empty(mock_execute_query, mock_cursor):
+    side_effects, expected = mock_execute_helper(
+        [
+            (
+                mock_cursor([(None,)], []),
+                mock.call("select current_schema()"),
+            ),
+            (None, mock.call('use schema "new schema"')),
+        ]
+    )
+    mock_execute_query.side_effect = side_effects
+
+    with sql_facade._use_schema_optional("new schema"):  # noqa: SLF001
+        pass
+
+    assert mock_execute_query.mock_calls == expected
+
+
 @pytest.mark.parametrize(
     "error_raised, error_caught, error_message",
     [