83 lines
2.1 KiB
Python
83 lines
2.1 KiB
Python
import pytest
|
|
import capnp
|
|
import os
|
|
|
|
this_dir = os.path.dirname(__file__)
|
|
|
|
@pytest.fixture
|
|
def addressbook():
|
|
return capnp.load(os.path.join(this_dir, 'addressbook.capnp'))
|
|
|
|
def build_message(addressbook):
|
|
addresses = addressbook.AddressBook.new_message()
|
|
people = addresses.init('people', 2)
|
|
|
|
alice = people[0]
|
|
alice.id = 123
|
|
alice.name = 'Alice'
|
|
alice.email = 'alice@example.com'
|
|
alicePhones = alice.init('phones', 1)
|
|
alicePhones[0].number = "555-1212"
|
|
alicePhones[0].type = 'mobile'
|
|
alice.employment.school = "MIT"
|
|
|
|
bob = people[1]
|
|
bob.id = 456
|
|
bob.name = 'Bob'
|
|
bob.email = 'bob@example.com'
|
|
bobPhones = bob.init('phones', 2)
|
|
bobPhones[0].number = "555-4567"
|
|
bobPhones[0].type = 'home'
|
|
bobPhones[1].number = "555-7654"
|
|
bobPhones[1].type = 'work'
|
|
bob.employment.unemployed = None
|
|
|
|
return addresses
|
|
|
|
def check_msg(addresses):
|
|
people = addresses.people
|
|
|
|
alice = people[0]
|
|
assert alice.id == 123
|
|
assert alice.name == 'Alice'
|
|
assert alice.email == 'alice@example.com'
|
|
alicePhones = alice.phones
|
|
assert alicePhones[0].number == "555-1212"
|
|
assert alicePhones[0].type == 'mobile'
|
|
assert alice.employment.school == "MIT"
|
|
|
|
bob = people[1]
|
|
assert bob.id == 456
|
|
assert bob.name == 'Bob'
|
|
assert bob.email == 'bob@example.com'
|
|
bobPhones = bob.phones
|
|
assert bobPhones[0].number == "555-4567"
|
|
assert bobPhones[0].type == 'home'
|
|
assert bobPhones[1].number == "555-7654"
|
|
assert bobPhones[1].type == 'work'
|
|
assert bob.employment.unemployed == None
|
|
|
|
|
|
def test_roundtrip_file(addressbook):
|
|
f = open('example', 'w')
|
|
msg = build_message(addressbook)
|
|
msg.write(f)
|
|
|
|
f = open('example', 'r')
|
|
msg = addressbook.AddressBook.read(f)
|
|
check_msg(msg)
|
|
|
|
def test_roundtrip_bytes(addressbook):
|
|
msg = build_message(addressbook)
|
|
message_bytes = msg.to_bytes()
|
|
|
|
msg = addressbook.AddressBook.from_bytes(message_bytes)
|
|
check_msg(msg)
|
|
|
|
def test_roundtrip_dict(addressbook):
|
|
msg = build_message(addressbook)
|
|
d = msg.to_dict()
|
|
|
|
msg = addressbook.AddressBook.from_dict(d)
|
|
check_msg(msg)
|