Rename some call sites for read_headers.

Fixes tests.
This commit is contained in:
Jacob Hoffman-Andrews
2020-11-15 13:34:15 -08:00
committed by Martin Algesten
parent f652a9e449
commit 4cb856196e
4 changed files with 7 additions and 7 deletions

View File

@@ -1,6 +1,6 @@
#![allow(dead_code)] #![allow(dead_code)]
use crate::testserver::{read_headers, TestServer}; use crate::testserver::{read_request, TestServer};
use std::io::{self, Read, Write}; use std::io::{self, Read, Write};
use std::net::TcpStream; use std::net::TcpStream;
use std::time::Duration; use std::time::Duration;
@@ -10,7 +10,7 @@ use super::super::*;
// Handler that answers with a simple HTTP response, and times // Handler that answers with a simple HTTP response, and times
// out idle connections after 2 seconds. // out idle connections after 2 seconds.
fn idle_timeout_handler(mut stream: TcpStream) -> io::Result<()> { fn idle_timeout_handler(mut stream: TcpStream) -> io::Result<()> {
read_headers(&stream); read_request(&stream);
stream.write_all(b"HTTP/1.1 200 OK\r\nContent-Length: 8\r\n\r\nresponse")?; stream.write_all(b"HTTP/1.1 200 OK\r\nContent-Length: 8\r\n\r\nresponse")?;
stream.set_read_timeout(Some(Duration::from_secs(2)))?; stream.set_read_timeout(Some(Duration::from_secs(2)))?;
Ok(()) Ok(())
@@ -73,7 +73,7 @@ fn custom_resolver() {
#[cfg(feature = "cookies")] #[cfg(feature = "cookies")]
#[cfg(test)] #[cfg(test)]
fn cookie_and_redirect(mut stream: TcpStream) -> io::Result<()> { fn cookie_and_redirect(mut stream: TcpStream) -> io::Result<()> {
let headers = read_headers(&stream); let headers = read_request(&stream);
match headers.path() { match headers.path() {
"/first" => { "/first" => {
stream.write_all(b"HTTP/1.1 302 Found\r\n")?; stream.write_all(b"HTTP/1.1 302 Found\r\n")?;
@@ -134,7 +134,7 @@ fn test_cookies_on_redirect() -> Result<(), Error> {
#[test] #[test]
fn dirty_streams_not_returned() -> Result<(), Error> { fn dirty_streams_not_returned() -> Result<(), Error> {
let testserver = TestServer::new(|mut stream: TcpStream| -> io::Result<()> { let testserver = TestServer::new(|mut stream: TcpStream| -> io::Result<()> {
read_headers(&stream); read_request(&stream);
stream.write_all(b"HTTP/1.1 200 OK\r\n")?; stream.write_all(b"HTTP/1.1 200 OK\r\n")?;
stream.write_all(b"Transfer-Encoding: chunked\r\n")?; stream.write_all(b"Transfer-Encoding: chunked\r\n")?;
stream.write_all(b"\r\n")?; stream.write_all(b"\r\n")?;

View File

@@ -97,7 +97,7 @@ fn redirect_host() {
// of example.invalid. We can probably do better by, e.g. // of example.invalid. We can probably do better by, e.g.
// overriding the resolver. // overriding the resolver.
let srv = TestServer::new(|mut stream: TcpStream| -> io::Result<()> { let srv = TestServer::new(|mut stream: TcpStream| -> io::Result<()> {
testserver::read_headers(&stream); testserver::read_request(&stream);
write!(stream, "HTTP/1.1 302 Found\r\n")?; write!(stream, "HTTP/1.1 302 Found\r\n")?;
write!(stream, "Location: http://example.invalid/\r\n")?; write!(stream, "Location: http://example.invalid/\r\n")?;
write!(stream, "\r\n")?; write!(stream, "\r\n")?;

View File

@@ -9,7 +9,7 @@ use super::super::*;
// Send an HTTP response on the TcpStream at a rate of two bytes every 10 // Send an HTTP response on the TcpStream at a rate of two bytes every 10
// milliseconds, for a total of 600 bytes. // milliseconds, for a total of 600 bytes.
fn dribble_body_respond(mut stream: TcpStream, contents: &[u8]) -> io::Result<()> { fn dribble_body_respond(mut stream: TcpStream, contents: &[u8]) -> io::Result<()> {
read_headers(&stream); read_request(&stream);
let headers = format!( let headers = format!(
"HTTP/1.1 200 OK\r\nContent-Length: {}\r\n\r\n", "HTTP/1.1 200 OK\r\nContent-Length: {}\r\n\r\n",
contents.len() * 2 contents.len() * 2

View File

@@ -93,7 +93,7 @@ pub fn read_request(stream: &TcpStream) -> TestHeaders {
for line in BufReader::new(stream).lines() { for line in BufReader::new(stream).lines() {
match line { match line {
Err(e) => { Err(e) => {
eprintln!("testserver: in read_headers: {}", e); eprintln!("testserver: in read_request: {}", e);
break; break;
} }
Ok(line) if line == "" => break, Ok(line) if line == "" => break,