def test_404_on_no_routes():
poller = select.poll()
server = htserver.HttpServer(poller, [], CONFIG)
finished_client = False
def _client(ADDRESS):
nonlocal finished_client
sock = socket.socket()
sock.setblocking(True)
sock.connect(ADDRESS)
sock.send(b"GET /foo HTTP/1.1\r\n\r\n")
data = _drain(sock)
assert data == b'HTTP/1.0 404 Not Found\r\n\r\n'
finished_client = True
with threaded(_client, ADDRESS):
while True:
ready = poller.poll(1)
if not ready and not server.clients:
break
for info in ready:
server.handle_ready_socket(poller, *info)
assert finished_client
def test_large_header():
size = 1024
poller = select.poll()
routes = [('GET', '/foo', 0, _200_ok("hi"))]
server = htserver.HttpServer(poller, routes, CONFIG)
finished_clients = 0
def _foo_client(ADDRESS):
nonlocal finished_clients
sock = socket.socket()
sock.setblocking(True)
sock.connect(ADDRESS)
sock.send(
bytes(
"GET /foo HTTP/1.1\r\nX-Custom-Header: {}\r\n\r\n".format(
'foop' * size), 'ascii'))
data = _drain(sock)
assert str(data,
'ascii') == 'HTTP/1.0 200 OK\r\nContent-Length: 2\r\n\r\nhi'
finished_clients += 1
with threaded(_foo_client, ADDRESS):
while True:
ready = poller.poll(1)
if not ready and not server.clients:
break
for info in ready:
server.handle_ready_socket(poller, *info)
assert finished_clients == 1
def __init__(self):
iv = interrupt_vector.InterruptVector()
btns = buttons.Buttons(iv, conf.config().get('buttons', {}))
weather_devices = weather_stations.WeatherStations(conf.config().get(
'weather_stations', []))
self.target_loop_time = conf.config().get("main_loop",
{}).get("interval", 32)
self.minimum_loop_time = conf.config().get("main_loop",
{}).get("minimum", 1)
self.thinkers = [
iv, # buttons, switches, et al.
btns,
weather_devices
]
routes = [('GET', '/weather_stations', 0, weather_devices.list),
('GET', '/weather_stations/(.+)', 1, weather_devices.show)]
self.poller = select.poll()
if not hasattr(self.poller, 'ipoll'):
import uselect
self.poller = uselect.poll()
self.server = htserver.HttpServer(self.poller, routes,
conf.config().get('http_server', {}))
def test_regex_routes():
poller = select.poll()
routes = [('GET', re.compile('/foo/(.+)'), 1, _200_ok("")),
('GET', re.compile('/bar/(.+)/(.+)/id'), 2, _200_ok(""))]
server = htserver.HttpServer(poller, routes, CONFIG)
finished_clients = 0
def _foo_client(ADDRESS):
nonlocal finished_clients
sock = socket.socket()
sock.setblocking(True)
sock.connect(ADDRESS)
sock.send(b"GET /foo/my-id HTTP/1.1\r\n\r\n")
data = _drain(sock)
assert str(
data,
'ascii') == 'HTTP/1.0 200 OK\r\nContent-Length: 5\r\n\r\nmy-id'
finished_clients += 1
def _bar_client(ADDRESS):
nonlocal finished_clients
sock = socket.socket()
sock.setblocking(True)
sock.connect(ADDRESS)
sock.send(b"GET /bar/aaaa/bbbb/id HTTP/1.1\r\n\r\n")
data = _drain(sock)
assert str(
data,
'ascii') == 'HTTP/1.0 200 OK\r\nContent-Length: 9\r\n\r\naaaa bbbb'
finished_clients += 1
with threaded(_foo_client, ADDRESS):
while True:
ready = poller.poll(1)
if not ready and not server.clients:
break
for info in ready:
server.handle_ready_socket(poller, *info)
with threaded(_bar_client, ADDRESS):
while True:
ready = poller.poll(1)
if not ready and not server.clients:
break
for info in ready:
server.handle_ready_socket(poller, *info)
assert finished_clients == 2
def test_basic_routes():
poller = select.poll()
routes = [('GET', '/foo', 0, _200_ok("foo")),
('GET', '/bar', 0, _200_ok("bar"))]
server = htserver.HttpServer(poller, routes, CONFIG)
finished_clients = 0
def _foo_client(ADDRESS):
nonlocal finished_clients
sock = socket.socket()
sock.setblocking(True)
sock.connect(ADDRESS)
sock.send(b"GET /foo HTTP/1.1\r\n\r\n")
data = _drain(sock)
assert str(
data, 'ascii') == 'HTTP/1.0 200 OK\r\nContent-Length: 3\r\n\r\nfoo'
finished_clients += 1
def _bar_client(ADDRESS):
nonlocal finished_clients
sock = socket.socket()
sock.setblocking(True)
sock.connect(ADDRESS)
sock.send(b"GET /bar HTTP/1.1\r\n\r\n")
data = _drain(sock)
assert str(
data, 'ascii') == 'HTTP/1.0 200 OK\r\nContent-Length: 3\r\n\r\nbar'
finished_clients += 1
with threaded(_foo_client, ADDRESS):
while True:
ready = poller.poll(1)
if not ready and not server.clients:
break
for info in ready:
server.handle_ready_socket(poller, *info)
with threaded(_bar_client, ADDRESS):
while True:
ready = poller.poll(1)
if not ready and not server.clients:
break
for info in ready:
server.handle_ready_socket(poller, *info)
assert finished_clients == 2
def test_concurrent_clients():
read_size = 1024
write_size = 1024 * 1024 * 10
poller = select.poll()
routes = [('GET', '/foo', 0, _200_ok("foop" * write_size))]
server = htserver.HttpServer(poller, routes, CONFIG)
import threading
lock = threading.Lock()
finished_clients = 0
def _foo_client(ADDRESS):
nonlocal finished_clients
sock = socket.socket()
sock.setblocking(True)
sock.connect(ADDRESS)
sock.send(
bytes(
"GET /foo HTTP/1.1\r\nContent-Length: {}\r\n\r\n{}".format(
4 * read_size, 'foop' * read_size), 'ascii'))
data = _drain(sock)
assert str(
data, 'ascii'
) == 'HTTP/1.0 200 OK\r\nContent-Length: {}\r\n\r\n{}'.format(
4 * write_size, 'foop' * write_size)
with lock:
finished_clients += 1
with threaded(_foo_client, ADDRESS):
with threaded(_foo_client, ADDRESS):
with threaded(_foo_client, ADDRESS):
with threaded(_foo_client, ADDRESS):
with threaded(_foo_client, ADDRESS):
with threaded(_foo_client, ADDRESS):
while True:
ready = poller.poll(1)
if not ready and not server.clients:
break
for info in ready:
server.handle_ready_socket(poller, *info)
assert finished_clients == 6