update tests

simularium · Sep 22, 2023 · 33afb92 · 33afb92
1 parent 1531bf2
commit 33afb92
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/test/SelectionInterface.test.ts b/src/test/SelectionInterface.test.ts
@@ -23,10 +23,12 @@ const idMapping = {
 };
 
 const color = "";
-const initialColorChanges = {
-    agents: [],
-    color,
-};
+const initialColorChanges = [
+    {
+        agents: [],
+        color,
+    },
+];
 
 describe("SelectionInterface module", () => {
     describe("decode", () => {
@@ -427,9 +429,10 @@ describe("SelectionInterface module", () => {
             };
             const si = new SelectionInterface();
             si.parse(idMapping);
+            const uiDisplayData = si.getUIDisplayData();
             const oldColors = si.getColorsForName("A");
             expect(oldColors).not.toContain(newColor);
-            si.updateAgentColors(agentIds, colorChanges);
+            si.updateAgentColors(agentIds, colorChanges, uiDisplayData);
             const colors = si.getColorsForName("A");
             expect(colors).toContain(newColor);
         });