From 38a5fddb9d5086108bdd7ee3362ee45787b77004 Mon Sep 17 00:00:00 2001
From: Veronica Berglyd Olsen <1619840+vkbo@users.noreply.github.com>
Date: Thu, 30 May 2024 20:49:55 +0200
Subject: [PATCH] Update tests

---
 tests/test_gui/test_gui_statusbar.py | 26 +++++++++++++-------------
 1 file changed, 13 insertions(+), 13 deletions(-)

diff --git a/tests/test_gui/test_gui_statusbar.py b/tests/test_gui/test_gui_statusbar.py
index 228f6896..15e66921 100644
--- a/tests/test_gui/test_gui_statusbar.py
+++ b/tests/test_gui/test_gui_statusbar.py
@@ -25,7 +25,7 @@ import time
 import pytest
 
 from novelwriter import CONFIG, SHARED
-from novelwriter.extensions.statusled import StatusLED
+from novelwriter.enum import nwTrinary
 
 from tests.tools import C, buildTestProject
 
@@ -46,20 +46,20 @@ def testGuiStatusBar_Main(qtbot, nwGUI, projPath, mockRnd):
     assert nwGUI.mainStatus._refTime == refTime
 
     # Project Status
-    nwGUI.mainStatus.setProjectStatus(StatusLED.S_NONE)
-    assert nwGUI.mainStatus.projIcon._theCol == nwGUI.mainStatus.projIcon._colNone
-    nwGUI.mainStatus.setProjectStatus(StatusLED.S_BAD)
-    assert nwGUI.mainStatus.projIcon._theCol == nwGUI.mainStatus.projIcon._colBad
-    nwGUI.mainStatus.setProjectStatus(StatusLED.S_GOOD)
-    assert nwGUI.mainStatus.projIcon._theCol == nwGUI.mainStatus.projIcon._colGood
+    nwGUI.mainStatus.setProjectStatus(nwTrinary.NEUTRAL)
+    assert nwGUI.mainStatus.projIcon.state == nwTrinary.NEUTRAL
+    nwGUI.mainStatus.setProjectStatus(nwTrinary.NEGATIVE)
+    assert nwGUI.mainStatus.projIcon.state == nwTrinary.NEGATIVE
+    nwGUI.mainStatus.setProjectStatus(nwTrinary.POSITIVE)
+    assert nwGUI.mainStatus.projIcon.state == nwTrinary.POSITIVE
 
     # Document Status
-    nwGUI.mainStatus.setDocumentStatus(StatusLED.S_NONE)
-    assert nwGUI.mainStatus.docIcon._theCol == nwGUI.mainStatus.docIcon._colNone
-    nwGUI.mainStatus.setDocumentStatus(StatusLED.S_BAD)
-    assert nwGUI.mainStatus.docIcon._theCol == nwGUI.mainStatus.docIcon._colBad
-    nwGUI.mainStatus.setDocumentStatus(StatusLED.S_GOOD)
-    assert nwGUI.mainStatus.docIcon._theCol == nwGUI.mainStatus.docIcon._colGood
+    nwGUI.mainStatus.setDocumentStatus(nwTrinary.NEUTRAL)
+    assert nwGUI.mainStatus.docIcon.state == nwTrinary.NEUTRAL
+    nwGUI.mainStatus.setDocumentStatus(nwTrinary.NEGATIVE)
+    assert nwGUI.mainStatus.docIcon.state == nwTrinary.NEGATIVE
+    nwGUI.mainStatus.setDocumentStatus(nwTrinary.POSITIVE)
+    assert nwGUI.mainStatus.docIcon.state == nwTrinary.POSITIVE
 
     # Idle Status
     CONFIG.stopWhenIdle = False