关于python+websocket+unittest的接口封装

在用python+websocket+unittest编写测试代码，但是如何将连接的接口进行封装，登陆接口直接调用呢，求指导

连接代码：

class connection():
ws = 'ws://43.248.79.180:10040/'

def setUp(self):
    self.ws = create_connection(self.ws)
    self.ws.settimeout(20)

# 采用断言，测试连接状态
def test_connect_sataus(self):
    self.assertEqual(101, self.ws.getstatus(),msg='成功')

def tearDown(self):
    self.ws.close()

登陆代码：
class test_login(unittest.TestCase):

# 发送参数
def test_send_logininfo(self):
    params = {
        "T": "Req",
        "ID": 1,
        "A": "User.Login",
        "P": {
            "udid": "test_003",
            "pkgType": "Moye.Standalone.Test.Main",
            "data": {
                "type": "test"
            },
            "lang": "zh"
        }
    }

    # 调用json.dumps()方法并将字典中的参数传入给该方法
    self.ws.send(json.dumps(params))
    result = self.ws.recv()
    res = json.loads(result)
    print('接收结果：', res)
    expt = 'Resp'
    exptmsg = res['T']
    assert expt == exptmsg
    print('断言结果：成功')

if name == 'main':
unittest.main()