def test_load_serialized_pb_as_proto():
expected = cpb.FtpSource()
expected.address = "abcd"
with TemporaryDirectory() as tmp_dir:
path = Path(tmp_dir).joinpath("ftp.pb")
with open(path, 'wb') as tmp_file:
tmp_file.write(expected.SerializeToString())
actual = proto_util.load_serialized_pb_as_proto(
path=path,
proto_obj=cpb.FtpSource()
)
assert actual == expected
def test_load_text_as_proto():
expected = cpb.FtpSource()
expected.address = "abcd"
with TemporaryDirectory() as tmp_dir:
path = Path(tmp_dir).joinpath("ftp.config")
with open(path, 'w') as tmp_file:
tmp_file.write("""
address: "abcd"
""")
actual = proto_util.load_text_as_proto(
path=path,
proto_obj=cpb.FtpSource()
)
assert actual == expected
def test_set_field_unnested(): expected = cpb.FtpSource() expected.address = "new_addr_val" actual = cpb.FtpSource() proto_util.set_field(actual, "address", "new_addr_val") assert actual == expected
def test_parse_proto_fields_ftp_source(): expected = set(["address", "workdir"]) actual = set(proto_util.get_full_field_names(cpb.FtpSource())) assert actual == expected