feat: add support for cloud spec

scenario/mocking.py

@@ -21,7 +21,7 @@
 )
 
 from ops import JujuVersion, pebble
-from ops.model import ModelError, RelationNotFoundError
+from ops.model import CloudSpec, ModelError, RelationNotFoundError
 from ops.model import Secret as Secret_Ops  # lol
 from ops.model import (
     SecretInfo,
@@ -631,6 +631,13 @@ def resource_get(self, resource_name: str) -> str:
                 f"resource {resource_name} not found in State. please pass it.",
             )
 
+    def credential_get(self) -> CloudSpec:
+        if not self._state.cloud_spec:
+            raise ModelError(
+                "ERROR cloud spec is empty, initialise it with `scenario.State(cloud_spec=...)`",
+            )
+        return self._state.cloud_spec
+
 
 class _MockPebbleClient(_TestingPebbleClient):
     def __init__(

scenario/state.py

@@ -32,7 +32,7 @@
 import yaml
 from ops import pebble
 from ops.charm import CharmBase, CharmEvents
-from ops.model import SecretRotate, StatusBase
+from ops.model import CloudSpec, SecretRotate, StatusBase
 
 from scenario.logger import logger as scenario_logger
 
@@ -919,6 +919,7 @@ class State(_DCBase):
     """Status of the unit."""
     workload_version: str = ""
     """Workload version."""
+    cloud_spec: Optional[CloudSpec] = None
 
     def __post_init__(self):
         for name in ["app_status", "unit_status"]:

tests/test_e2e/test_cloud_spec.py

@@ -0,0 +1,40 @@
+import ops
+import pytest
+
+import scenario
+
+
+@pytest.fixture(scope="function")
+def mycharm():
+    class MyCharm(ops.CharmBase):
+        def __init__(self, framework: ops.Framework):
+            super().__init__(framework)
+            for evt in self.on.events().values():
+                self.framework.observe(evt, self._on_event)
+
+        def _on_event(self, event):
+            pass
+
+    return MyCharm
+
+
+def test_get_cloud_spec(mycharm):
+    cloud_spec = ops.model.CloudSpec.from_dict(
+        {
+            "name": "localhost",
+            "type": "lxd",
+            "endpoint": "https://127.0.0.1:8443",
+            "credential": {
+                "auth-type": "certificate",
+                "attrs": {
+                    "client-cert": "foo",
+                    "client-key": "bar",
+                    "server-cert": "baz",
+                },
+            },
+        }
+    )
+
+    ctx = scenario.Context(mycharm, meta={"name": "foo"})
+    with ctx.manager("start", scenario.State(cloud_spec=cloud_spec)) as mgr:
+        assert mgr.charm.model.get_cloud_spec() == cloud_spec