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)