c14fd7036a
* change to_dict return bytes for DATA type * reformat * reformat * add includes .h --------- Co-authored-by: Brian Xu <brian.xu1@bytedance.com>
22 lines
538 B
Python
22 lines
538 B
Python
import os
|
|
import capnp
|
|
import base64
|
|
import pytest
|
|
|
|
this_dir = os.path.dirname(__file__)
|
|
|
|
|
|
@pytest.fixture
|
|
def blob_schema():
|
|
return capnp.load(os.path.join(this_dir, "blob_test.capnp"))
|
|
|
|
|
|
def test_blob_to_dict(blob_schema):
|
|
blob_value = b"hello world"
|
|
blob = blob_schema.BlobTest(blob=blob_value)
|
|
blob_dict = blob.to_dict(encode_bytes_as_base64=True)
|
|
assert base64.b64decode(blob_dict["blob"]) == blob_value
|
|
msg = blob_schema.BlobTest.new_message()
|
|
msg.from_dict(blob_dict)
|
|
assert blob.blob == blob_value
|