connect: mark core one as buddy printer (#182)

prusa3d · Dec 23, 2024 · edc792b · edc792b
1 parent b21dd18
commit edc792b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/prusaerrors/connect/codes.py b/prusaerrors/connect/codes.py
@@ -15,7 +15,8 @@
                                       unique_codes)
 
 BUDDY = [
-    'MINI', 'MK4', 'IX', 'XL', 'MK3.5', 'MK4S', 'MK3.9', 'MK3.9S', 'MK3.5S'
+    'MINI', 'MK4', 'IX', 'XL', 'MK3.5', 'MK4S', 'MK3.9', 'MK3.9S', 'MK3.5S',
+    'COREONE'
 ]
 
 

diff --git a/tests/test_connect.py b/tests/test_connect.py
@@ -69,13 +69,18 @@ def test_MK35S(self):
         assert code.id
 
     def test_COREONE(self):
+        # unique code for COREONE
         code = PrinterCodes.get("31701")
         assert code.printer == Printer.COREONE
         assert code.category == Category(7)
         assert code.error == 1
         assert code.title
         assert code.message
         assert code.id
+        # common code for buddy printers
+        code = PrinterCodes.get("31801")
+        assert code.printer == Printer.COREONE
+
 
     def test_no_MK35S(self):
         """MK35S doesn't have puppies."""