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

1from sbe2.schema import Message, Messages 

2from pytest import raises 

3 

4def test_add_get(): 

5 

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 

11 

12 with raises(KeyError): 

13 _ = m['WrongName'] 

14 with raises(KeyError): 

15 _ = m[8] 

16 

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