WIP

prusaerrors/sl1/codes.py

@@ -16,6 +16,8 @@
 
 from prusaerrors.shared.codes import Category, Code, Printer, unique_codes, Codes
 
+PRINTER_MODEL_PATH = Path("/run/model")
+
 if "_" not in vars(builtins):
 
     def _(value):
@@ -32,14 +34,13 @@ def decor(cls):
         assert "Errors" in data
 
         printer = Printer.SL1
-        path = Path("/run/model")
-        model = [x.name for x in path.iterdir() if x.is_file()]
+        model = [x.name for x in PRINTER_MODEL_PATH.iterdir() if x.is_file()]
         if len(model) == 1:
             a = Printer.M1.name.lower()
             if a in model:
                 printer = Printer.M1
         else:
-            raise KeyError("None or multiple model files found. Check /run/model folder.")
+            raise KeyError("None or multiple model files found. Check %s folder.", PRINTER_MODEL_PATH)
 
         re = re_compile(
             r"^(?P<printer>([0-9][0-9]|XX))"

tests/test_sl1.py

@@ -6,7 +6,9 @@
 # pylint: disable = missing-class-docstring
 # pylint: disable = missing-module-docstring
 
+from tempfile import TemporaryDirectory
 import unittest
+from unittest.mock import MagicMock, PropertyMock, patch
 
 from prusaerrors.sl1.codes import Sl1Codes
 
@@ -15,8 +17,18 @@ class TestErrors(unittest.TestCase):
     def test_str_conversion(self):
         self.assertEqual("#10500", str(Sl1Codes.NONE))
 
+
     def test_code_lookup(self):
-        self.assertEqual(Sl1Codes.NONE, Sl1Codes.get("#10500"))
+        with patch("prusaerrors.sl1.codes.PRINTER_MODEL_PATH") as mock:
+            with TemporaryDirectory() as dir:
+                mock.return_value = dir
+
+                with open(str(dir) + "/sl1", "w+", encoding="utf-8") as f: 
+                    self.assertEqual(Sl1Codes.NONE, Sl1Codes.get("#10500"))
+                with open(str(dir) + "/sl1s", "w+", encoding="utf-8") as f: 
+                    self.assertEqual(Sl1Codes.NONE, Sl1Codes.get("#10500"))
+                with open(str(dir) + "/m1", "w+", encoding="utf-8") as f: 
+                    self.assertEqual(Sl1Codes.NONE, Sl1Codes.get("#29500"))
 
     def test_unknown_code_lookup(self):
         self.assertEqual(Sl1Codes.UNKNOWN, Sl1Codes.get("random string"))