Пример #1
0
def test_read_response_head():
    rfile = BytesIO(b"HTTP/1.1 418 I'm a teapot\r\n" b"Content-Length: 4\r\n" b"\r\n" b"skip")
    rfile.reset_timestamps = Mock()
    rfile.first_byte_timestamp = 42
    r = read_response_head(rfile)
    assert r.status_code == 418
    assert r.headers["Content-Length"] == "4"
    assert r.content is None
    assert rfile.reset_timestamps.called
    assert r.timestamp_start == 42
    assert rfile.read() == b"skip"
Пример #2
0
def test_read_request_head():
    rfile = BytesIO(b"GET / HTTP/1.1\r\n" b"Content-Length: 4\r\n" b"\r\n" b"skip")
    rfile.reset_timestamps = Mock()
    rfile.first_byte_timestamp = 42
    r = read_request_head(rfile)
    assert r.method == "GET"
    assert r.headers["Content-Length"] == "4"
    assert r.content is None
    assert rfile.reset_timestamps.called
    assert r.timestamp_start == 42
    assert rfile.read() == b"skip"
Пример #3
0
def test_read_response_head():
    rfile = BytesIO(b"HTTP/1.1 418 I'm a teapot\r\n"
                    b"Content-Length: 4\r\n"
                    b"\r\n"
                    b"skip")
    rfile.reset_timestamps = Mock()
    rfile.first_byte_timestamp = 42
    r = read_response_head(rfile)
    assert r.status_code == 418
    assert r.headers["Content-Length"] == "4"
    assert r.content is None
    assert rfile.reset_timestamps.called
    assert r.timestamp_start == 42
    assert rfile.read() == b"skip"
Пример #4
0
def test_read_request_head():
    rfile = BytesIO(b"GET / HTTP/1.1\r\n"
                    b"Content-Length: 4\r\n"
                    b"\r\n"
                    b"skip")
    rfile.reset_timestamps = Mock()
    rfile.first_byte_timestamp = 42
    r = read_request_head(rfile)
    assert r.method == "GET"
    assert r.headers["Content-Length"] == "4"
    assert r.content is None
    assert rfile.reset_timestamps.called
    assert r.timestamp_start == 42
    assert rfile.read() == b"skip"
Пример #5
0
 def test_reset_timestamps(self):
     s = BytesIO(b"foobar\nfoobar")
     s = tcp.Reader(s)
     s.first_byte_timestamp = 500
     s.reset_timestamps()
     assert not s.first_byte_timestamp
Пример #6
0
 def test_reset_timestamps(self):
     s = BytesIO(b"foobar\nfoobar")
     s = tcp.Reader(s)
     s.first_byte_timestamp = 500
     s.reset_timestamps()
     assert not s.first_byte_timestamp