From 563f396b4b45fe5a8d36ed08d2354388751bbc93 Mon Sep 17 00:00:00 2001
From: philon- <philon-@users.noreply.github.com>
Date: Wed, 7 May 2025 22:30:06 +0200
Subject: [PATCH] Update tests

---
 src/components/generic/Table/index.test.tsx | 37 ++++++++++++++++-----
 1 file changed, 29 insertions(+), 8 deletions(-)

diff --git a/src/components/generic/Table/index.test.tsx b/src/components/generic/Table/index.test.tsx
index 61bd4063..0c81f66f 100644
--- a/src/components/generic/Table/index.test.tsx
+++ b/src/components/generic/Table/index.test.tsx
@@ -40,25 +40,46 @@ describe("Generic Table", () => {
 
   // A simplified version of the rows in pages/Nodes.tsx for testing purposes
   const mockDevicesWithShortNameAndConnection = [
-    { user: { shortName: "TST1" }, hopsAway: 1, lastHeard: Date.now() + 1000 },
-    { user: { shortName: "TST2" }, hopsAway: 0, lastHeard: Date.now() + 4000 },
-    { user: { shortName: "TST3" }, hopsAway: 4, lastHeard: Date.now() },
-    { user: { shortName: "TST4" }, hopsAway: 3, lastHeard: Date.now() + 2000 },
+    {
+      user: { shortName: "TST1" },
+      hopsAway: 1,
+      lastHeard: Date.now() + 1000,
+      viaMqtt: false,
+    },
+    {
+      user: { shortName: "TST2" },
+      hopsAway: 0,
+      lastHeard: Date.now() + 4000,
+      viaMqtt: true,
+    },
+    {
+      user: { shortName: "TST3" },
+      hopsAway: 4,
+      lastHeard: Date.now(),
+      viaMqtt: false,
+    },
+    {
+      user: { shortName: "TST4" },
+      hopsAway: 3,
+      lastHeard: Date.now() + 2000,
+      viaMqtt: true,
+    },
   ];
 
   const mockRows = mockDevicesWithShortNameAndConnection.map((node) => [
     <h1 data-testshortname key={node.user.shortName}>{node.user.shortName}</h1>,
-    <React.Fragment key={node.user.shortName}>
+    <Mono key="lastHeard" data-testheard>
       <TimeAgo timestamp={node.lastHeard * 1000} />
-    </React.Fragment>,
+    </Mono>,
     <Mono key="hops" data-testhops>
       {node.lastHeard !== 0
-        ? node.hopsAway === 0
+        ? node.viaMqtt === false && node.hopsAway === 0
           ? "Direct"
           : `${node.hopsAway?.toString()} ${
-            node.hopsAway > 1 ? "hops" : "hop"
+            node.hopsAway ?? 0 > 1 ? "hops" : "hop"
           } away`
         : "-"}
+      {node.viaMqtt === true ? ", via MQTT" : ""}
     </Mono>,
   ]);