def setUp(self):
self.email = "*****@*****.**"
self.new_antiPhishing_code = "1111111111"
self.modify_antiPhishing_code = "2222222222"
check_email_result = get_data_from_db.get_user_id_by_email(self.email)
if check_email_result == None:
print("用户", self.email, "未存在，注册开始")
register.register_action(self.email)
else:
print("用户", self.email, "已存在")
self.userid = get_data_from_db.get_user_id_by_email(self.email)
print("self.userid:", self.userid)
check_result = get_data_from_db.get_user_GA_secret(self.userid)
print("check_result:", check_result)
if (check_result == None):
print("用户", self.email, "未绑定GA，绑定GA开始")
self.userid = get_data_from_db.get_user_id_by_email(self.email)
self.token = login.login(self.email)
response = bind_GA_page.bind_GA(self.email, self.token)
else:
print("用户", self.email, "已绑定GA")
self.token = login.login(self.email)
apCode_result = get_data_from_db.get_apcode_by_userid(
self.userid).get("ap_code")
if apCode_result == None:
print("用户", self.email, "未绑定防钓鱼码，开始绑定防钓鱼码")
self.secret = get_data_from_db.get_bind_GA_secret_fromdb(
self.userid)
self.GA_code = get_GA_code.get_google_code(self.secret)
response = antiPhishing_code_page.add_antiPhishing_code_api(
self.new_antiPhishing_code, self.GA_code, self.token)
else:
print("用户", self.email, "已设置防钓鱼码")
self.token = login.login(self.email)
def change_GA(userid, token):
send_change_GA_email(token)
vcode = get_data_from_redis.get_change_GA_emailcode_from_redis(userid)
print("vcode", vcode)
response = bind_GA_page.get_GA_secret_api(token)
new_secret = get_data_from_db.get_bind_GA_secret_fromdb(userid)
new_GA_code = get_GA_code.get_google_code(new_secret)
# print("new_GA_code",new_GA_code)
old_secret = get_data_from_db.get_user_GA_secret(userid).get("secretKey")
old_GA_code = get_GA_code.get_google_code(old_secret)
response = change_GA_api(old_GA_code, new_GA_code, vcode, token)
if response.get("status") != "success":
print("change_GA fail,response: ", response)
else:
print("change_GA success!")
return response
def test_add_antiPhishing_code_success(self):
"""add antiPhishing_code success"""
self.secret = get_data_from_db.get_bind_GA_secret_fromdb(self.userid)
self.GA_code = get_GA_code.get_google_code(self.secret)
response = antiPhishing_code_page.add_antiPhishing_code_api(
self.new_antiPhishing_code, self.GA_code, self.token)
print(response)
self.assertEqual(response.get("status"), "success",
"add antiPhinshing_code success")
def test_modify_antiPhishing_code_success(self):
"""modify antiPhishing_code success"""
self.secret = get_data_from_db.get_user_GA_secret(
self.userid).get("secretKey")
self.GA_code = get_GA_code.get_google_code(self.secret)
response = antiPhishing_code_page.modify_antiPhishing_code_api(
self.new_antiPhishing_code, self.modify_antiPhishing_code,
self.GA_code, self.token)
print(response)
self.assertEqual(response.get("status"), "success",
"modify antiPhinshing_code success")
def bind_GA(email, token):
get_GA_secret_api(token)
userid = get_data_from_db.get_user_id_by_email(email)
secret = get_data_from_db.get_bind_GA_secret_fromdb(userid)
GA_code = get_GA_code.get_google_code(secret)
bind_GA_send_email_api(token)
email_code = get_data_from_redis.get_bind_GA_email_code_from_redis(userid)
response = bind_GA_api(email_code, GA_code, token)
if response.get("status") != "success":
print("bind_GA fail,response: ", response)
else:
print("bind_GA success!")
return response
def login(email):
"""根据email登录并拿到token"""
config = get_config()
base_url = "https://" + config.get("host") + "/v1/sign-in"
headers = get_headers()
# if email is None:
# email = "*****@*****.**"
# secretKey = "OFCSSHNSK47IL4TJJ4X4GYBUY3V7BMJS"
# ga = ga_auth.get_totp_token(secretKey)
# else:
# email = email
# secretKey = check_bind_ga(email)
# if secretKey is None:
# ga = ""
# else:
# print(secretKey)
# # ga = ga_auth.get_totp_token(secretKey)
# userid = get_user_id_by_email(email)
# secretKey = get_user_GA_secret(userid).get("secretKey")
# ga = get_GA_code.get_google_code(secretKey)
# print(type(ga))
email = email
secretKey = check_bind_ga(email)
if secretKey is None:
ga = ""
else:
print(secretKey)
# ga = ga_auth.get_totp_token(secretKey)
userid = get_user_id_by_email(email)
secretKey = get_user_GA_secret(userid).get("secretKey")
ga = get_GA_code.get_google_code(secretKey)
print(type(ga))
password = hmac_password(email, b"Password01")
payload = {"email": email, "password": password, "ga": ga}
print("ga= " + str(ga))
response = requests.request("POST", base_url, headers=headers, data=json.dumps(payload))
status = response.json()['status']
if status != "success":
print("login fail！ response: " + response.text)
else:
print("login success, response: " + response.text)
my_token = response.json()['data']['token']
print("my_token=" + my_token)
return my_token
def setUp(self):
# 准备email
# 注册
# 登录获取token
now_time = time.strftime("%Y-%m-%d_%H_%M_%S")
# email = now_time + "@1.com"
email = global_variable.cir_var + "_" + now_time + "@1.com"
print("email:", email)
register.register_action(email)
self.token = login.login_with_param(email)
bind_GA_page.get_GA_secret_api(self.token)
self.userid = get_data_from_db.get_user_id_by_email(email)
self.secret = get_data_from_db.get_bind_GA_secret_fromdb(self.userid)
print("secret", self.secret)
self.GA_code = get_GA_code.get_google_code(self.secret)
print("GA_code", self.GA_code)
bind_GA_page.bind_GA_send_email_api(self.token)
self.email_code = get_data_from_redis.get_bind_GA_email_code_from_redis(
self.userid)
print("email_code", self.email_code)