blob: 2c5e797c0cf9d927c85df8f5f9c7e18097fa28d8 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
from dataclasses import dataclass
import pytest
import aristaproto
def test_oneof_pattern_matching():
@dataclass
class Sub(aristaproto.Message):
val: int = aristaproto.int32_field(1)
@dataclass
class Foo(aristaproto.Message):
bar: int = aristaproto.int32_field(1, group="group1")
baz: str = aristaproto.string_field(2, group="group1")
sub: Sub = aristaproto.message_field(3, group="group2")
abc: str = aristaproto.string_field(4, group="group2")
foo = Foo(baz="test1", abc="test2")
match foo:
case Foo(bar=_):
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=_):
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=v):
pytest.fail("Matched 'abc' instead of 'sub'")
case _:
pytest.fail("Matched neither 'sub' nor 'abc'")
|
