New test for NWStatus class

2020-12-03 20:22:22 +01:00
parent c7c70a35a0
commit 0cb36c02f3
5 changed files with 132 additions and 16 deletions
@@ -0,0 +1,126 @@
+# -*- coding: utf-8 -*-
+"""novelWriter Status Class Tester
+"""
+
+import pytest
+
+from lxml import etree
+
+from nw.core.status import NWStatus
+
+@pytest.mark.core
+def testCoreStatus_Entries():
+    """Test all the simple setters for the NWItem class.
+    """
+    theStatus = NWStatus()
+
+    # Add entries
+    theStatus.addEntry("New",   (100, 100, 100))
+    theStatus.addEntry("Minor", (200, 50,  0))
+    theStatus.addEntry("Major", (200, 150, 0))
+    theStatus.addEntry("Main",  (50,  200, 0))
+
+    assert theStatus._theLabels == ["New", "Minor", "Major", "Main"]
+    assert theStatus._theColours == [(100, 100, 100), (200, 50, 0), (200, 150, 0), (50, 200, 0)]
+    assert theStatus._theCounts == [0, 0, 0, 0]
+    assert theStatus._theMap["New"] == 0
+    assert theStatus._theMap["Minor"] == 1
+    assert theStatus._theMap["Major"] == 2
+    assert theStatus._theMap["Main"] == 3
+    assert theStatus._theLength == 4
+
+    # Lookups
+    assert theStatus.lookupEntry(None) is None
+    assert theStatus.lookupEntry("dummy") is None
+    assert theStatus.lookupEntry("Main") == 3
+
+    # Checks
+    assert theStatus.checkEntry(123) == "New"
+    assert theStatus.checkEntry("Stuff") == "New"
+    assert theStatus.checkEntry("New ") == "New"
+    assert theStatus.checkEntry("  Main ") == "Main"
+
+    # Set new list
+    newList = [
+        ("New", 1, 1, 1, "New"),
+        ("Minor", 2, 2, 2, "Minor"),
+        ("Major", 3, 3, 3, "Major"),
+        ("Min", 4, 4, 4, "Main"),
+        ("Max", 5, 5, 5, None),
+    ]
+    assert theStatus.setNewEntries(None) == {}
+    assert theStatus.setNewEntries(newList) == {"Main": "Min"}
+
+    assert theStatus._theLabels == ["New", "Minor", "Major", "Min", "Max"]
+    assert theStatus._theColours == [(1, 1, 1), (2, 2, 2), (3, 3, 3), (4, 4, 4), (5, 5, 5)]
+    assert theStatus._theCounts == [0, 0, 0, 0, 0]
+    assert theStatus._theMap["New"] == 0
+    assert theStatus._theMap["Minor"] == 1
+    assert theStatus._theMap["Major"] == 2
+    assert theStatus._theMap["Min"] == 3
+    assert theStatus._theMap["Max"] == 4
+    assert theStatus._theLength == 5
+
+    # Add counts
+    countTo = [3, 5, 7, 9, 11]
+    for i, n in enumerate(countTo):
+        for _ in range(n):
+            theStatus.countEntry(theStatus._theLabels[i])
+    assert theStatus._theCounts == countTo
+
+    # Iterate
+    for i, (sA, sB, sC) in enumerate(theStatus):
+        assert sA == theStatus._theLabels[i]
+        assert sB == theStatus._theColours[i]
+        assert sC == theStatus._theCounts[i]
+
+    assert theStatus[9] == (None, None, None)
+
+    # Clear counts
+    theStatus.resetCounts()
+    assert theStatus._theCounts == [0, 0, 0, 0, 0]
+
+# END Test testCoreStatus_Entries
+
+@pytest.mark.core
+def testCoreStatus_XMLPackUnpack():
+    """Test all the simple setters for the NWItem class.
+    """
+    theStatus = NWStatus()
+    theStatus.addEntry("New",   (100, 100, 100))
+    theStatus.addEntry("Minor", (200, 50,  0))
+    theStatus.addEntry("Major", (200, 150, 0))
+    theStatus.addEntry("Main",  (50,  200, 0))
+
+    countTo = [3, 5, 7, 9]
+    for i, n in enumerate(countTo):
+        for _ in range(n):
+            theStatus.countEntry(theStatus._theLabels[i])
+
+    nwXML = etree.Element("novelWriterXML")
+
+    # Pack
+    xStatus = etree.SubElement(nwXML, "status")
+    theStatus.packXML(xStatus)
+    assert etree.tostring(xStatus, pretty_print=False, encoding="utf-8") == (
+        b"<status>"
+        b"<entry blue=\"100\" green=\"100\" red=\"100\">New</entry>"
+        b"<entry blue=\"0\" green=\"50\" red=\"200\">Minor</entry>"
+        b"<entry blue=\"0\" green=\"150\" red=\"200\">Major</entry>"
+        b"<entry blue=\"0\" green=\"200\" red=\"50\">Main</entry>"
+        b"</status>"
+    )
+
+    # Unpack
+    theStatus = NWStatus()
+    assert theStatus.unpackXML(xStatus)
+    assert theStatus._theLabels == ["New", "Minor", "Major", "Main"]
+    assert theStatus._theColours == [(100, 100, 100), (200, 50, 0), (200, 150, 0), (50, 200, 0)]
+    assert theStatus._theCounts == [0, 0, 0, 0]
+    assert theStatus._theMap["New"] == 0
+    assert theStatus._theMap["Minor"] == 1
+    assert theStatus._theMap["Major"] == 2
+    assert theStatus._theMap["Main"] == 3
+    assert theStatus._theLength == 4
+
+# END Test testCoreStatus_XMLPackUnpack