From 07282d95e0f9d5f24a2a6aebae51f76a093b6084 Mon Sep 17 00:00:00 2001
From: dotCooCoo <RobertLeeLW@gmail.com>
Date: Sun, 1 Nov 2026 08:38:15 -0800
Subject: [PATCH] Variables: a graph variable's data set applies its CSV as
 graph.setData does

Choosing a data set gave a graph variable's CSV to the graph without the
two steps graph.setData takes after new data: rows with neither a label
nor a value were kept and drawn as empty slots, and a right-axis series
the new data no longer had stayed assigned, so a series added by a later
row started on the right axis. The data set now goes through the same
steps.
---
 crates/engine/src/cmd/graph/data.rs  | 20 +++++++++++---------
 crates/engine/src/tests/variables.rs | 28 ++++++++++++++++++++++++++++
 2 files changed, 39 insertions(+), 9 deletions(-)

diff --git a/crates/engine/src/cmd/graph/data.rs b/crates/engine/src/cmd/graph/data.rs
index e81f24feb..362f988f0 100644
--- a/crates/engine/src/cmd/graph/data.rs
+++ b/crates/engine/src/cmd/graph/data.rs
@@ -86,7 +86,8 @@ pub(super) fn to_csv(g: &GraphSpec) -> String {
     g.to_csv()
 }
 
-/// Apply CSV graph data to the graph root `id` and rebuild its art (Variables data merge).
+/// Apply CSV graph data to the graph root `id` and rebuild its art (Variables data merge), as `graph.setData` with
+/// `csv` does.
 pub(crate) fn apply_csv(d: &mut Document, id: NodeId, csv: &str) -> Result<()> {
     let Some(node) = d.node(id) else {
         return Err(EngineError::NoNode(id));
@@ -94,13 +95,17 @@ pub(crate) fn apply_csv(d: &mut Document, id: NodeId, csv: &str) -> Result<()> {
     let Some(mut spec) = node.extras().graph.as_deref().cloned() else {
         return Err(bad("graph.applyCsv", format!("node {} is not a graph", id.0)));
     };
-    let (series, categories, rows) = parse_csv(csv);
-    spec.series = series;
-    spec.categories = categories;
-    spec.set_cells(rows);
+    apply_data(&mut spec, &json!({ "csv": csv }));
+    keep_right_series(&mut spec);
     regenerate(d, id, spec)
 }
 
+/// Drop right-axis series the data no longer has, so a series added later starts on the left.
+fn keep_right_series(g: &mut GraphSpec) {
+    let count = g.rows.iter().map(Vec::len).max().unwrap_or(0);
+    g.right_series.retain(|i| *i < count);
+}
+
 pub(super) fn apply_data(g: &mut GraphSpec, p: &Value) -> bool {
     let mut changed = false;
     if let Some(csv) = str_param(p, "csv") {
@@ -185,10 +190,7 @@ pub(super) fn set_data(s: &mut Session, p: &Value) -> Result<Value> {
         // Blank cells come back as null, so the rows can be edited and sent back as they are.
         return Ok(json!({ "csv": to_csv(&spec), "series": spec.series, "categories": spec.categories, "rows": spec.cells() }));
     }
-    // Series assigned to the right axis that the new data no longer has are dropped, so a series added later starts
-    // on the left.
-    let count = spec.rows.iter().map(Vec::len).max().unwrap_or(0);
-    spec.right_series.retain(|i| *i < count);
+    keep_right_series(&mut spec);
     s.edit("Graph Data", |d, _| regenerate(d, id, spec))?;
     Ok(json!({ "id": id.0 }))
 }
diff --git a/crates/engine/src/tests/variables.rs b/crates/engine/src/tests/variables.rs
index bdacd0e1b..fb252d854 100644
--- a/crates/engine/src/tests/variables.rs
+++ b/crates/engine/src/tests/variables.rs
@@ -1632,6 +1632,34 @@ fn locked_variables_refuse_a_library_that_deletes_or_defines_variables() {
     assert_eq!(vars.datasets.iter().map(|d| d.name.as_str()).collect::<Vec<_>>(), ["B"]);
 }
 
+/// A graph variable's data set applies its CSV as `graph.setData` does: a right-axis series the row's data no longer
+/// has is dropped, and a row with neither a label nor a value is not drawn as an empty slot.
+#[test]
+fn graph_variable_applies_csv_as_set_data() {
+    let mut s = session();
+    let create = json!({"type": "column", "x": 10, "y": 10, "width": 120, "height": 80, "csv": ",a,b\nQ1,10,1000\nQ2,5,500"});
+    let id = s.execute("graph.create", &create).unwrap()["id"].as_u64().unwrap();
+    let other = s.execute("graph.create", &create).unwrap()["id"].as_u64().unwrap();
+    for g in [id, other] {
+        s.execute("graph.setType", &json!({"id": g, "seriesIndexes": [1], "valueAxis": "right"})).unwrap();
+    }
+    s.execute("select.set", &json!({"ids": [id]})).unwrap();
+    s.execute("variable.makeGraphDynamic", &json!({})).unwrap();
+    let name = s.execute("variable.list", &json!({})).unwrap()["variables"][0]["name"].as_str().unwrap().to_string();
+    let spec = |s: &Session, g: u64| s.doc().unwrap().doc.node(NodeId(g)).unwrap().extras().graph.as_deref().unwrap().clone();
+    for (row, csv) in [("One", ",a\nQ1,10\nQ2,5"), ("Blank", ",a\nQ1,10\n,\nQ2,5")] {
+        s.execute("dataset.new", &json!({"name": row, "values": {name.clone(): csv}})).unwrap();
+        s.execute("dataset.select", &json!({"name": row})).unwrap();
+        s.execute("graph.setData", &json!({"id": other, "csv": csv})).unwrap();
+        let (merged, set) = (spec(&s, id), spec(&s, other));
+        assert_eq!(merged.right_series, set.right_series, "{row}");
+        assert_eq!(merged.categories, set.categories, "{row}");
+        assert_eq!(merged.cells(), set.cells(), "{row}");
+    }
+    assert!(spec(&s, id).right_series.is_empty());
+    assert_eq!(spec(&s, id).categories, ["Q1", "Q2"]);
+}
+
 #[test]
 fn graph_variable_applies_csv_and_captures() {
     let mut s = session();