Python BytesIO.first_byte_timestamp Examples

Example #1

File: test_read.py Project: MatthewShao/mitmproxy

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"

Example #2

File: test_read.py Project: MatthewShao/mitmproxy

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"

Example #3

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"

Example #4

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"

Example #5

File: test_tcp.py Project: tdickers/mitmproxy

 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

Example #6

File: test_tcp.py Project: bemre/mitmproxy

 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