vrr/g3
2
0
Fork 0
mirror of https://github.com/bytedance/g3.git synced 2026-05-19 07:54:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

g3proxy: add more post/put testcase (#373)

Browse source
This commit is contained in:
Zhang Jingqiang 2024-11-17 19:24:19 +08:00 • committed by GitHub
parent 35b344e71c
commit d5dfd98134
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 26 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

27
g3proxy/ci/python3+curl/test_httpbin.py
View file

@ -5,6 +5,7 @@ import sys
import unittest
import base64
from io import BytesIO
from urllib.parse import urlencode
import pycurl
@ -72,7 +73,7 @@ class TestHttpBin(unittest.TestCase):
self.assertEqual(self.c.getinfo(pycurl.RESPONSE_CODE), 200)
self.assertEqual(self.buffer.getvalue(), b"HTTPBIN is awesome")
def test_post_continue(self):
def test_post_small(self):
data = "Content to post"
self.set_url_and_request_target('/post')
@ -80,10 +81,34 @@ class TestHttpBin(unittest.TestCase):
self.c.perform()
self.assertEqual(self.c.getinfo(pycurl.RESPONSE_CODE), 200)
# add Expect and try again
self.c.setopt(pycurl.HTTPHEADER, ['Expect: 100-continue'])
self.c.perform()
self.assertEqual(self.c.getinfo(pycurl.RESPONSE_CODE), 200)
def test_post_large(self):
post_data = {'data': "Content to post" * 1024 * 100}
post_fields = urlencode(post_data)
self.set_url_and_request_target('/post')
self.c.setopt(pycurl.POSTFIELDS, post_fields)
self.c.perform()
self.assertEqual(self.c.getinfo(pycurl.RESPONSE_CODE), 200)
# disable Expect and try again
self.c.setopt(pycurl.HTTPHEADER, ['Expect:'])
self.c.perform()
self.assertEqual(self.c.getinfo(pycurl.RESPONSE_CODE), 200)
def test_put_file(self):
self.set_url_and_request_target('/put')
self.c.setopt(pycurl.UPLOAD, 1)
file = open(__file__)
self.c.setopt(pycurl.READDATA, file)
self.c.perform()
self.assertEqual(self.c.getinfo(pycurl.RESPONSE_CODE), 200)
file.close()
if __name__ == '__main__':
parser = argparse.ArgumentParser()