Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(129)

Unified Diff: recipe_engine/third_party/requests/tests/test_testserver.py

Issue 2164713003: Vendor requests. (Closed) Base URL: https://chromium.googlesource.com/external/github.com/luci/recipes-py@master
Patch Set: Fix deps.pyl Created 4 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: recipe_engine/third_party/requests/tests/test_testserver.py
diff --git a/recipe_engine/third_party/requests/tests/test_testserver.py b/recipe_engine/third_party/requests/tests/test_testserver.py
deleted file mode 100644
index 027f8e5048c795c2ddcca8da9ca488f6946771c1..0000000000000000000000000000000000000000
--- a/recipe_engine/third_party/requests/tests/test_testserver.py
+++ /dev/null
@@ -1,137 +0,0 @@
-import threading
-import socket
-import time
-
-import pytest
-import requests
-from tests.testserver.server import Server
-
-class TestTestServer:
- def test_basic(self):
- """messages are sent and received properly"""
- question = b"sucess?"
- answer = b"yeah, success"
- def handler(sock):
- text = sock.recv(1000)
- assert text == question
- sock.sendall(answer)
-
- with Server(handler) as (host, port):
- sock = socket.socket()
- sock.connect((host, port))
- sock.sendall(question)
- text = sock.recv(1000)
- assert text == answer
- sock.close()
-
- def test_server_closes(self):
- """the server closes when leaving the context manager"""
- with Server.basic_response_server() as (host, port):
- sock = socket.socket()
- sock.connect((host, port))
-
- sock.close()
-
- with pytest.raises(socket.error):
- new_sock = socket.socket()
- new_sock.connect((host, port))
-
- def test_text_response(self):
- """the text_response_server sends the given text"""
- server = Server.text_response_server(
- "HTTP/1.1 200 OK\r\n" +
- "Content-Length: 6\r\n" +
- "\r\nroflol"
- )
-
- with server as (host, port):
- r = requests.get('http://{0}:{1}'.format(host, port))
-
- assert r.status_code == 200
- assert r.text == u'roflol'
- assert r.headers['Content-Length'] == '6'
-
- def test_basic_response(self):
- """the basic response server returns an empty http response"""
- with Server.basic_response_server() as (host, port):
- r = requests.get('http://{0}:{1}'.format(host, port))
- assert r.status_code == 200
- assert r.text == u''
- assert r.headers['Content-Length'] == '0'
-
- def test_basic_waiting_server(self):
- """the server waits for the block_server event to be set before closing"""
- block_server = threading.Event()
-
- with Server.basic_response_server(wait_to_close_event=block_server) as (host, port):
- sock = socket.socket()
- sock.connect((host, port))
- sock.sendall(b'send something')
- time.sleep(2.5)
- sock.sendall(b'still alive')
- block_server.set() # release server block
-
- def test_multiple_requests(self):
- """multiple requests can be served"""
- requests_to_handle = 5
-
- server = Server.basic_response_server(requests_to_handle=requests_to_handle)
-
- with server as (host, port):
- server_url = 'http://{0}:{1}'.format(host, port)
- for _ in range(requests_to_handle):
- r = requests.get(server_url)
- assert r.status_code == 200
-
- # the (n+1)th request fails
- with pytest.raises(requests.exceptions.ConnectionError):
- r = requests.get(server_url)
-
- def test_request_recovery(self):
- """can check the requests content"""
- server = Server.basic_response_server(requests_to_handle=2)
- first_request = b'put your hands up in the air'
- second_request = b'put your hand down in the floor'
-
- with server as address:
- sock1 = socket.socket()
- sock2 = socket.socket()
-
- sock1.connect(address)
- sock1.sendall(first_request)
- sock1.close()
-
- sock2.connect(address)
- sock2.sendall(second_request)
- sock2.close()
-
- assert server.handler_results[0] == first_request
- assert server.handler_results[1] == second_request
-
- def test_requests_after_timeout_are_not_received(self):
- """the basic response handler times out when receiving requests"""
- server = Server.basic_response_server(request_timeout=1)
-
- with server as address:
- sock = socket.socket()
- sock.connect(address)
- time.sleep(1.5)
- sock.sendall(b'hehehe, not received')
- sock.close()
-
- assert server.handler_results[0] == b''
-
-
- def test_request_recovery_with_bigger_timeout(self):
- """a biggest timeout can be specified"""
- server = Server.basic_response_server(request_timeout=3)
- data = b'bananadine'
-
- with server as address:
- sock = socket.socket()
- sock.connect(address)
- time.sleep(1.5)
- sock.sendall(data)
- sock.close()
-
- assert server.handler_results[0] == data
« no previous file with comments | « recipe_engine/third_party/requests/tests/test_structures.py ('k') | recipe_engine/third_party/requests/tests/test_utils.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698