Tests: added tests for "return" action.

This commit is contained in:
Andrei Zeliankou 2020-03-27 15:50:09 +00:00
parent 5f2d07019c
commit 6e5b5d2a0b
4 changed files with 191 additions and 270 deletions

103
test/test_return.py Normal file
View file

@ -0,0 +1,103 @@
import re
import unittest
from unit.applications.proto import TestApplicationProto
class TestReturn(TestApplicationProto):
prerequisites = {}
def setUp(self):
super().setUp()
self._load_conf(
{
"listeners": {"*:7080": {"pass": "routes"}},
"routes": [{"action": {"return": 200}}],
"applications": {},
}
)
def get_resps_sc(self, req=10):
to_send = b"""GET / HTTP/1.1
Host: localhost
""" * (
req - 1
)
to_send += b"""GET / HTTP/1.1
Host: localhost
Connection: close
"""
return self.http(to_send, raw_resp=True, raw=True)
def test_return(self):
resp = self.get()
self.assertEqual(resp['status'], 200)
self.assertIn('Server', resp['headers'])
self.assertIn('Date', resp['headers'])
self.assertEqual(resp['headers']['Content-Length'], '0')
self.assertEqual(resp['headers']['Connection'], 'close')
self.assertEqual(resp['body'], '', 'body')
resp = self.post(body='blah')
self.assertEqual(resp['status'], 200)
self.assertEqual(resp['body'], '', 'body')
resp = self.get_resps_sc()
self.assertEqual(len(re.findall('200 OK', resp)), 10)
self.assertEqual(len(re.findall('Connection:', resp)), 1)
self.assertEqual(len(re.findall('Connection: close', resp)), 1)
resp = self.get(http_10=True)
self.assertEqual(resp['status'], 200)
self.assertIn('Server', resp['headers'])
self.assertIn('Date', resp['headers'])
self.assertEqual(resp['headers']['Content-Length'], '0')
self.assertNotIn('Connection', resp['headers'])
self.assertEqual(resp['body'], '', 'body')
def test_return_update(self):
self.assertIn('success', self.conf('0', 'routes/0/action/return'))
resp = self.get()
self.assertEqual(resp['status'], 0)
self.assertEqual(resp['body'], '')
self.assertIn('success', self.conf('404', 'routes/0/action/return'))
resp = self.get()
self.assertEqual(resp['status'], 404)
self.assertNotEqual(resp['body'], '')
self.assertIn('success', self.conf('598', 'routes/0/action/return'))
resp = self.get()
self.assertEqual(resp['status'], 598)
self.assertNotEqual(resp['body'], '')
self.assertIn('success', self.conf('999', 'routes/0/action/return'))
resp = self.get()
self.assertEqual(resp['status'], 999)
self.assertEqual(resp['body'], '')
def test_return_invalid(self):
def check_error(conf):
self.assertIn('error', self.conf(conf, 'routes/0/action'))
check_error({"return": "200"})
check_error({"return": []})
check_error({"return": 80.})
check_error({"return": 1000})
check_error({"return": 200, "share": "/blah"})
self.assertIn(
'error', self.conf('001', 'routes/0/action/return'), 'leading zero'
)
if __name__ == '__main__':
TestReturn.main()

View file

@ -16,27 +16,10 @@ class TestRouting(TestApplicationProto):
"routes": [
{
"match": {"method": "GET"},
"action": {"pass": "applications/empty"},
"action": {"return": 200},
}
],
"applications": {
"empty": {
"type": "python",
"processes": {"spare": 0},
"path": self.current_dir + '/python/empty',
"working_directory": self.current_dir
+ '/python/empty',
"module": "wsgi",
},
"mirror": {
"type": "python",
"processes": {"spare": 0},
"path": self.current_dir + '/python/mirror',
"working_directory": self.current_dir
+ '/python/mirror',
"module": "wsgi",
},
},
"applications": {},
}
),
'routing configure',
@ -48,18 +31,14 @@ class TestRouting(TestApplicationProto):
def route_match(self, match):
self.assertIn(
'success',
self.route(
{"match": match, "action": {"pass": "applications/empty"}}
),
self.route({"match": match, "action": {"return": 200}}),
'route match configure',
)
def route_match_invalid(self, match):
self.assertIn(
'error',
self.route(
{"match": match, "action": {"pass": "applications/empty"}}
),
self.route({"match": match, "action": {"return": 200}}),
'route match configure invalid',
)
@ -233,24 +212,7 @@ class TestRouting(TestApplicationProto):
{
"listeners": {"*:7080": {"pass": "routes/main"}},
"routes": {"main": []},
"applications": {
"empty": {
"type": "python",
"processes": {"spare": 0},
"path": self.current_dir + '/python/empty',
"working_directory": self.current_dir
+ '/python/empty',
"module": "wsgi",
},
"mirror": {
"type": "python",
"processes": {"spare": 0},
"path": self.current_dir + '/python/mirror',
"working_directory": self.current_dir
+ '/python/mirror',
"module": "wsgi",
},
},
"applications": {},
}
),
'route empty configure',
@ -272,7 +234,7 @@ class TestRouting(TestApplicationProto):
def test_routes_route_match_absent(self):
self.assertIn(
'success',
self.conf([{"action": {"pass": "applications/empty"}}], 'routes'),
self.conf([{"action": {"return": 200}}], 'routes'),
'route match absent configure',
)
@ -349,14 +311,8 @@ class TestRouting(TestApplicationProto):
'success',
self.conf(
[
{
"match": {"method": "GET"},
"action": {"pass": "applications/empty"},
},
{
"match": {"method": "POST"},
"action": {"pass": "applications/mirror"},
},
{"match": {"method": "GET"}, "action": {"return": 200}},
{"match": {"method": "POST"}, "action": {"return": 201}},
],
'routes',
),
@ -364,18 +320,7 @@ class TestRouting(TestApplicationProto):
)
self.assertEqual(self.get()['status'], 200, 'rules two match first')
self.assertEqual(
self.post(
headers={
'Host': 'localhost',
'Content-Type': 'text/html',
'Connection': 'close',
},
body='X',
)['status'],
200,
'rules two match second',
)
self.assertEqual(self.post()['status'], 201, 'rules two match second')
def test_routes_two(self):
self.assertIn(
@ -393,20 +338,11 @@ class TestRouting(TestApplicationProto):
"second": [
{
"match": {"host": "localhost"},
"action": {"pass": "applications/empty"},
"action": {"return": 200},
}
],
},
"applications": {
"empty": {
"type": "python",
"processes": {"spare": 0},
"path": self.current_dir + '/python/empty',
"working_directory": self.current_dir
+ '/python/empty',
"module": "wsgi",
}
},
"applications": {},
}
),
'routes two configure',
@ -556,7 +492,7 @@ class TestRouting(TestApplicationProto):
self.assertIn(
'success',
self.conf([{"action": {"pass": "applications/empty"}}], 'routes'),
self.conf([{"action": {"return": 200}}], 'routes'),
'redefine 2',
)
self.assertEqual(self.get()['status'], 200, 'redefine request 2')
@ -569,19 +505,8 @@ class TestRouting(TestApplicationProto):
self.conf(
{
"listeners": {"*:7080": {"pass": "routes/main"}},
"routes": {
"main": [{"action": {"pass": "applications/empty"}}]
},
"applications": {
"empty": {
"type": "python",
"processes": {"spare": 0},
"path": self.current_dir + '/python/empty',
"working_directory": self.current_dir
+ '/python/empty',
"module": "wsgi",
}
},
"routes": {"main": [{"action": {"return": 200}}]},
"applications": {},
}
),
'redefine 4',
@ -595,25 +520,19 @@ class TestRouting(TestApplicationProto):
self.assertIn(
'success',
self.conf_post(
{"action": {"pass": "applications/empty"}}, 'routes/main'
),
self.conf_post({"action": {"return": 200}}, 'routes/main'),
'redefine 6',
)
self.assertEqual(self.get()['status'], 200, 'redefine request 6')
self.assertIn(
'error',
self.conf(
{"action": {"pass": "applications/empty"}}, 'routes/main/2'
),
self.conf({"action": {"return": 200}}, 'routes/main/2'),
'redefine 7',
)
self.assertIn(
'success',
self.conf(
{"action": {"pass": "applications/empty"}}, 'routes/main/1'
),
self.conf({"action": {"return": 201}}, 'routes/main/1'),
'redefine 8',
)
@ -631,10 +550,7 @@ class TestRouting(TestApplicationProto):
self.assertIn(
'success',
self.conf_post(
{
"match": {"method": "POST"},
"action": {"pass": "applications/empty"},
},
{"match": {"method": "POST"}, "action": {"return": 200}},
'routes',
),
'routes edit configure 2',
@ -654,9 +570,7 @@ class TestRouting(TestApplicationProto):
self.assertEqual(self.post()['status'], 200, 'routes edit POST 2')
self.assertIn(
'success',
self.conf_delete('routes/0'),
'routes edit configure 3',
'success', self.conf_delete('routes/0'), 'routes edit configure 3',
)
self.assertEqual(self.get()['status'], 404, 'routes edit GET 3')
@ -682,9 +596,7 @@ class TestRouting(TestApplicationProto):
self.assertEqual(self.post()['status'], 200, 'routes edit POST 4')
self.assertIn(
'success',
self.conf_delete('routes/0'),
'routes edit configure 5',
'success', self.conf_delete('routes/0'), 'routes edit configure 5',
)
self.assertEqual(self.get()['status'], 404, 'routes edit GET 5')
@ -693,10 +605,7 @@ class TestRouting(TestApplicationProto):
self.assertIn(
'success',
self.conf_post(
{
"match": {"method": "POST"},
"action": {"pass": "applications/empty"},
},
{"match": {"method": "POST"}, "action": {"return": 200},},
'routes',
),
'routes edit configure 6',
@ -710,19 +619,8 @@ class TestRouting(TestApplicationProto):
self.conf(
{
"listeners": {"*:7080": {"pass": "routes/main"}},
"routes": {
"main": [{"action": {"pass": "applications/empty"}}]
},
"applications": {
"empty": {
"type": "python",
"processes": {"spare": 0},
"path": self.current_dir + '/python/empty',
"working_directory": self.current_dir
+ '/python/empty',
"module": "wsgi",
}
},
"routes": {"main": [{"action": {"return": 200}}]},
"applications": {},
}
),
'route edit configure 7',
@ -1838,20 +1736,11 @@ class TestRouting(TestApplicationProto):
"second": [
{
"match": {"destination": ["127.0.0.1:7081"]},
"action": {"pass": "applications/empty"},
"action": {"return": 200},
}
],
},
"applications": {
"empty": {
"type": "python",
"processes": {"spare": 0},
"path": self.current_dir + "/python/empty",
"working_directory": self.current_dir
+ "/python/empty",
"module": "wsgi",
}
},
"applications": {},
}
),
'proxy configure',

View file

@ -2,9 +2,9 @@ from unit.applications.tls import TestApplicationTLS
class TestRoutingTLS(TestApplicationTLS):
prerequisites = {'modules': ['python', 'openssl']}
prerequisites = {'modules': ['openssl']}
def test_routes_match_scheme(self):
def test_routes_match_scheme_tls(self):
self.certificate()
self.assertIn(
@ -21,35 +21,21 @@ class TestRoutingTLS(TestApplicationTLS):
"routes": [
{
"match": {"scheme": "http"},
"action": {"pass": "applications/empty"},
"action": {"return": 200},
},
{
"match": {"scheme": "https"},
"action": {"pass": "applications/204_no_content"},
"action": {"return": 201},
},
],
"applications": {
"empty": {
"type": "python",
"processes": {"spare": 0},
"path": self.current_dir + "/python/empty",
"module": "wsgi",
},
"204_no_content": {
"type": "python",
"processes": {"spare": 0},
"path": self.current_dir
+ "/python/204_no_content",
"module": "wsgi",
},
},
"applications": {},
}
),
'scheme configure',
)
self.assertEqual(self.get()['status'], 200, 'http')
self.assertEqual(self.get_ssl(port=7081)['status'], 204, 'https')
self.assertEqual(self.get_ssl(port=7081)['status'], 201, 'https')
if __name__ == '__main__':

View file

@ -1,10 +1,10 @@
import os
import unittest
from unit.applications.lang.python import TestApplicationPython
from unit.applications.proto import TestApplicationProto
class TestStatic(TestApplicationPython):
prerequisites = {'modules': ['python']}
class TestStatic(TestApplicationProto):
prerequisites = {}
def setUp(self):
super().setUp()
@ -20,19 +20,10 @@ class TestStatic(TestApplicationPython):
{
"listeners": {
"*:7080": {"pass": "routes"},
"*:7081": {"pass": "applications/empty"},
"*:7081": {"pass": "routes"},
},
"routes": [{"action": {"share": self.testdir + "/assets"}}],
"applications": {
"empty": {
"type": "python",
"processes": {"spare": 0},
"path": self.current_dir + "/python/empty",
"working_directory": self.current_dir
+ "/python/empty",
"module": "wsgi",
}
},
"applications": {},
}
)
@ -41,37 +32,22 @@ class TestStatic(TestApplicationPython):
super().tearDown()
def action_update(self, conf):
self.assertIn('success', self.conf(conf, 'routes/0/action'))
def test_fallback(self):
self.assertIn(
'success',
self.conf({"share": "/blah"}, 'routes/0/action'),
'configure bad path no fallback',
)
self.action_update({"share": "/blah"})
self.assertEqual(self.get()['status'], 404, 'bad path no fallback')
self.assertIn(
'success',
self.conf(
{"share": "/blah", "fallback": {"pass": "applications/empty"}},
'routes/0/action',
),
'configure bad path fallback',
)
self.action_update({"share": "/blah", "fallback": {"return": 200}})
resp = self.get()
self.assertEqual(resp['status'], 200, 'bad path fallback status')
self.assertEqual(resp['body'], '', 'bad path fallback')
def test_fallback_valid_path(self):
self.assertIn(
'success',
self.conf(
{
"share": self.testdir + "/assets",
"fallback": {"pass": "applications/empty"},
},
'routes/0/action',
),
'configure fallback',
self.action_update(
{"share": self.testdir + "/assets", "fallback": {"return": 200}}
)
resp = self.get()
self.assertEqual(resp['status'], 200, 'fallback status')
@ -90,36 +66,28 @@ class TestStatic(TestApplicationPython):
)
def test_fallback_nested(self):
self.assertIn(
'success',
self.conf(
{
"share": "/blah",
"fallback": {
"share": "/blah/blah",
"fallback": {"pass": "applications/empty"},
},
self.action_update(
{
"share": "/blah",
"fallback": {
"share": "/blah/blah",
"fallback": {"return": 200},
},
'routes/0/action',
),
'configure fallback nested',
}
)
resp = self.get()
self.assertEqual(resp['status'], 200, 'fallback nested status')
self.assertEqual(resp['body'], '', 'fallback nested')
def test_fallback_share(self):
self.assertIn(
'success',
self.conf(
{
"share": "/blah",
"fallback": {"share": self.testdir + "/assets"},
},
'routes/0/action',
),
'configure fallback share',
self.action_update(
{
"share": "/blah",
"fallback": {"share": self.testdir + "/assets"},
}
)
resp = self.get()
self.assertEqual(resp['status'], 200, 'fallback share status')
self.assertEqual(resp['body'], '0123456789', 'fallback share')
@ -136,76 +104,51 @@ class TestStatic(TestApplicationPython):
self.assertIn(
'success',
self.conf(
{
"share": "/blah",
"fallback": {"proxy": "http://127.0.0.1:7081"},
},
'routes/0/action',
[
{
"match": {"destination": "*:7081"},
"action": {"return": 200},
},
{
"action": {
"share": "/blah",
"fallback": {"proxy": "http://127.0.0.1:7081"},
}
},
],
'routes',
),
'configure fallback proxy',
'configure fallback proxy route',
)
resp = self.get()
self.assertEqual(resp['status'], 200, 'fallback proxy status')
self.assertEqual(resp['body'], '', 'fallback proxy')
@unittest.skip('not yet')
def test_fallback_proxy_cycle(self):
self.assertIn(
'success',
self.conf(
{
"share": "/blah",
"fallback": {"proxy": "http://127.0.0.1:7080"},
},
'routes/0/action',
),
'configure fallback cycle',
self.action_update(
{
"share": "/blah",
"fallback": {"proxy": "http://127.0.0.1:7080"},
}
)
self.assertNotEqual(self.get()['status'], 200, 'fallback cycle')
self.assertIn(
'success', self.conf_delete('listeners/*:7081'), 'delete listener'
)
self.assertIn('success', self.conf_delete('listeners/*:7081'))
self.assertNotEqual(self.get()['status'], 200, 'fallback cycle 2')
def test_fallback_invalid(self):
self.assertIn(
'error',
self.conf({"share": "/blah", "fallback": {}}, 'routes/0/action'),
'configure fallback empty',
)
self.assertIn(
'error',
self.conf({"share": "/blah", "fallback": ""}, 'routes/0/action'),
'configure fallback not object',
)
self.assertIn(
'error',
self.conf(
{
"proxy": "http://127.0.0.1:7081",
"fallback": {"share": "/blah"},
},
'routes/0/action',
),
'configure fallback proxy invalid',
)
self.assertIn(
'error',
self.conf(
{
"pass": "applications/empty",
"fallback": {"share": "/blah"},
},
'routes/0/action',
),
'configure fallback pass invalid',
)
self.assertIn(
'error',
self.conf({"fallback": {"share": "/blah"}}, 'routes/0/action'),
'configure fallback only',
def check_error(conf):
self.assertIn('error', self.conf(conf, 'routes/0/action'))
check_error({"share": "/blah", "fallback": {}})
check_error({"share": "/blah", "fallback": ""})
check_error({"return": 200, "fallback": {"share": "/blah"}})
check_error(
{"proxy": "http://127.0.0.1:7081", "fallback": {"share": "/blah"}}
)
check_error({"fallback": {"share": "/blah"}})
if __name__ == '__main__':