Added option for alternative server config server-profiles

Author: kzscisoft

simvue/config/user.py

@@ -55,10 +55,14 @@ class SimvueConfiguration(pydantic.BaseModel):
     server: ServerSpecifications = pydantic.Field(
         ..., description="Specifications for Simvue server"
     )
+    profiles: dict[str, ServerSpecifications] = pydantic.Field(
+        default_factory=dict[str, ServerSpecifications]
+    )
     run: DefaultRunSpecifications = DefaultRunSpecifications()
     offline: OfflineSpecifications = OfflineSpecifications()
     metrics: MetricsSpecifications = MetricsSpecifications()
     eco: EcoConfig = EcoConfig()
+    current_profile: str = "default"
 
     @classmethod
     def _load_pyproject_configs(cls) -> dict | None:
@@ -158,6 +162,7 @@ def fetch(
         mode: typing.Literal["offline", "online", "disabled"],
         server_url: str | None = None,
         server_token: str | None = None,
+        profile: str = "default",
     ) -> "SimvueConfiguration":
         """Retrieve the Simvue configuration from this project
 
@@ -190,10 +195,18 @@ def fetch(
 
         except FileNotFoundError:
             if not server_token or not server_url:
-                _config_dict = {"server": {}}
+                _config_dict |= {"server": {}}
                 logger.debug("No config file found, checking environment variables")
 
-        _config_dict["server"] = _config_dict.get("server", {})
+        if profile == "default":
+            _config_dict["server"] = _config_dict.get("server", {})
+        elif not _config_dict.get("profiles", {}).get(profile):
+            raise RuntimeError(
+                f"Cannot load server configuration for '{profile}', "
+                "profile not found in configurations."
+            )
+        else:
+            _config_dict["server"] = _config_dict["profiles"][profile]
 
         _config_dict["offline"] = _config_dict.get("offline", {})
 
@@ -234,7 +247,7 @@ def fetch(
         _config_dict["server"]["url"] = _server_url
         _config_dict["run"]["mode"] = _run_mode
 
-        return SimvueConfiguration(**_config_dict)
+        return SimvueConfiguration(profile=profile, **_config_dict)
 
     @classmethod
     @functools.lru_cache

tests/functional/test_config.py

@@ -21,18 +21,25 @@
     "use_args", (True, False),
     ids=("args", "no_args")
 )
+@pytest.mark.parametrize(
+    "profile", ("default", "other"),
+    ids=("default_profile", "alt_profile")
+)
 def test_config_setup(
     use_env: bool,
-    use_file: str | None,
+    use_file: typing.Literal["basic", "extended", "pyproject.toml"] | None,
     use_args: bool,
+    profile: typing.Literal["default", "other"],
     monkeypatch: pytest.MonkeyPatch,
     mocker: pytest_mock.MockerFixture
 ) -> None:
     _token: str = f"{uuid.uuid4()}".replace('-', '')
     _other_token: str = f"{uuid.uuid4()}".replace('-', '')
     _arg_token: str = f"{uuid.uuid4()}".replace('-', '')
+    _alt_token: str = f"{uuid.uuid4()}".replace("-", "")
     _url: str = "https://simvue.example.com/"
     _other_url: str = "http://simvue.example.com/"
+    _alt_url: str = "https://simvue-dev.example.com/"
     _arg_url: str = "http://simvue.example.io/"
     _description: str = "test case for runs"
     _description_ppt: str = "test case for runs using pyproject.toml"
@@ -75,6 +82,10 @@ def test_config_setup(
                     url = "{_url}"
                     token = "{_token}"
 
+                    [profiles.other]
+                    url = "{_alt_url}"
+                    token = "{_alt_token}"
+
                     [offline]
                     cache = "{_windows_safe}"
                     """
@@ -104,14 +115,22 @@ def _mocked_find(file_names: list[str], *_, ppt_file=_ppt_file, conf_file=_confi
                 simvue.config.user.SimvueConfiguration.fetch(mode="online")
             return
         elif use_args:
-            _config = simvue.config.user.SimvueConfiguration.fetch(
+            _config: SimvueConfiguration = simvue.config.user.SimvueConfiguration.fetch(
                 server_url=_arg_url,
                 server_token=_arg_token,
                 mode="online"
             )
+        elif profile == "other":
+            if not use_file:
+                with pytest.raises(RuntimeError):
+                    _ = simvue.config.user.SimvueConfiguration.fetch(mode="online", profile="other")
+                return
+            else:
+                _config = simvue.config.user.SimvueConfiguration.fetch(mode="online", profile="other")
+
         else:
             _config = simvue.config.user.SimvueConfiguration.fetch(mode="online")
-            
+
         if use_file and use_file != "pyproject.toml":
             assert _config.config_file() == _config_file
 
@@ -121,6 +140,10 @@ def _mocked_find(file_names: list[str], *_, ppt_file=_ppt_file, conf_file=_confi
         elif use_args:
             assert _config.server.url == f"{_arg_url}api"
             assert _config.server.token.get_secret_value() == _arg_token
+        elif use_file and profile == "other":
+            assert _config.server.url == f"{_alt_url}api"
+            assert _config.server.token.get_secret_value() == _alt_token
+            assert f"{_config.offline.cache}" == temp_d
         elif use_file and use_file != "pyproject.toml":
             assert _config.server.url == f"{_url}api"
             assert _config.server.token.get_secret_value() == _token