| #!/usr/bin/env python3 |
| # -*- coding: utf-8 -*- |
| #*************************************************************************** |
| # _ _ ____ _ |
| # Project ___| | | | _ \| | |
| # / __| | | | |_) | | |
| # | (__| |_| | _ <| |___ |
| # \___|\___/|_| \_\_____| |
| # |
| # Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al. |
| # |
| # This software is licensed as described in the file COPYING, which |
| # you should have received as part of this distribution. The terms |
| # are also available at https://curl.se/docs/copyright.html. |
| # |
| # You may opt to use, copy, modify, merge, publish, distribute and/or sell |
| # copies of the Software, and permit persons to whom the Software is |
| # furnished to do so, under the terms of the COPYING file. |
| # |
| # This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY |
| # KIND, either express or implied. |
| # |
| # SPDX-License-Identifier: curl |
| # |
| ########################################################################### |
| # |
| import difflib |
| import filecmp |
| import logging |
| import os |
| import time |
| import pytest |
| |
| from testenv import Env, CurlClient |
| |
| |
| log = logging.getLogger(__name__) |
| |
| |
| class TestUpload: |
| |
| @pytest.fixture(autouse=True, scope='class') |
| def _class_scope(self, env, httpd, nghttpx): |
| if env.have_h3(): |
| nghttpx.start_if_needed() |
| env.make_data_file(indir=env.gen_dir, fname="data-63k", fsize=63*1024) |
| env.make_data_file(indir=env.gen_dir, fname="data-64k", fsize=64*1024) |
| env.make_data_file(indir=env.gen_dir, fname="data-100k", fsize=100*1024) |
| env.make_data_file(indir=env.gen_dir, fname="data-10m", fsize=10*1024*1024) |
| httpd.clear_extra_configs() |
| httpd.reload() |
| |
| # upload small data, check that this is what was echoed |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_01_upload_1_small(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 fails here") |
| data = '0123456789' |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]' |
| r = curl.http_upload(urls=[url], data=data, alpn_proto=proto) |
| r.check_stats(count=1, http_status=200, exitcode=0) |
| respdata = open(curl.response_file(0)).readlines() |
| assert respdata == [data] |
| |
| # upload large data, check that this is what was echoed |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_02_upload_1_large(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 fails here") |
| fdata = os.path.join(env.gen_dir, 'data-100k') |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]' |
| r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto) |
| r.check_stats(count=1, http_status=200, exitcode=0) |
| indata = open(fdata).readlines() |
| respdata = open(curl.response_file(0)).readlines() |
| assert respdata == indata |
| |
| # upload data sequentially, check that they were echoed |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_10_upload_sequential(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 stalls here") |
| count = 50 |
| data = '0123456789' |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-{count-1}]' |
| r = curl.http_upload(urls=[url], data=data, alpn_proto=proto) |
| r.check_stats(count=count, http_status=200, exitcode=0) |
| for i in range(count): |
| respdata = open(curl.response_file(i)).readlines() |
| assert respdata == [data] |
| |
| # upload data parallel, check that they were echoed |
| @pytest.mark.parametrize("proto", ['h2', 'h3']) |
| def test_07_11_upload_parallel(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 stalls here") |
| # limit since we use a separate connection in h1 |
| count = 50 |
| data = '0123456789' |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-{count-1}]' |
| r = curl.http_upload(urls=[url], data=data, alpn_proto=proto, |
| extra_args=['--parallel']) |
| r.check_stats(count=count, http_status=200, exitcode=0) |
| for i in range(count): |
| respdata = open(curl.response_file(i)).readlines() |
| assert respdata == [data] |
| |
| # upload large data sequentially, check that this is what was echoed |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_12_upload_seq_large(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 stalls here") |
| fdata = os.path.join(env.gen_dir, 'data-100k') |
| count = 20 |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-{count-1}]' |
| r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto) |
| r.check_response(count=count, http_status=200) |
| indata = open(fdata).readlines() |
| r.check_stats(count=count, http_status=200, exitcode=0) |
| for i in range(count): |
| respdata = open(curl.response_file(i)).readlines() |
| assert respdata == indata |
| |
| # upload very large data sequentially, check that this is what was echoed |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_13_upload_seq_large(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 stalls here") |
| fdata = os.path.join(env.gen_dir, 'data-10m') |
| count = 2 |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-{count-1}]' |
| r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto) |
| r.check_stats(count=count, http_status=200, exitcode=0) |
| indata = open(fdata).readlines() |
| for i in range(count): |
| respdata = open(curl.response_file(i)).readlines() |
| assert respdata == indata |
| |
| # upload data parallel, check that they were echoed |
| @pytest.mark.parametrize("proto", ['h2', 'h3']) |
| def test_07_20_upload_parallel(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 stalls here") |
| # limit since we use a separate connection in h1 |
| count = 20 |
| data = '0123456789' |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-{count-1}]' |
| r = curl.http_upload(urls=[url], data=data, alpn_proto=proto, |
| extra_args=['--parallel']) |
| r.check_stats(count=count, http_status=200, exitcode=0) |
| for i in range(count): |
| respdata = open(curl.response_file(i)).readlines() |
| assert respdata == [data] |
| |
| # upload large data parallel, check that this is what was echoed |
| @pytest.mark.parametrize("proto", ['h2', 'h3']) |
| def test_07_21_upload_parallel_large(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 stalls here") |
| fdata = os.path.join(env.gen_dir, 'data-100k') |
| # limit since we use a separate connection in h1 |
| count = 20 |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-{count-1}]' |
| r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto, |
| extra_args=['--parallel']) |
| r.check_response(count=count, http_status=200) |
| self.check_download(count, fdata, curl) |
| |
| # upload large data parallel to a URL that denies uploads |
| @pytest.mark.parametrize("proto", ['h2', 'h3']) |
| def test_07_22_upload_parallel_fail(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 stalls here") |
| fdata = os.path.join(env.gen_dir, 'data-10m') |
| count = 100 |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}'\ |
| f'/curltest/tweak?status=400&delay=5ms&chunks=1&body_error=reset&id=[0-{count-1}]' |
| r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto, |
| extra_args=['--parallel']) |
| exp_exit = 92 if proto == 'h2' else 95 |
| r.check_stats(count=count, exitcode=exp_exit) |
| |
| # PUT 100k |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_30_put_100k(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 fails here") |
| fdata = os.path.join(env.gen_dir, 'data-100k') |
| count = 1 |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?id=[0-{count-1}]' |
| r = curl.http_put(urls=[url], fdata=fdata, alpn_proto=proto, |
| extra_args=['--parallel']) |
| r.check_stats(count=count, http_status=200, exitcode=0) |
| exp_data = [f'{os.path.getsize(fdata)}'] |
| r.check_response(count=count, http_status=200) |
| for i in range(count): |
| respdata = open(curl.response_file(i)).readlines() |
| assert respdata == exp_data |
| |
| # PUT 10m |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_31_put_10m(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 fails here") |
| fdata = os.path.join(env.gen_dir, 'data-10m') |
| count = 1 |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?id=[0-{count-1}]&chunk_delay=2ms' |
| r = curl.http_put(urls=[url], fdata=fdata, alpn_proto=proto, |
| extra_args=['--parallel']) |
| r.check_stats(count=count, http_status=200, exitcode=0) |
| exp_data = [f'{os.path.getsize(fdata)}'] |
| r.check_response(count=count, http_status=200) |
| for i in range(count): |
| respdata = open(curl.response_file(i)).readlines() |
| assert respdata == exp_data |
| |
| # issue #10591 |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_32_issue_10591(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 fails here") |
| fdata = os.path.join(env.gen_dir, 'data-10m') |
| count = 1 |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?id=[0-{count-1}]' |
| r = curl.http_put(urls=[url], fdata=fdata, alpn_proto=proto) |
| r.check_stats(count=count, http_status=200, exitcode=0) |
| |
| # issue #11157, upload that is 404'ed by server, needs to terminate |
| # correctly and not time out on sending |
| def test_07_33_issue_11157a(self, env: Env, httpd, nghttpx, repeat): |
| proto = 'h2' |
| fdata = os.path.join(env.gen_dir, 'data-10m') |
| # send a POST to our PUT handler which will send immediately a 404 back |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put' |
| curl = CurlClient(env=env) |
| r = curl.run_direct(with_stats=True, args=[ |
| '--resolve', f'{env.authority_for(env.domain1, proto)}:127.0.0.1', |
| '--cacert', env.ca.cert_file, |
| '--request', 'POST', |
| '--max-time', '5', '-v', |
| '--url', url, |
| '--form', 'idList=12345678', |
| '--form', 'pos=top', |
| '--form', 'name=mr_test', |
| '--form', f'fileSource=@{fdata};type=application/pdf', |
| ]) |
| assert r.exit_code == 0, f'{r}' |
| r.check_stats(1, 404) |
| |
| # issue #11157, send upload that is slowly read in |
| def test_07_33_issue_11157b(self, env: Env, httpd, nghttpx, repeat): |
| proto = 'h2' |
| fdata = os.path.join(env.gen_dir, 'data-10m') |
| # tell our test PUT handler to read the upload more slowly, so |
| # that the send buffering and transfer loop needs to wait |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?chunk_delay=2ms' |
| curl = CurlClient(env=env) |
| r = curl.run_direct(with_stats=True, args=[ |
| '--verbose', '--trace-config', 'ids,time', |
| '--resolve', f'{env.authority_for(env.domain1, proto)}:127.0.0.1', |
| '--cacert', env.ca.cert_file, |
| '--request', 'PUT', |
| '--max-time', '10', '-v', |
| '--url', url, |
| '--form', 'idList=12345678', |
| '--form', 'pos=top', |
| '--form', 'name=mr_test', |
| '--form', f'fileSource=@{fdata};type=application/pdf', |
| ]) |
| assert r.exit_code == 0, r.dump_logs() |
| r.check_stats(1, 200) |
| |
| def test_07_34_issue_11194(self, env: Env, httpd, nghttpx, repeat): |
| proto = 'h2' |
| # tell our test PUT handler to read the upload more slowly, so |
| # that the send buffering and transfer loop needs to wait |
| fdata = os.path.join(env.gen_dir, 'data-100k') |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put' |
| curl = CurlClient(env=env) |
| r = curl.run_direct(with_stats=True, args=[ |
| '--verbose', '--trace-config', 'ids,time', |
| '--resolve', f'{env.authority_for(env.domain1, proto)}:127.0.0.1', |
| '--cacert', env.ca.cert_file, |
| '--request', 'PUT', |
| '--digest', '--user', 'test:test', |
| '--data-binary', f'@{fdata}', |
| '--url', url, |
| ]) |
| assert r.exit_code == 0, r.dump_logs() |
| r.check_stats(1, 200) |
| |
| # upload large data on a h1 to h2 upgrade |
| def test_07_35_h1_h2_upgrade_upload(self, env: Env, httpd, nghttpx, repeat): |
| fdata = os.path.join(env.gen_dir, 'data-100k') |
| curl = CurlClient(env=env) |
| url = f'http://{env.domain1}:{env.http_port}/curltest/echo?id=[0-0]' |
| r = curl.http_upload(urls=[url], data=f'@{fdata}', extra_args=[ |
| '--http2' |
| ]) |
| r.check_response(count=1, http_status=200) |
| # apache does not Upgrade on request with a body |
| assert r.stats[0]['http_version'] == '1.1', f'{r}' |
| indata = open(fdata).readlines() |
| respdata = open(curl.response_file(0)).readlines() |
| assert respdata == indata |
| |
| # upload to a 301,302,303 response |
| @pytest.mark.parametrize("redir", ['301', '302', '303']) |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_36_upload_30x(self, env: Env, httpd, nghttpx, repeat, redir, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 fails here") |
| data = '0123456789' * 10 |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo{redir}?id=[0-0]' |
| r = curl.http_upload(urls=[url], data=data, alpn_proto=proto, extra_args=[ |
| '-L', '--trace-config', 'http/2,http/3' |
| ]) |
| r.check_response(count=1, http_status=200) |
| respdata = open(curl.response_file(0)).readlines() |
| assert respdata == [] # was transformed to a GET |
| |
| # upload to a 307 response |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_37_upload_307(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 fails here") |
| data = '0123456789' * 10 |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo307?id=[0-0]' |
| r = curl.http_upload(urls=[url], data=data, alpn_proto=proto, extra_args=[ |
| '-L', '--trace-config', 'http/2,http/3' |
| ]) |
| r.check_response(count=1, http_status=200) |
| respdata = open(curl.response_file(0)).readlines() |
| assert respdata == [data] # was POST again |
| |
| # POST form data, yet another code path in transfer |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_38_form_small(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 fails here") |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]' |
| r = curl.http_form(urls=[url], alpn_proto=proto, form={ |
| 'name1': 'value1', |
| }) |
| r.check_stats(count=1, http_status=200, exitcode=0) |
| |
| # POST data urlencoded, small enough to be sent with request headers |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_39_post_urlenc_small(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 fails here") |
| fdata = os.path.join(env.gen_dir, 'data-63k') |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]' |
| r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto, extra_args=[ |
| '--trace-config', 'http/2,http/3' |
| ]) |
| r.check_stats(count=1, http_status=200, exitcode=0) |
| indata = open(fdata).readlines() |
| respdata = open(curl.response_file(0)).readlines() |
| assert respdata == indata |
| |
| # POST data urlencoded, large enough to be sent separate from request headers |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_40_post_urlenc_large(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 fails here") |
| fdata = os.path.join(env.gen_dir, 'data-64k') |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]' |
| r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto, extra_args=[ |
| '--trace-config', 'http/2,http/3' |
| ]) |
| r.check_stats(count=1, http_status=200, exitcode=0) |
| indata = open(fdata).readlines() |
| respdata = open(curl.response_file(0)).readlines() |
| assert respdata == indata |
| |
| # POST data urlencoded, small enough to be sent with request headers |
| # and request headers are so large that the first send is larger |
| # than our default upload buffer length (64KB). |
| # Unfixed, this will fail when run with CURL_DBG_SOCK_WBLOCK=80 most |
| # of the time |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_41_post_urlenc_small(self, env: Env, httpd, nghttpx, repeat, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| if proto == 'h3' and env.curl_uses_lib('msh3'): |
| pytest.skip("msh3 fails here") |
| if proto == 'h3' and env.curl_uses_lib('quiche'): |
| pytest.skip("quiche has CWND issues with large requests") |
| fdata = os.path.join(env.gen_dir, 'data-63k') |
| curl = CurlClient(env=env) |
| extra_args = ['--trace-config', 'http/2,http/3'] |
| # add enough headers so that the first send chunk is > 64KB |
| for i in range(63): |
| extra_args.extend(['-H', f'x{i:02d}: {"y"*1019}']) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]' |
| r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto, extra_args=extra_args) |
| r.check_stats(count=1, http_status=200, exitcode=0) |
| indata = open(fdata).readlines() |
| respdata = open(curl.response_file(0)).readlines() |
| assert respdata == indata |
| |
| def check_download(self, count, srcfile, curl): |
| for i in range(count): |
| dfile = curl.download_file(i) |
| assert os.path.exists(dfile) |
| if not filecmp.cmp(srcfile, dfile, shallow=False): |
| diff = "".join(difflib.unified_diff(a=open(srcfile).readlines(), |
| b=open(dfile).readlines(), |
| fromfile=srcfile, |
| tofile=dfile, |
| n=1)) |
| assert False, f'download {dfile} differs:\n{diff}' |
| |
| # speed limited on put handler |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_50_put_speed_limit(self, env: Env, httpd, nghttpx, proto, repeat): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| count = 1 |
| fdata = os.path.join(env.gen_dir, 'data-100k') |
| up_len = 100 * 1024 |
| speed_limit = 20 * 1024 |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?id=[0-0]' |
| r = curl.http_put(urls=[url], fdata=fdata, alpn_proto=proto, |
| with_headers=True, extra_args=[ |
| '--limit-rate', f'{speed_limit}' |
| ]) |
| r.check_response(count=count, http_status=200) |
| assert r.responses[0]['header']['received-length'] == f'{up_len}', f'{r.responses[0]}' |
| up_speed = r.stats[0]['speed_upload'] |
| assert (speed_limit * 0.5) <= up_speed <= (speed_limit * 1.5), f'{r.stats[0]}' |
| |
| # speed limited on echo handler |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_07_51_echo_speed_limit(self, env: Env, httpd, nghttpx, proto, repeat): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| count = 1 |
| fdata = os.path.join(env.gen_dir, 'data-100k') |
| speed_limit = 20 * 1024 |
| curl = CurlClient(env=env) |
| url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]' |
| r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto, |
| with_headers=True, extra_args=[ |
| '--limit-rate', f'{speed_limit}' |
| ]) |
| r.check_response(count=count, http_status=200) |
| up_speed = r.stats[0]['speed_upload'] |
| assert (speed_limit * 0.5) <= up_speed <= (speed_limit * 1.5), f'{r.stats[0]}' |
| |