def test_tpayload_pickle(): ab = thriftpy.load_module("addressbook_thrift") person = ab.Person(name="Bob") person_2 = pickle.loads(PICKLED_BYTES) assert person == person_2
def test_load_module(): ab = thriftpy.load_module("addressbook_thrift") assert ab.__name__ == "addressbook_thrift" assert sys.modules["addressbook_thrift"] == ab # note we can import after load_module import addressbook_thrift as ab2 assert ab2 == ab
def _load_service(service_name): thrift_module = thriftpy.load_module( 'walis.thirdparty.thrift.{}_thrift'.format(service_name)) for key, value in thrift_module.__dict__.items(): if key.lower().endswith('service'): thrift_module.t_service = value return thrift_module