run api_server in multiprocessing.Thread other than gevent.pywsgi.WSGIServer

This commit is contained in:
debugtalk
2017-06-20 13:25:59 +08:00
parent b775e8c217
commit bb199bd42e
2 changed files with 11 additions and 21 deletions

View File

@@ -1,2 +0,0 @@
from gevent import monkey
monkey.patch_all()

View File

@@ -1,7 +1,7 @@
import gevent
import gevent.pywsgi
import requests
import multiprocessing
import time
import unittest
import requests
from . import api_server
class TestApiServer(unittest.TestCase):
@@ -10,16 +10,17 @@ class TestApiServer(unittest.TestCase):
"""
def setUp(self):
super(TestApiServer, self).setUp()
self._api_server = gevent.pywsgi.WSGIServer(("127.0.0.1", 0), api_server.app, log=None)
gevent.spawn(lambda: self._api_server.serve_forever())
gevent.sleep(0.01)
self.host = "http://127.0.0.1:%i" % self._api_server.server_port
self.api_server_process = multiprocessing.Process(
target=api_server.app.run
)
self.api_server_process.start()
time.sleep(0.1)
self.host = "http://127.0.0.1:5000"
self.api_client = requests.Session()
def tearDown(self):
super(TestApiServer, self).tearDown()
self._api_server.stop_accepting()
self._api_server.stop()
self.api_server_process.terminate()
def clear_users(self):
url = "%s/api/users" % self.host
@@ -59,24 +60,21 @@ class TestApiServer(unittest.TestCase):
self.assertEqual(True, resp.json()['success'])
def test_create_user_not_existed(self):
self.clear_users()
resp = self.create_user(1000, 'user1', '123456')
self.assertEqual(201, resp.status_code)
self.assertEqual(True, resp.json()['success'])
def test_create_user_existed(self):
self.clear_users()
resp = self.create_user(1000, 'user1', '123456')
resp = self.create_user(1000, 'user1', '123456')
self.assertEqual(500, resp.status_code)
def test_get_users_empty(self):
self.clear_users()
resp = self.get_users()
self.assertEqual(200, resp.status_code)
self.assertEqual(resp.json()['count'], 0)
def test_get_users_not_empty(self):
self.clear_users()
resp = self.create_user(1000, 'user1', '123456')
resp = self.get_users()
self.assertEqual(200, resp.status_code)
@@ -88,39 +86,33 @@ class TestApiServer(unittest.TestCase):
self.assertEqual(resp.json()['count'], 2)
def test_get_user_not_existed(self):
self.clear_users()
resp = self.get_user(1000)
self.assertEqual(404, resp.status_code)
self.assertEqual(resp.json()['success'], False)
def test_get_user_existed(self):
self.clear_users()
self.create_user(1000, 'user1', '123456')
resp = self.get_user(1000)
self.assertEqual(200, resp.status_code)
self.assertEqual(resp.json()['success'], True)
def test_update_user_not_existed(self):
self.clear_users()
resp = self.update_user(1000, 'user1', '123456')
self.assertEqual(404, resp.status_code)
self.assertEqual(resp.json()['success'], False)
def test_update_user_existed(self):
self.clear_users()
self.create_user(1000, 'user1', '123456')
resp = self.update_user(1000, 'user2', '123456')
self.assertEqual(200, resp.status_code)
self.assertEqual(resp.json()['data']['name'], 'user2')
def test_delete_user_not_existed(self):
self.clear_users()
resp = self.delete_user(1000)
self.assertEqual(404, resp.status_code)
self.assertEqual(resp.json()['success'], False)
def test_delete_user_existed(self):
self.clear_users()
self.create_user(1000, 'leo', '123456')
resp = self.delete_user(1000)
self.assertEqual(200, resp.status_code)