diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/requests/test_registration.py (renamed from test/requests/test-registration.py) | 38 |
1 files changed, 10 insertions, 28 deletions
diff --git a/test/requests/test-registration.py b/test/requests/test_registration.py index bcb1642f..0047e8a6 100644 --- a/test/requests/test-registration.py +++ b/test/requests/test_registration.py @@ -1,27 +1,8 @@ import sys -import unittest import requests -import logging -from elasticsearch import Elasticsearch, TransportError -#from utility.tools import ELASTICSEARCH_HOST, ELASTICSEARCH_PORT +from parametrized_test import ParametrizedTest -GN2_SERVER = None -ES_SERVER = None - -class TestRegistration(unittest.TestCase): - - - def setUp(self): - self.url = GN2_SERVER+"/n/register" - self.es = Elasticsearch([ES_SERVER]) - self.es_cleanup = [] - - es_logger = logging.getLogger("elasticsearch") - es_logger.addHandler( - logging.FileHandler("/tmp/es_TestRegistrationInfo.log")) - es_trace_logger = logging.getLogger("elasticsearch.trace") - es_trace_logger.addHandler( - logging.FileHandler("/tmp/es_TestRegistrationTrace.log")) +class TestRegistration(ParametrizedTest): def tearDown(self): for item in self.es_cleanup: @@ -36,24 +17,25 @@ class TestRegistration(unittest.TestCase): "password": "test_password", "password_confirm": "test_password" } - requests.post(self.url, data) + requests.post(self.gn2_url+"/n/register", data) response = self.es.search( index="users" , doc_type="local" , body={ "query": {"match": {"email_address": "test@user.com"}}}) self.assertEqual(len(response["hits"]["hits"]), 1) - self.es_cleanup.append(response["hits"]["hits"][0]) else: self.skipTest("The elasticsearch server is down") -def main(): +def main(gn2, es): + import unittest suite = unittest.TestSuite() - suite.addTest(TestRegistration("testRegistrationPage")) + suite.addTest(TestRegistration(methodName="testRegistrationPage", gn2_url=gn2, es_url=es)) runner = unittest.TextTestRunner() runner.run(suite) if __name__ == "__main__": - GN2_SERVER = sys.argv[1] - ES_SERVER = sys.argv[2] - main() + if len(sys.argv) < 3: + raise Exception("Required arguments missing") + else: + main(sys.argv[1], sys.argv[2]) |