fixed scenarios

lib/charms/grafana_k8s/v0/grafana_source.py

@@ -438,7 +438,10 @@ def get_source_uids(self) -> Dict[str, Dict[str, str]]:
         for rel in self._charm.model.relations.get(self._relation_name, []):
             if not rel:
                 continue
-            uids[rel.app.name] = json.loads(rel.data[rel.app]["datasource_uids"])
+            ds_uids_raw = rel.data[rel.app].get("datasource_uids")
+            if not ds_uids_raw:
+                continue
+            uids[rel.app.name] = json.loads(ds_uids_raw)
         return uids
 
     def _set_sources_from_event(self, event: RelationJoinedEvent) -> None:

tests/scenario/conftest.py

@@ -28,6 +28,12 @@ def prometheus_charm():
         yield PrometheusCharm
 
 
+@pytest.fixture(autouse=True)
+def disable_charm_tracing_buffer():
+    with patch("charms.tempo_coordinator_k8s.v0.charm_tracing._Buffer.save", lambda *_: None):
+        yield
+
+
 @pytest.fixture(scope="function")
 def context(prometheus_charm):
     return Context(charm_type=prometheus_charm, juju_version="3.0.3")

tests/scenario/test_server_scheme.py

@@ -34,9 +34,9 @@ def initial_state(self, context, fqdn, leader) -> State:
 
             # Add relations
             rels = [
-                Relation("self-metrics-endpoint", relation_id=10),  # external self-monitoring
-                Relation("grafana-source", relation_id=11),  # grafana
-                Relation("receive-remote-write", relation_id=12),  # grafana-agent
+                Relation("self-metrics-endpoint"),  # external self-monitoring
+                Relation("grafana-source"),  # grafana
+                Relation("receive-remote-write"),  # grafana-agent
             ]
             for rel in rels:
                 state = add_relation_sequence(context, state, rel)