fix tests

pedohorse · Jan 20, 2024 · ac2b31c · ac2b31c
1 parent 08b9a28
commit ac2b31c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_ui_protocol_data.py b/tests/test_ui_protocol_data.py
@@ -3,6 +3,7 @@
 from io import BytesIO
 from lifeblood.enums import TaskState, WorkerType, WorkerState, TaskGroupArchivedState
 from lifeblood.buffered_connection import BufferedReaderWrapper
+from lifeblood.worker_metadata import WorkerMetadata
 from lifeblood.ui_protocol_data import UiData, NodeGraphStructureData, \
     NodeData, NodeConnectionData, TaskBatchData, TaskData, WorkerData, WorkerResources, WorkerBatchData, \
     TaskGroupData, TaskGroupStatisticsData, TaskGroupBatchData, TaskDelta
@@ -26,7 +27,8 @@ def test_trivial1(self):
             }), WorkerBatchData(0, {
                 121: WorkerData(121, WorkerResources(12.3, 23.4, 929283838, 939384848, 56.7, 67.8, 84847575, 85857676),
                                 '1928374', '127.1.2.333:blo', 1234567, WorkerState.BUSY, WorkerType.STANDARD,
-                                1534, 8273, 7573, 55.5, {"borker", "gorker"})
+                                1534, 8273, 7573, 55.5, {"borker", "gorker"},
+                                WorkerMetadata("fooo"))
             }), TaskGroupBatchData(0, {
                 'grooup foo': TaskGroupData('grooup foo', 2345678, TaskGroupArchivedState.ARCHIVED, 22.3,
                                             TaskGroupStatisticsData(45, 56, 67, 78))