From 46a50608e29f71724a395089ca9eb56536266e68 Mon Sep 17 00:00:00 2001 From: Unai Uribarri Date: Mon, 11 Jan 2021 12:55:14 +0100 Subject: [PATCH] Update room names when receiving a M message Also fix unit tests, which did not expect room name updates. --- maxcube/cube.py | 2 ++ tests/test_cube.py | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/maxcube/cube.py b/maxcube/cube.py index 2219852..eeb33d0 100644 --- a/maxcube/cube.py +++ b/maxcube/cube.py @@ -185,6 +185,8 @@ def parse_m_message(self, message): room.id = room_id room.name = name self.rooms.append(room) + else: + room.name = name num_devices = data[pos] pos += 1 diff --git a/tests/test_cube.py b/tests/test_cube.py index 4aa4281..5356a6d 100644 --- a/tests/test_cube.py +++ b/tests/test_cube.py @@ -211,10 +211,10 @@ def test_parse_m_message(self): self.assertEqual('KEQ0839778', self.cube.devices[2].serial) self.assertEqual(1, self.cube.devices[3].room_id) - self.assertEqual('Badezimmer', self.cube.rooms[0].name) + self.assertEqual('Kitchen', self.cube.rooms[0].name) self.assertEqual(1, self.cube.rooms[0].id) - self.assertEqual('Wohnzimmer', self.cube.rooms[1].name) + self.assertEqual('Living', self.cube.rooms[1].name) self.assertEqual(2, self.cube.rooms[1].id) def test_parse_l_message(self):