test: Test string subclass

python-sdbus · May 21, 2022 · 1a8ab05 · 1a8ab05 · igo95862 · May 21, 2022
1 parent b1db762
commit 1a8ab05
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test/test_read_write_dbus_types.py b/test/test_read_write_dbus_types.py
@@ -415,6 +415,18 @@ def test_message_properties(self) -> None:
         self.assertEqual(message.member,
                          'GetUnit')
 
+    def test_string_subclass(self) -> None:
+        from enum import Enum
+
+        class TestEnum(str, Enum):
+            SOMETHING = 'test'
+
+        message = create_message(self.bus)
+        message.append_data('s', TestEnum.SOMETHING)
+        message.seal()
+
+        self.assertEqual(message.get_contents(), TestEnum.SOMETHING)
+
 
 if __name__ == "__main__":
     main()