betterproto: support Struct and Value (#551)

* betterproto: support `Struct` and `Value` Signed-off-by: William Woodruff <[email protected]> * betterproto: handle struct in to_dict as well Signed-off-by: William Woodruff <[email protected]> * tests: add Struct roundtrip tests Signed-off-by: William Woodruff <[email protected]> * specialize from_dict and to_dict on Struct ...rather than special-casing in the Message ABC. Signed-off-by: William Woodruff <[email protected]> * betterproto: `poe format` Signed-off-by: William Woodruff <[email protected]> * Update src/betterproto/__init__.py Co-authored-by: James Hilton-Balfe <[email protected]> * remove future annotations Signed-off-by: William Woodruff <[email protected]> * replace type[...] with typing.T Signed-off-by: William Woodruff <[email protected]> * quote instead Signed-off-by: William Woodruff <[email protected]> --------- Signed-off-by: William Woodruff <[email protected]> Co-authored-by: James Hilton-Balfe <[email protected]>
danielgtaylor · Jan 2, 2024 · 5666393 · 5666393
1 parent ce5093e
commit 5666393
Show file tree

Hide file tree

Showing 2 changed files with 55 additions and 0 deletions.
diff --git a/src/betterproto/lib/google/protobuf/__init__.py b/src/betterproto/lib/google/protobuf/__init__.py
diff --git a/tests/test_struct.py b/tests/test_struct.py
@@ -0,0 +1,24 @@
+import json
+
+from betterproto.lib.google.protobuf import Struct
+
+
+def test_struct_roundtrip():
+    data = {
+        "foo": "bar",
+        "baz": None,
+        "quux": 123,
+        "zap": [1, {"two": 3}, "four"],
+    }
+    data_json = json.dumps(data)
+
+    struct_from_dict = Struct().from_dict(data)
+    assert struct_from_dict.fields == data
+    assert struct_from_dict.to_dict() == data
+    assert struct_from_dict.to_json() == data_json
+
+    struct_from_json = Struct().from_json(data_json)
+    assert struct_from_json.fields == data
+    assert struct_from_json.to_dict() == data
+    assert struct_from_json == struct_from_dict
+    assert struct_from_json.to_json() == data_json