pip install winney
from winney.winney import Address
from winney import Winney, retry
from winney.mock import Mock
class UserMock(Mock):
data = {"name": "olivetree"}
class UserCenter(object):
def __init__(self):
self.winney = Winney(host="localhost", port=5000)
self.init_functions()
def init_functions(self):
self.winney.register(method="post",
name="login",
uri="/api/login",
mock=False,
mock_data=None)
self.winney.register(method="get",
name="get_user",
uri="/api/user",
mock=True,
mock_data=UserMock())
def login(self, account, password):
r = self.winney.login(json={"account": account, "password": password})
return r.json()
def get_user(self, user_id):
r = self.winney.get_user(data={"user_id": user_id})
return r.json()
if __name__ == "__main__":
uc = UserCenter()
uc.login("hello", "123456")