diff --git a/test/base.py b/test/base.py new file mode 100644 index 00000000..1813a89b --- /dev/null +++ b/test/base.py @@ -0,0 +1,20 @@ +import multiprocessing +import time +import unittest +from . import api_server + +class ApiServerUnittest(unittest.TestCase): + """ + Test case class that sets up an HTTP server which can be used within the tests + """ + @classmethod + def setUpClass(cls): + cls.api_server_process = multiprocessing.Process( + target=api_server.app.run + ) + cls.api_server_process.start() + time.sleep(0.1) + + @classmethod + def tearDownClass(cls): + cls.api_server_process.terminate() diff --git a/test/test_apiserver.py b/test/test_apiserver.py index 9aef84ac..73ae3bc3 100644 --- a/test/test_apiserver.py +++ b/test/test_apiserver.py @@ -1,25 +1,7 @@ -import multiprocessing -import time -import unittest import requests -from . import api_server - -class TestApiServer(unittest.TestCase): - """ - Test case class that sets up an HTTP server which can be used within the tests - """ - @classmethod - def setUpClass(cls): - cls.api_server_process = multiprocessing.Process( - target=api_server.app.run - ) - cls.api_server_process.start() - time.sleep(0.1) - - @classmethod - def tearDownClass(cls): - cls.api_server_process.terminate() +from .base import ApiServerUnittest +class TestApiServer(ApiServerUnittest): def setUp(self): super(TestApiServer, self).setUp() self.host = "http://127.0.0.1:5000"