Coverage for tests/test_schema/test_messages_collection.py: 100%
16 statements
« prev ^ index » next coverage.py v7.9.1, created at 2025-06-21 15:58 +0200
« prev ^ index » next coverage.py v7.9.1, created at 2025-06-21 15:58 +0200
1from sbe2.schema import Message, Messages
2from pytest import raises
4def test_add_get():
6 m = Messages()
7 msg = Message(name="TestMessage", description='', id=5, fields=[], groups=[], datas=[], package='package')
8 m.add(msg)
9 assert m['TestMessage'] is msg
10 assert m[5] is msg
12 with raises(KeyError):
13 _ = m['WrongName']
14 with raises(KeyError):
15 _ = m[8]
17 assert m.get('TestMessage') is msg
18 assert m.get(5) is msg
19 assert m.get('WrongName') is None
20 assert m.get(8) is None