67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
# Copyright 2014 Hewlett-Packard Development Company, L.P.
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
"""Tests for `glance.wsgi`."""
|
|
|
|
import re
|
|
import socket
|
|
import time
|
|
|
|
from oslo.config import cfg
|
|
import testtools
|
|
|
|
from glance.common import wsgi
|
|
|
|
CONF = cfg.CONF
|
|
|
|
|
|
class TestWSGIServer(testtools.TestCase):
|
|
"""WSGI server tests."""
|
|
def test_client_socket_timeout(self):
|
|
CONF.set_default("workers", 0)
|
|
CONF.set_default("client_socket_timeout", 0.1)
|
|
"""Verify connections are timed out as per 'client_socket_timeout'"""
|
|
greetings = 'Hello, World!!!'
|
|
|
|
def hello_world(env, start_response):
|
|
start_response('200 OK', [('Content-Type', 'text/plain')])
|
|
return [greetings]
|
|
|
|
server = wsgi.Server()
|
|
server.start(hello_world, 0)
|
|
port = server.sock.getsockname()[1]
|
|
sock1 = socket.socket()
|
|
sock1.connect(("127.0.0.1", port))
|
|
|
|
fd = sock1.makefile('rw')
|
|
fd.write(b'GET / HTTP/1.1\r\nHost: localhost\r\n\r\n')
|
|
fd.flush()
|
|
|
|
buf = fd.read()
|
|
# Should succeed - no timeout
|
|
self.assertTrue(re.search(greetings, buf))
|
|
|
|
sock2 = socket.socket()
|
|
sock2.connect(("127.0.0.1", port))
|
|
time.sleep(0.2)
|
|
|
|
fd = sock2.makefile('rw')
|
|
fd.write(b'GET / HTTP/1.1\r\nHost: localhost\r\n\r\n')
|
|
fd.flush()
|
|
|
|
buf = fd.read()
|
|
# Should fail - connection timed out so we get nothing from the server
|
|
self.assertFalse(buf)
|