Python server.handle_connection函数代码示例

def test_non_existant_file():
    conn = FakeConnection("GET /ddd.txt HTTP/1.0\r\n\r\n")

    server.handle_connection(conn)

    assert 'HTTP/1.0 404 NOT FOUND' in conn.sent, 'Got: %s' % (repr(conn.sent),)
    assert '<h1>404 NOT FOUND</h1>' in conn.sent, 'Got: %s' % (repr(conn.sent),)

def test_form_get():
    conn =  FakeConnection("GET /submit?firstname=hello&lastname=world HTTP/1.0\r\n\r\n")


    server.handle_connection(conn)

    assert "Mr. hello world" in conn.sent, 'Got: %s' % (repr(conn.sent),)

def test_form_get():
    conn = FakeConnection("GET /submit?firstname=Usman&lastname=Majeed HTTP/1.0\r\n\r\n")
    
    server.handle_connection(conn)

    assert 'HTTP/1.0200 OK\r\n' in conn.sent, 'Got: %s' % (repr(conn.sent),)
    assert 'Hello Usman Majeed!' in conn.sent, 'Got: %s' % (repr(conn.sent),)

def test_handle_connection():
    conn = FakeConnection("GET / HTTP/1.0\r\n\r\n")
    expected_return = 'HTTP/1.0 200 OK'
    server.handle_connection(conn)
    splitconn = conn.sent.split('\r\n')[0]

    assert splitconn == expected_return, 'Got: %s' %(repr(conn.sent),)

def test_handle_not_found():
    conn = FakeConnection("GET /fake HTTP/1.0\r\n\r\n")

    app = make_app()
    server.handle_connection(conn, app)
    assert 'HTTP/1.0 404' in conn.sent , \
    'Got: %s' % (repr(conn.sent),)

def test_handle_empty_request():
    conn = FakeConnection("\r\n\r\n")

    server.handle_connection(conn)

    assert 'HTTP/1.0 404' in conn.sent and 'want' in conn.sent, \
    'Got: %s' % (repr(conn.sent),)

def test_handle_connenction_file():
    conn = FakeConnection("GET /file HTTP/1.0\r\n\r\n")
    
    server.handle_connection(conn)
 
    assert 'HTTP/1.0 200' in conn.sent and 'file' in conn.sent, \
    'Got: %s' % (repr(conn.sent),)

def test_handle_connection():
    conn = FakeConnection("GET / HTTP/1.0\r\n\r\n")

    server.handle_connection(conn, "arctic.cse.msu.edu", "9943")

    assert 'HTTP/1.0 200' in conn.sent and 'form' in conn.sent, \
    'Got: %s' % (repr(conn.sent),)

def test_handle_not_found():
    conn = FakeConnection("GET /poop HTTP/1.0\r\n\r\n")

    server.handle_connection(conn, "arctic.cse.msu.edu", "9943")

    assert 'HTTP/1.0 404' in conn.sent and 'want' in conn.sent, \
    'Got: %s' % (repr(conn.sent),)

def test_submit_post_urlencoded():
    conn = FakeConnection("POST /submit HTTP/1.0\r\n" + \
                           "Content-Length: 26\r\n" + \
                           "Content-Type: application/x-www-form-urlencoded\r\n\r\n" + \
                           "firstname=Matt&lastname=Ao\r\n")
    server.handle_connection(conn, 80, "myapp")
    assert conn.sent[:len(expected_OK)] == expected_OK, 'Got: %s' % (repr(conn.sent),)

def test_handle_connection():
    conn = FakeConnection("GET / HTTP/1.0\r\n\r\n")
    server.handle_connection(conn, 0, APP)

    assert 'HTTP/1.0 200' in conn.sent
    assert 'ettemaet' in conn.sent, \
        'Got: %s' % (repr(conn.sent),)

def test_submit_post_multipart():
    conn = FakeConnection("POST /submit HTTP/1.0\r\n" + \
                          "Content-Length: 374\r\n" + \
                          "Content-Type: multipart/form-data; " + \
                          "boundary=32452685f36942178a5f36fd94e34b63\r\n\r\n" + \
                          "--32452685f36942178a5f36fd94e34b63\r\n" + \
                          "Content-Disposition: form-data; name=\"lastname\";" + \
                          " filename=\"lastname\"\r\n\r\n" + \
                          "taylor\r\n" + \
                          "--32452685f36942178a5f36fd94e34b63\r\n" + \
                          "Content-Disposition: form-data; name=\"firstname\";" + \
                          " filename=\"firstname\"\r\n\r\n" + \
                          "ben\r\n" + \
                          "--32452685f36942178a5f36fd94e34b63\r\n" + \
                          "Content-Disposition: form-data; name=\"key\";" + \
                          " filename=\"key\"\r\n\r\n" + \
                          "value\r\n" + \
                          "--32452685f36942178a5f36fd94e34b63--\r\n"
                    )
    fname = 'ben'
    lname = 'taylor'
    er = 'HTTP/1.0 200 OK\r\n'

    app = make_app()
    server.handle_connection(conn, 80, app)

    assert conn.sent[:len(er)] == er, 'Got: %s' % (repr(conn.sent),)

def test_file():
    conn = FakeConnection("GET /file HTTP/1.0\r\n\r\n")

    server.handle_connection(conn)

    assert 'HTTP/1.0 200 OK\r\n' in conn.sent, 'Got: %s' % (repr(conn.sent),)
    assert '<iframe src="/file.txt" type="text/plain"></iframe>' in conn.sent, 'Got: %s' % (repr(conn.sent),)

def test_content():
    conn = FakeConnection("GET /content HTTP/1.0\r\n\r\n")

    server.handle_connection(conn)

    assert 'HTTP/1.0 200 OK\r\n' in conn.sent, 'Got: %s' % (repr(conn.sent),)
    assert '<h1>Content page!</h1>' in conn.sent, 'Got: %s' % (repr(conn.sent),)

def test_handle_file_get():
    conn = FakeConnection("GET /file HTTP/1.0\r\n\r\n")
    header = 'HTTP/1.0 200 OK\r\n' + \
             'Content-type: text/plain\r\n' + \
             '\r\n'
    server.handle_connection(conn, 8500)
    assert conn.sent.startswith(header), 'Got: %s' % (repr(conn.sent),)