python(进阶2)实现自动化注册和登录

1. 分析需求

后端完成接口以后，工作中可能会涉及到自测通断，a接口和b接口之间可能有关联关系，例如:a接口注册，b接口登录，就需要a接口返回的参数传递到b接口

2. 环境准备

需要这些类包

import requests
import random

3. 实战代码

先准备2个接口
注册接口：https://xxx/register
登录接口:https://xxx/login
注册需要的参数

register_data = {
    "email": random_str + "@189.com",
    "passwd": "12345678",
    "repasswd": "12345678",
}

这里需要的是邮箱注册，由于需要反复测试，所以写了一个随机字符方法

def generate_random_string(length):
    characters = string.ascii_letters + string.digits  # 小写字母+大写字母+数字
    random_string = ''.join(random.choice(characters) for _ in range(length))
    return random_string

发送注册请求

with requests.Session() as s:
    register_response = s.post(register_url, data=register_data)
    register_result = register_response.json()  

    # 检查注册是否成功
    if register_result['code'] == 200:
        print("注册成功！")

这里判断注册是否成功，成功后再执行登录方法

        # 使用注册时提供的email和密码进行登录
        login_data = {
            "email": register_data['email'],  # 使用注册时提供的email
            "passwd": register_data['passwd']  # 使用注册时设置的密码
        }

        # 发送登录请求
        login_response = s.post(login_url, data=login_data)

        # 检查登录是否成功（这里需要根据实际的响应格式来判断）
        if login_response.status_code == 200:
           print('登录成功')

这样就完成了在连贯的流程里自动化测试，如果还有更多的接口都可以直接再下面去扩展。