import unittest, ddt, json
from common import commonMethod, requestMethod
from process import loginProcess, useDatabase
local_request_method = requestMethod.RequestMethod()
local_common_method = commonMethod.CommonMethod()
local_login_process = loginProcess.LoginProcess()
local_use_database = useDatabase.SelectData()
headers = local_login_process.api_header_token()
datas = local_common_method.readExcel("syncApis")
@ddt.ddt
class TestOpertionRoles(unittest.TestCase):
@ddt.data(*datas[0:2])
def test_user_ak_sync(self, datas):
uri = local_common_method.getUrl("HTTP", "sit", datas[2])
try:
result = local_request_method.request(datas[1], uri, headers,
datas[3])
result_dict = json.loads(result.text)
if "records" in result_dict["result"].keys():
self.checkResult(len(result_dict["result"]["records"]),
datas[4])
else:
self.checkResult(result_dict["result"]["total"], datas[4])
except Exception as e:
raise e
@ddt.data(*datas[2:4])
def test_user_info_sync(self, datas):
import requests
import json
from common import commonMethod
localCommonMethod = commonMethod.CommonMethod()
class RequestMethod:
def post(self, url, headers, data):
res = requests.post(url=url, headers=headers, data=data)
return res
def get(self, url, headers, data):
res = requests.get(url=url, headers=headers, params=data)
return res
def delete(self, url, headers, data):
res = requests.delete(url=url, headers=headers, data=data)
return res
def request(self, method, url, headers, data):
if method == "post":
res = self.post(url, headers, data)
elif method == "get":
res = self.get(url, headers, data)
else:
res = self.delete(url, headers, data)
return res
def login(self, env):
headers = {"Content-Type": "application/json"}