fix test

Author: ohmayr

.generator/cli.py

@@ -775,7 +775,9 @@ def _generate_api(
         _run_protoc_command(command, source)
 
         # 3. Determine staging location
-        staging_child_directory = _get_staging_child_directory(api_path, is_proto_only_library)
+        staging_child_directory = _get_staging_child_directory(
+            api_path, is_proto_only_library
+        )
         staging_dir = os.path.join(
             output, "owl-bot-staging", library_id, staging_child_directory
         )

.generator/test_cli.py

@@ -1440,4 +1440,6 @@ def test_stage_gapic_library(mocker):
     _stage_gapic_library(tmp_dir, staging_dir)
 
     # Assertion: Check copytree was called exactly once with the correct arguments
-    mock_shutil_copytree.assert_called_once_with(tmp_dir, staging_dir)
+    mock_shutil_copytree.assert_called_once_with(
+        tmp_dir, staging_dir, dirs_exist_ok=True
+    )