def test_import_hook():
ab_1 = thriftpy2.load("addressbook.thrift")
print("Load file succeed.")
assert ab_1.DEFAULT_LIST_SIZE == 10
try:
import addressbook_thrift as ab # noqa
except ImportError:
print("Import hook not installed.")
thriftpy2.install_import_hook()
import addressbook_thrift as ab_2
print("Magic import succeed.")
assert ab_2.DEFAULT_LIST_SIZE == 10
# -*- coding: utf-8 -*-
import thriftpy2
thriftpy2.install_import_hook()
import const_thrift as const # noqa
def test_num_const():
assert -10 == const.NEGATIVE_I16
assert -123.456 == const.NEGATIVE_DOUBLE
assert 10 == const.I16_CONST
assert 100000 == const.I32_CONST
assert 123.456 == const.DOUBLE_CONST
def test_string_const():
assert "hello" == const.DOUBLE_QUOTED_CONST
assert "hello" == const.SINGLE_QUOTED_CONST
def test_const_with_sep():
assert "hello" == const.CONST_WITH_SEP1
assert "hello" == const.CONST_WITH_SEP2
def test_list_const():
assert [1, 2, 3] == const.I32_LIST_CONST
assert [1.1, 2.2, 3.3] == const.DOUBLE_LIST_CONST
assert ["hello", "world"] == const.STRING_LIST_CONST
# -*- coding: utf-8 -*-
from __future__ import absolute_import
import os
import multiprocessing
import socket
import time
import pytest
import thriftpy2
thriftpy2.install_import_hook() # noqa
from thriftpy2.http import make_server, make_client, client_context # noqa
from thriftpy2.thrift import TApplicationException # noqa
addressbook = thriftpy2.load(
os.path.join(os.path.dirname(__file__), "addressbook.thrift"))
class Dispatcher():
def __init__(self):
self.ab = addressbook.AddressBook()
self.ab.people = {}
def ping(self):
return True
def hello(self, name):
return "hello " + name
