Fix match test

AdrienVannson · AdrienVannson · commit bc9d79ed593c · 2024-11-12T10:43:41.000+01:00
diff --git a/tests/test_oneof_pattern_matching.py b/tests/test_oneof_pattern_matching.py
@@ -0,0 +1,52 @@
+from dataclasses import dataclass
+
+import pytest
+import sys
+from typing import Optional
+
+import betterproto
+
+
+@pytest.mark.skipif(
+    sys.version_info < (3, 10),
+    reason="pattern matching is only supported in python3.10+",
+)
+def test_oneof_pattern_matching():
+    @dataclass
+    class Sub(betterproto.Message):
+        val: int = betterproto.int32_field(1)
+
+    @dataclass
+    class Foo(betterproto.Message):
+        bar: Optional[int] = betterproto.int32_field(1, group="group1")
+        baz: Optional[str] = betterproto.string_field(2, group="group1")
+        sub: Optional[Sub] = betterproto.message_field(3, group="group2")
+        abc: Optional[str] = betterproto.string_field(4, group="group2")
+
+    foo = Foo(baz="test1", abc="test2")
+
+    match foo:
+        case Foo(bar=int(_)):
+            pytest.fail("Matched 'bar' instead of 'baz'")
+        case Foo(baz=v):
+            assert v == "test1"
+        case _:
+            pytest.fail("Matched neither 'bar' nor 'baz'")
+
+    match foo:
+        case Foo(sub=Sub(_)):
+            pytest.fail("Matched 'sub' instead of 'abc'")
+        case Foo(abc=v):
+            assert v == "test2"
+        case _:
+            pytest.fail("Matched neither 'sub' nor 'abc'")
+
+    foo.sub = Sub(val=1)
+
+    match foo:
+        case Foo(sub=Sub(val=v)):
+            assert v == 1
+        case Foo(abc=str(v)):
+            pytest.fail("Matched 'abc' instead of 'sub'")
+        case _:
+            pytest.fail("Matched neither 'sub' nor 'abc'")
