Tests: added basic infrastructure.
This commit is contained in:
Andrey Zelenkov
14
test/run.py
Executable file
14
test/run.py
Executable file
@@ -0,0 +1,14 @@
|
||||
#!/usr/bin/python3
|
||||
|
||||
import unittest
|
||||
import os
|
||||
|
||||
loader = unittest.TestLoader()
|
||||
suite = unittest.TestSuite()
|
||||
|
||||
this_dir = os.path.dirname(__file__)
|
||||
tests = loader.discover(start_dir=this_dir)
|
||||
suite.addTests(tests)
|
||||
|
||||
runner = unittest.TextTestRunner(verbosity=3)
|
||||
result = runner.run(suite)
|
||||
163
test/test_basic.py
Normal file
163
test/test_basic.py
Normal file
@@ -0,0 +1,163 @@
|
||||
import unit
|
||||
import unittest
|
||||
|
||||
class TestUnitBasic(unit.TestUnitControl):
|
||||
|
||||
def test_get(self):
|
||||
resp = self.get()
|
||||
self.assertEqual(resp, {'listeners': {}, 'applications': {}}, 'empty')
|
||||
self.assertEqual(self.get('/listeners'), {}, 'empty listeners prefix')
|
||||
self.assertEqual(self.get('/applications'), {},
|
||||
'empty applications prefix')
|
||||
|
||||
self.put('/applications', """
|
||||
{
|
||||
"app": {
|
||||
"type": "python",
|
||||
"workers": 1,
|
||||
"path": "/app",
|
||||
"module": "wsgi"
|
||||
}
|
||||
}
|
||||
""")
|
||||
|
||||
resp = self.get()
|
||||
|
||||
self.assertEqual(resp['listeners'], {}, 'python empty listeners')
|
||||
self.assertEqual(resp['applications'],
|
||||
{
|
||||
"app": {
|
||||
"type": "python",
|
||||
"workers": 1,
|
||||
"path": "/app",
|
||||
"module": "wsgi"
|
||||
}
|
||||
},
|
||||
'python applications')
|
||||
|
||||
self.assertEqual(self.get('/applications'),
|
||||
{
|
||||
"app": {
|
||||
"type": "python",
|
||||
"workers": 1,
|
||||
"path": "/app",
|
||||
"module":"wsgi"
|
||||
}
|
||||
},
|
||||
'python applications prefix')
|
||||
|
||||
self.assertEqual(self.get('/applications/app'),
|
||||
{
|
||||
"type": "python",
|
||||
"workers": 1,
|
||||
"path": "/app",
|
||||
"module": "wsgi"
|
||||
},
|
||||
'python applications prefix 2')
|
||||
|
||||
self.assertEqual(self.get('/applications/app/type'), 'python',
|
||||
'python applications type')
|
||||
self.assertEqual(self.get('/applications/app/workers'), 1,
|
||||
'python applications workers')
|
||||
|
||||
self.put('/listeners', '{"*:8080":{"application":"app"}}')
|
||||
|
||||
self.assertEqual(self.get()['listeners'],
|
||||
{"*:8080":{"application":"app"}}, 'python listeners')
|
||||
self.assertEqual(self.get('/listeners'),
|
||||
{"*:8080":{"application":"app"}}, 'python listeners prefix')
|
||||
self.assertEqual(self.get('/listeners/*:8080'),
|
||||
{"application":"app"}, 'python listeners prefix 2')
|
||||
self.assertEqual(self.get('/listeners/*:8080/application'), 'app',
|
||||
'python listeners application')
|
||||
|
||||
def test_put(self):
|
||||
self.put('/', """
|
||||
{
|
||||
"listeners": {
|
||||
"*:8080": {
|
||||
"application": "app"
|
||||
}
|
||||
},
|
||||
"applications": {
|
||||
"app": {
|
||||
"type": "python",
|
||||
"workers": 1,
|
||||
"path": "/app",
|
||||
"module": "wsgi"
|
||||
}
|
||||
}
|
||||
}
|
||||
""")
|
||||
|
||||
resp = self.get()
|
||||
|
||||
self.assertEqual(resp['listeners'], {"*:8080":{"application":"app"}},
|
||||
'put listeners')
|
||||
|
||||
self.assertEqual(resp['applications'],
|
||||
{
|
||||
"app": {
|
||||
"type": "python",
|
||||
"workers": 1,
|
||||
"path": "/app",
|
||||
"module": "wsgi"
|
||||
}
|
||||
},
|
||||
'put applications')
|
||||
|
||||
self.put('/listeners', '{"*:8081":{"application":"app"}}')
|
||||
self.assertEqual(self.get('/listeners'),
|
||||
{"*:8081": {"application":"app"}}, 'put listeners prefix')
|
||||
|
||||
self.put('/listeners/*:8080', '{"application":"app"}')
|
||||
|
||||
self.assertEqual(self.get('/listeners'),
|
||||
{
|
||||
"*:8080": {
|
||||
"application": "app"
|
||||
},
|
||||
"*:8081": {
|
||||
"application": "app"
|
||||
}
|
||||
},
|
||||
'put listeners prefix 3')
|
||||
|
||||
self.put('/applications/app/workers', '30')
|
||||
self.assertEqual(self.get('/applications/app/workers'), 30,
|
||||
'put applications workers')
|
||||
|
||||
self.put('/applications/app/path', '"/www"')
|
||||
self.assertEqual(self.get('/applications/app/path'), '/www',
|
||||
'put applications path')
|
||||
|
||||
def test_delete(self):
|
||||
self.put('/', """
|
||||
{
|
||||
"listeners": {
|
||||
"*:8080": {
|
||||
"application": "app"
|
||||
}
|
||||
},
|
||||
"applications": {
|
||||
"app": {
|
||||
"type": "python",
|
||||
"workers": 1,
|
||||
"path": "/app",
|
||||
"module": "wsgi"
|
||||
}
|
||||
}
|
||||
}
|
||||
""")
|
||||
|
||||
self.assertIn('error', self.delete('/applications/app'),
|
||||
'delete app before listener')
|
||||
self.assertIn('success', self.delete('/listeners/*:8080'),
|
||||
'delete listener')
|
||||
self.assertIn('success', self.delete('/applications/app'),
|
||||
'delete app after listener')
|
||||
self.assertIn('error', self.delete('/applications/app'),
|
||||
'delete app again')
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
136
test/test_configuration.py
Normal file
136
test/test_configuration.py
Normal file
@@ -0,0 +1,136 @@
|
||||
import unit
|
||||
import unittest
|
||||
|
||||
class TestUnitConfiguration(unit.TestUnitControl):
|
||||
|
||||
def test_json_applications(self):
|
||||
self.assertIn('error', self.put('/applications', '"{}"'),
|
||||
'applications string')
|
||||
self.assertIn('error', self.put('/applications', '{'),
|
||||
'applications miss brace')
|
||||
|
||||
self.assertIn('error', self.put('/applications', """
|
||||
{
|
||||
app": {
|
||||
"type": "python",
|
||||
"workers": 1,
|
||||
"path": "/app",
|
||||
"module": "wsgi"
|
||||
}
|
||||
}
|
||||
"""), 'applications miss quote')
|
||||
|
||||
self.assertIn('error', self.put('/applications', """
|
||||
{
|
||||
"app" {
|
||||
"type": "python",
|
||||
"workers": 1,
|
||||
"path": "/app",
|
||||
"module": "wsgi"
|
||||
}
|
||||
}
|
||||
"""), 'applications miss colon')
|
||||
|
||||
self.assertIn('error', self.put('/applications', """
|
||||
{
|
||||
"app": {
|
||||
"type": "python"
|
||||
"workers": 1,
|
||||
"path": "/app",
|
||||
"module": "wsgi"
|
||||
}
|
||||
}
|
||||
"""), 'applications miss comma')
|
||||
|
||||
self.assertIn('success', self.put('/applications', b'{ \n\r\t}'),
|
||||
'skip space')
|
||||
|
||||
self.assertIn('success', self.put('/applications', """
|
||||
{
|
||||
"app": {
|
||||
"type": "python",
|
||||
"workers": 1,
|
||||
"path": "../app",
|
||||
"module": "wsgi"
|
||||
}
|
||||
}
|
||||
"""), 'relative path')
|
||||
|
||||
self.assertIn('success', self.put('/applications', b"""
|
||||
{
|
||||
"ap\u0070": {
|
||||
"type": "\u0070ython",
|
||||
"workers": 1,
|
||||
"path": "\u002Fapp",
|
||||
"module": "wsgi"
|
||||
}
|
||||
}
|
||||
"""), 'unicode')
|
||||
|
||||
self.assertIn('success', self.put('/applications', """
|
||||
{
|
||||
"приложение": {
|
||||
"type": "python",
|
||||
"workers": 1,
|
||||
"path": "/app",
|
||||
"module": "wsgi"
|
||||
}
|
||||
}
|
||||
"""), 'unicode 2')
|
||||
|
||||
self.assertIn('error', self.put('/applications', b"""
|
||||
{
|
||||
"app": {
|
||||
"type": "python",
|
||||
"workers": \u0031,
|
||||
"path": "/app",
|
||||
"module": "wsgi"
|
||||
}
|
||||
}
|
||||
"""), 'unicode number')
|
||||
|
||||
def test_json_listeners(self):
|
||||
self.assertIn('error', self.put('/listeners',
|
||||
'{"*:8080":{"application":"app"}}'), 'listeners no app')
|
||||
|
||||
self.put('/applications', """
|
||||
{
|
||||
"app": {
|
||||
"type": "python",
|
||||
"workers": 1,
|
||||
"path": "/app",
|
||||
"module": "wsgi"
|
||||
}
|
||||
}
|
||||
""")
|
||||
|
||||
self.assertIn('success', self.put('/listeners',
|
||||
'{"*:8080":{"application":"app"}}'), 'listeners wildcard')
|
||||
self.assertIn('success', self.put('/listeners',
|
||||
'{"127.0.0.1:8081":{"application":"app"}}'), 'listeners explicit')
|
||||
self.assertIn('success', self.put('/listeners',
|
||||
'{"[::1]:8082":{"application":"app"}}'), 'listeners explicit ipv6')
|
||||
self.assertIn('error', self.put('/listeners',
|
||||
'{"127.0.0.1":{"application":"app"}}'), 'listeners no port')
|
||||
|
||||
@unittest.skip("TODO")
|
||||
def test_broken(self):
|
||||
self.assertIn('error', self.put('/', '00'), 'leading zero')
|
||||
self.assertIn('error', self.put('/listeners', '{"*:8080":{}}'),
|
||||
'listener empty')
|
||||
self.assertIn('error', self.put('/applications', '"type":"python"'),
|
||||
'application type only')
|
||||
|
||||
self.assertIn('error', self.put('/applications', """
|
||||
{
|
||||
"app": {
|
||||
"type": "python",
|
||||
"workers": 1,
|
||||
"path": "/app",
|
||||
"module": "wsgi"
|
||||
}
|
||||
}
|
||||
"""), 'negative workers')
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
112
test/unit.py
Normal file
112
test/unit.py
Normal file
@@ -0,0 +1,112 @@
|
||||
import os
|
||||
import re
|
||||
import sys
|
||||
import json
|
||||
import time
|
||||
import shutil
|
||||
import socket
|
||||
import tempfile
|
||||
import unittest
|
||||
import subprocess
|
||||
|
||||
class TestUnit(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
self.testdir = tempfile.mkdtemp(prefix='unit-test-')
|
||||
|
||||
os.mkdir(self.testdir + '/state')
|
||||
|
||||
pardir = os.path.abspath(os.path.join(os.path.dirname(__file__),
|
||||
os.pardir))
|
||||
|
||||
print()
|
||||
|
||||
subprocess.call([pardir + '/build/unitd',
|
||||
# TODO '--no-daemon',
|
||||
'--modules', pardir + '/build',
|
||||
'--state', self.testdir + '/state',
|
||||
'--pid', self.testdir + '/unit.pid',
|
||||
'--log', self.testdir + '/unit.log',
|
||||
'--control', 'unix:' + self.testdir + '/control.unit.sock'])
|
||||
|
||||
time_wait = 0
|
||||
while time_wait < 5 and not (os.path.exists(self.testdir + '/unit.pid')
|
||||
and os.path.exists(self.testdir + '/unit.log')
|
||||
and os.path.exists(self.testdir + '/control.unit.sock')):
|
||||
time.sleep(0.1)
|
||||
time_wait += 0.1
|
||||
|
||||
# TODO dependency check
|
||||
|
||||
def tearDown(self):
|
||||
with open(self.testdir + '/unit.pid', 'r') as f:
|
||||
pid = f.read().rstrip()
|
||||
|
||||
subprocess.call(['kill', pid])
|
||||
|
||||
time_wait = 0
|
||||
while time_wait < 5 and os.path.exists(self.testdir + '/unit.pid'):
|
||||
time.sleep(0.1)
|
||||
time_wait += 0.1
|
||||
|
||||
if '--log' in sys.argv:
|
||||
with open(self.testdir + '/unit.log', 'r') as f:
|
||||
print(f.read())
|
||||
|
||||
if '--leave' not in sys.argv:
|
||||
shutil.rmtree(self.testdir)
|
||||
|
||||
class TestUnitControl(TestUnit):
|
||||
|
||||
# TODO socket reuse
|
||||
# TODO http client
|
||||
|
||||
def get(self, path='/'):
|
||||
|
||||
with self._control_sock() as sock:
|
||||
sock.sendall(('GET ' + path
|
||||
+ ' HTTP/1.1\r\nHost: localhost\r\n\r\n').encode())
|
||||
r = self._recvall(sock)
|
||||
|
||||
return self._body_json(r)
|
||||
|
||||
def delete(self, path='/'):
|
||||
|
||||
with self._control_sock() as sock:
|
||||
sock.sendall(('DELETE ' + path
|
||||
+ ' HTTP/1.1\r\nHost: localhost\r\n\r\n').encode())
|
||||
r = self._recvall(sock)
|
||||
|
||||
return self._body_json(r)
|
||||
|
||||
def put(self, path='/', data=''):
|
||||
|
||||
if isinstance(data, str):
|
||||
data = data.encode()
|
||||
|
||||
with self._control_sock() as sock:
|
||||
sock.sendall(('PUT ' + path + (' HTTP/1.1\nHost: localhost\n'
|
||||
'Content-Length: ') + str(len(data)) + '\r\n\r\n').encode()
|
||||
+ data)
|
||||
r = self._recvall(sock)
|
||||
|
||||
return self._body_json(r)
|
||||
|
||||
def _control_sock(self):
|
||||
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
||||
sock.connect(self.testdir + '/control.unit.sock')
|
||||
return sock
|
||||
|
||||
def _recvall(self, sock, buff_size=4096):
|
||||
data = ''
|
||||
while True:
|
||||
part = sock.recv(buff_size).decode()
|
||||
data += part
|
||||
if len(part) < buff_size:
|
||||
break
|
||||
|
||||
return data
|
||||
|
||||
def _body_json(self, resp):
|
||||
m = re.search('.*?\x0d\x0a?\x0d\x0a?(.*)', resp, re.M | re.S)
|
||||
return json.loads(m.group(1))
|
||||
Reference in New Issue
Block a user