import pytest
import allure
from Base.WebAPI.API.Base import request_handler

@allure.epic("API自动化测试")
@allure.feature("百度API测试")
class TestBaiduAPI:
    """百度接口测试用例"""

    @allure.story("搜索功能测试")
    @allure.title("测试百度搜索接口 - {search_keyword}")
    @allure.severity(allure.severity_level.CRITICAL)
    @allure.description("测试百度搜索功能，验证返回结果包含搜索关键词")
    @pytest.mark.parametrize("search_keyword", ["python接口测试", "自动化测试", "pytest"])
    def test_baidu_search(self, search_keyword):
        """测试百度搜索接口"""
        # 添加测试步骤
        with allure.step("准备请求参数"):
            params = {'wd': search_keyword}
            allure.attach(f"搜索关键词: {search_keyword}", name="请求参数")

        with allure.step("发送搜索请求"):
            # 添加更多请求头，模拟真实浏览器
            headers = {
                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
            }
            response = request_handler.get('/s', params=params, headers=headers)
            allure.attach(f"状态码: {response.status_code}", name="响应状态")

        with allure.step("验证响应结果"):
            # 检查是否触发了安全验证
            if '安全验证' in response.text or 'verify' in response.text.lower():
                pytest.xfail("百度安全验证被触发，跳过此测试")

            # 断言
            assert response.status_code == 200
            assert search_keyword in response.text
            allure.attach(f"响应内容包含关键词: {search_keyword}", name="验证结果")

    @allure.story("首页访问测试")
    @allure.title("测试百度首页访问")
    @allure.severity(allure.severity_level.NORMAL)
    @allure.description("测试百度首页访问，验证页面基本元素")
    def test_baidu_homepage(self):
        """测试百度首页"""
        with allure.step("发送首页请求"):
            headers = {
                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
            }
            response = request_handler.get('/', headers=headers)
            allure.attach(f"状态码: {response.status_code}", name="响应状态")

        with allure.step("验证首页内容"):
            # 检查是否触发了安全验证
            if '安全验证' in response.text or 'verify' in response.text.lower():
                pytest.xfail("百度安全验证被触发，跳过此测试")

            assert response.status_code == 200
            # 由于安全验证问题，我们只检查状态码
            allure.attach("首页访问成功", name="验证结果")


@allure.epic("API自动化测试")
@allure.feature("HTTPBin API测试")
class TestHTTPBin:
    """HTTPBin接口测试用例"""

    @allure.story("GET请求测试")
    @allure.title("测试HTTPBin GET接口")
    @allure.severity(allure.severity_level.CRITICAL)
    @allure.description("测试HTTPBin GET接口，验证参数传递和响应")
    def test_httpbin_get(self):
        """测试HTTPBin GET接口"""
        with allure.step("准备请求参数"):
            params = {
                'test': 'python接口测试',
                'number': 123
            }
            allure.attach(str(params), name="请求参数")

        with allure.step("发送GET请求"):
            response = request_handler.get('https://httpbin.org/get', params=params)
            allure.attach(f"状态码: {response.status_code}", name="响应状态")

        with allure.step("验证响应结果"):
            assert response.status_code == 200
            data = response.json()
            assert data['args']['test'] == 'python接口测试'
            assert data['args']['number'] == '123'
            allure.attach(str(data), name="响应数据")

    @allure.story("POST请求测试")
    @allure.title("测试HTTPBin POST接口")
    @allure.severity(allure.severity_level.CRITICAL)
    @allure.description("测试HTTPBin POST接口，验证JSON数据传递")
    def test_httpbin_post(self):
        """测试HTTPBin POST接口"""
        with allure.step("准备请求数据"):
            data = {
                'test': 'python接口测试',
                'number': 123
            }
            allure.attach(str(data), name="请求数据")

        with allure.step("发送POST请求"):
            response = request_handler.post('https://httpbin.org/post', json=data)
            allure.attach(f"状态码: {response.status_code}", name="响应状态")

        with allure.step("验证响应结果"):
            assert response.status_code == 200
            response_data = response.json()
            assert response_data['json']['test'] == 'python接口测试'
            assert response_data['json']['number'] == 123
            allure.attach(str(response_data), name="响应数据")


if __name__ == '__main__':
    pytest.main([__file__, '-v', '--alluredir', '../Reports/allure_results'])