From edc792b3a874e6066d7d9a7eafd8d8753296bf5c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=A0=C3=A1rka=20Faloutov=C3=A1?=
 <65911928+sarkafa@users.noreply.github.com>
Date: Mon, 23 Dec 2024 15:46:51 +0100
Subject: [PATCH] connect: mark core one as buddy printer (#182)

---
 prusaerrors/connect/codes.py | 3 ++-
 tests/test_connect.py        | 5 +++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/prusaerrors/connect/codes.py b/prusaerrors/connect/codes.py
index 33d7d73..45ebd47 100644
--- 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
index 58e4370..ab54f66 100644
--- a/tests/test_connect.py
+++ b/tests/test_connect.py
@@ -69,6 +69,7 @@ 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)
@@ -76,6 +77,10 @@ def test_COREONE(self):
         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."""