From 33afb92f23b566397061573bc5b5ec074fb57a61 Mon Sep 17 00:00:00 2001
From: Joe Heffernan <interim17@gmail.com>
Date: Fri, 22 Sep 2023 09:38:13 -0700
Subject: [PATCH] update tests

---
 src/test/SelectionInterface.test.ts | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/test/SelectionInterface.test.ts b/src/test/SelectionInterface.test.ts
index c48126e7..8c02d406 100644
--- 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);
         });