test_tk

# -*- coding: utf-8 -*-

"""
@Time    : 2021/12/17 14:25
@Author  : 阿涛@
"""
import hashlib
import json
import re
import unittest
from api.cms_tk import TK_CMSApi
import requests


def get_sig(data_dict) -> str:
    params_str = "".join(sorted([f"{k}={v}" for k, v in data_dict.items()])) + "irc_666."
    return hashlib.md5(params_str.encode("utf-8")).hexdigest()


class Test_Tk(unittest.TestCase):
    tk_cmsApi = TK_CMSApi("http://192.168.31.240:5002")

    def test_tk_001_login(self):
        url = "/user/login"
        data = {
            "username": "admin",
            "password": "123456"
        }
        data['sig'] = get_sig(data)
        print(data['sig'])  # 提交之前查看sig签名
        resp = self.tk_cmsApi.post(url, json=data)
        print(resp.text)
        assert "登录成功！" in resp.text