address PR feedback

Author: ohmayr

.generator/cli.py

@@ -20,7 +20,17 @@
 
 logger = logging.getLogger()
 
-LIBRARIAN = "/librarian"
+LIBRARIAN_DIR = "/librarian"
+GENERATOR_DIR = "/.generator"
+GENERATE_REQUEST_FILE = "generate-request.json"
+
+
+def _get_base_dir():
+    """Returns the correct base directory based on the environment."""
+    environment = os.getenv("PY_GENERATOR_ENV", "production").lower()
+    if environment == "test":
+        return GENERATOR_DIR
+    return LIBRARIAN_DIR
 
 
 # Helper function that reads a json file path and returns the loaded json content.
@@ -38,9 +48,8 @@ def handle_generate(dry_run=False):
 
     # Read a generate-request.json file
     if not dry_run:
-        request_path = f"{LIBRARIAN}/generate-request.json"
         try:
-            request_data = _read_json_file(request_path)
+            request_data = _read_json_file(f"{_get_base_dir()}/{GENERATE_REQUEST_FILE}")
         except Exception as e:
             logger.error(e)
             sys.exit(1)

.librarian/generate-request.json .generator/generate-request.json.librarian/generate-request.json renamed to .generator/generate-request.json

@@ -7,4 +7,3 @@
       }
     ]
   }
-  

.generator/test_cli.py

@@ -15,7 +15,25 @@
 import pytest
 import json
 
-from cli import _read_json_file, handle_generate, handle_build, handle_configure
+from unittest.mock import mock_open
+
+from cli import (
+    _read_json_file,
+    handle_generate,
+    handle_build,
+    handle_configure,
+    GENERATOR_DIR,
+    GENERATE_REQUEST_FILE,
+)
+
+
+# # The fixture is defined directly in the test file
+@pytest.fixture(autouse=True)
+def set_test_environment(monkeypatch):
+    """
+    Sets the environment variable only for tests in this file.
+    """
+    monkeypatch.setenv("PY_GENERATOR_ENV", "test")
 
 
 def test_handle_configure_dry_run():