import requests


def test_upload(url='http://127.0.0.1:8000/upload/', filepath='C:/Users/songlin/Desktop/网工_组网技术实验.docx'):
    """测试文件从客户端上传至服务端"""
    content = open(filepath, 'rb')
    response = requests.post(url, files={'file': content})
    print(response.text)


def test_download(url='http://127.0.0.1:8000/download'):
    """测试文件从服务端下载至客户端"""
    response = requests.get(url)
    assert response.status_code == 200

    response_headers = response.headers
    filename = response_headers['content-disposition'].split('=')[-1].replace('"', '')

    with open(filename, mode='wb') as f:
        f.write(response.text.encode())


if __name__ == '__main__':
    # test_upload()
    # test_download()
    test_upload(url='http://192.168.2.200:2345/upload/')
    # test_download(url='http://192.168.2.200:2345/download/')