playwright.request.new_context()

def testQueryStudent(playwright: Playwright):
    """
    Опрос студентов
    """
    url = 'http://localhost:8090/studentFindById'
    param = {
        'id': 105
    }
    request_context = playwright.request.new_context()
    response = request_context.get(url=url, params=param)
    assert response.ok
    assert response.json()
    print('\n', response.json())

def testAddStudent(playwright: Playwright):
    """
    Добавить нового студента
    :return:
    """
    url = 'http://localhost:8090/studentAdd'
    request_body = {
        "className": "banji",
        "courseName": "wuli",
        "email": "ales@qq.com",
        "name": "ales",
        "score": 70,
        "sex": "boy",
        "studentId": "92908290"
    }
    header = {"Content-Type": "application/json"}
    request_context = playwright.request.new_context()
    response = request_context.post(url=url, headers=header, data=request_body)
    assert response.ok
    assert response.json()
    print('\n', response.json())

def testUpdateStudents(playwright: Playwright):
    """
    Изменить студентов
    """
    url = 'http://localhost:8090/studentUpdate/100'
    param = {
        'studentId': "id" + str(100),
        'name': "name" + str(100),
        'score': 100,
        "sex": "girl",
        "className": "class" + str(100),
        "courseName": "course" + str(100),
        "email": str(100) + "email@qq.com"

    }
    request_context = playwright.request.new_context()
    response = request_context.put(url=url, form=param)
    assert response.ok
    assert response.json()
    print('\n', response.json())

def testDeleteStudents(playwright: Playwright):
    """
    Удалить студентов
    """
    url = 'http://localhost:8090/studentDelete/' + str(105)
    request_context = playwright.request.new_context()
    response = request_context.delete(url=url)
    assert response.ok
    assert response.json()
    print('\n', response.json())

def test_upload_file(playwright: Playwright):
    '''
    Загрузить файлы
    :param playwright:
    :return:
    '''
    # Создать контекст запроса
    request_context = playwright.request.new_context()

    # определение Загрузить файлыизURL
    upload_url = "http://localhost:8090/fileUpload"

    # путь к файлу
    file_path = "d:/demo.txt"

    # Получить имя файла и тип MIME
    filename = file_path.split('/')[-1]
    mime_type, _ = mimetypes.guess_type(file_path)
    if not mime_type:
        mime_type = 'application/octet-stream'

    # Чтение содержимого файла
    with open(file_path, 'rb') as file:
        file_content = file.read()

    # Создайте граничную строку multipart/form-data
    boundary = '---------------------' + str(random.randint(1e28, 1e29 - 1))

    # Создать тело запроса
    body = (
        f'--{boundary}\r\n'
        f'Content-Disposition: form-data; name="file"; filename="{filename}"\r\n'
        f'Content-Type: {mime_type}\r\n\r\n'
        f'{file_content.decode("utf-8") if mime_type.startswith("text/") else file_content.hex()}'
        f'\r\n--{boundary}--\r\n'
    ).encode('utf-8')

    # Установить заголовок запроса
    headers = {
        'Content-Type': f'multipart/form-data; boundary={boundary}',
    }
    # Инициировать POST-запрос
    response = request_context.post(upload_url, data=body, headers=headers)

    # Проверить ответ
    assert response.status == 200, f"Upload failed with status: {response.status}"
    assert response.ok
    assert response.json()
    print('\n', response.json())

# Чтение содержимого файла
with open(file_path, 'rb') as file:
    file_content = file.read()
    response = request_context.post(upload_url, multipart={
        "fileField": {
            "name": "demo.txt",
            "mimeType": "text/plain",
            "buffer": file_content,
        }
    })
print('\n', response.json())