Working test case for oneof

betterproto/tests/inputs/oneof/oneof-name.json

@@ -1,3 +1,3 @@
 {
-  "name": "foo"
+  "name": "foobar"
 }

betterproto/tests/inputs/oneof/oneof.json

@@ -1,3 +1,3 @@
 {
-  "count": 1
+  "count": 100
 }

betterproto/tests/inputs/oneof/test_oneof.py

@@ -0,0 +1,15 @@
+import betterproto
+from betterproto.tests.output_betterproto.oneof.oneof import Test
+from betterproto.tests.util import get_test_case_json_data
+
+
+def test_which_count():
+    message = Test()
+    message.from_json(get_test_case_json_data("oneof"))
+    assert betterproto.which_one_of(message, "foo") == ("count", 100)
+
+
+def test_which_name():
+    message = Test()
+    message.from_json(get_test_case_json_data("oneof", "oneof-name.json"))
+    assert betterproto.which_one_of(message, "foo") == ("name", "foobar")