aboutsummaryrefslogtreecommitdiffstats
path: root/app_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'app_test.py')
-rw-r--r--app_test.py227
1 files changed, 208 insertions, 19 deletions
diff --git a/app_test.py b/app_test.py
index 896a47cd..97e38f4d 100644
--- a/app_test.py
+++ b/app_test.py
@@ -1,15 +1,22 @@
-import json # pragma: no cover
-import os # pragma: no cover
+import json
+import os
-from app import app
+from app import manager
-from flask_testing import TestCase # pragma: no cover
+from flask import Blueprint
+
+from flask_testing import TestCase
+
+manager.app.tests_blueprint = Blueprint("tests", __name__)
+manager.load_views(manager.app.tests_blueprint, "pysite/views/tests")
+manager.app.register_blueprint(manager.app.tests_blueprint)
+app = manager.app
class SiteTest(TestCase):
- ''' extend TestCase with flask app instantiation '''
+ """ extend TestCase with flask app instantiation """
def create_app(self):
- ''' add flask app configuration settings '''
+ """ add flask app configuration settings """
server_name = 'pytest.local'
app.config['TESTING'] = True
app.config['LIVESERVER_TIMEOUT'] = 10
@@ -20,37 +27,45 @@ class SiteTest(TestCase):
return app
-class RootEndpoint(SiteTest):
- ''' test cases for the root endpoint and error handling '''
+class BaseEndpoints(SiteTest):
+ """ test cases for the base endpoints """
def test_index(self):
- ''' Check the root path reponds with 200 OK '''
+ """ Check the root path reponds with 200 OK """
response = self.client.get('/', 'http://pytest.local')
self.assertEqual(response.status_code, 200)
- def test_not_found(self):
- ''' Check paths without handlers returns 404 Not Found '''
- response = self.client.get('/nonexistentpath')
- self.assertEqual(response.status_code, 404)
-
def test_invite(self):
- ''' Check invite redirects '''
+ """ Check invite redirects """
response = self.client.get('/invite')
self.assertEqual(response.status_code, 302)
+ def test_ws_test(self):
+ """ check ws_test responds """
+ response = self.client.get('/ws_test')
+ self.assertEqual(response.status_code, 200)
+
+ def test_datadog_redirect(self):
+ """ Check datadog path redirects """
+ response = self.client.get('/datadog')
+ self.assertEqual(response.status_code, 302)
+
+
+class ApiEndpoints(SiteTest):
+ """ test cases for the api subdomain """
def test_api_unknown_route(self):
- ''' Check api unknown route '''
+ """ Check api unknown route """
response = self.client.get('/', app.config['API_SUBDOMAIN'])
self.assertEqual(response.json, {'error_code': 0, 'error_message': 'Unknown API route'})
self.assertEqual(response.status_code, 404)
def test_api_healthcheck(self):
- ''' Check healthcheck url responds '''
+ """ Check healthcheck url responds """
response = self.client.get('/healthcheck', app.config['API_SUBDOMAIN'])
self.assertEqual(response.json, {'status': 'ok'})
self.assertEqual(response.status_code, 200)
def test_api_tag(self):
- ''' Check tag api '''
+ """ Check tag api """
os.environ['BOT_API_KEY'] = 'abcdefg'
headers = {'X-API-Key': 'abcdefg', 'Content-Type': 'application/json'}
good_data = json.dumps({
@@ -80,7 +95,7 @@ class RootEndpoint(SiteTest):
self.assertEqual(response.status_code, 200)
def test_api_user(self):
- ''' Check insert user '''
+ """ Check insert user """
os.environ['BOT_API_KEY'] = 'abcdefg'
headers = {'X-API-Key': 'abcdefg', 'Content-Type': 'application/json'}
bad_data = json.dumps({'user_id': 1234, 'role': 5678})
@@ -94,3 +109,177 @@ class RootEndpoint(SiteTest):
response = self.client.post('/user', app.config['API_SUBDOMAIN'], headers=headers, data=good_data)
self.assertEqual(True, "inserted" in response.json)
+
+ def test_api_route_errors(self):
+ """ Check api route errors """
+ from pysite.base_route import APIView
+ from pysite.constants import ErrorCodes
+
+ av = APIView()
+ av.error(ErrorCodes.unauthorized)
+ av.error(ErrorCodes.bad_data_format)
+
+ def test_not_found(self):
+ """ Check paths without handlers returns 404 Not Found """
+ response = self.client.get('/nonexistentpath')
+ self.assertEqual(response.status_code, 404)
+
+
+class StaffEndpoints(SiteTest):
+ """ Test cases for staff subdomain """
+ def test_staff_view(self):
+ """ Check staff view renders """
+ from pysite.views.staff.index import StaffView
+ sv = StaffView()
+ result = sv.get()
+ self.assertEqual(result.startswith('<!DOCTYPE html>'), True)
+
+ response = self.client.get('/', app.config['STAFF_SUBDOMAIN'])
+ self.assertEqual(response.status_code, 200)
+
+
+class Utilities(SiteTest):
+ """ Test cases for internal utility code """
+ def test_logging_runtime_error(self):
+ """ Check that a wrong value for log level raises runtime error """
+ os.environ['LOG_LEVEL'] = 'wrong value'
+ try:
+ import pysite.__init__ # noqa: F401
+ except RuntimeError:
+ return True
+ finally:
+ os.environ['LOG_LEVEL'] = 'info'
+ raise Exception('Expected a failure due to wrong LOG_LEVEL attribute name')
+
+ def test_error_view_runtime_error(self):
+ """ Check that wrong values for error view setup raises runtime error """
+ import pysite.base_route
+
+ ev = pysite.base_route.ErrorView()
+ try:
+ ev.setup('sdf', 'sdfsdf')
+ except RuntimeError:
+ return True
+ raise Exception('Expected runtime error on setup() when giving wrongful arguments')
+
+
+class MixinTests(SiteTest):
+ """ Test cases for mixins """
+ def test_dbmixin_runtime_error(self):
+ """ Check that wrong values for error view setup raises runtime error """
+ from pysite.mixins import DBMixin
+
+ dbm = DBMixin()
+ try:
+ dbm.setup('sdf', 'sdfsdf')
+ except RuntimeError:
+ return True
+ raise Exception('Expected runtime error on setup() when giving wrongful arguments')
+
+ def test_dbmixin_table_property(self):
+ """ Check the table property returns correctly """
+ from pysite.mixins import DBMixin
+
+ try:
+ dbm = DBMixin()
+ dbm.table_name = 'Table'
+ self.assertEquals(dbm.table, 'Table')
+ except AttributeError:
+ pass
+
+ def test_handler_5xx(self):
+ """ Check error view returns error message """
+ from werkzeug.exceptions import InternalServerError
+ from pysite.views.error_handlers import http_5xx
+
+ error_view = http_5xx.Error404View()
+ error_message = error_view.get(InternalServerError)
+ self.assertEqual(error_message, ('Internal server error. Please try again later!', 500))
+
+ def test_route_view_runtime_error(self):
+ """ Check that wrong values for route view setup raises runtime error """
+ from pysite.base_route import RouteView
+
+ rv = RouteView()
+ try:
+ rv.setup('sdf', 'sdfsdf')
+ except RuntimeError:
+ return True
+ raise Exception('Expected runtime error on setup() when giving wrongful arguments')
+
+ def test_route_manager(self):
+ """ Check route manager """
+ from pysite.route_manager import RouteManager
+ os.environ['WEBPAGE_SECRET_KEY'] = 'super_secret'
+ rm = RouteManager()
+ self.assertEqual(rm.app.secret_key, 'super_secret')
+
+
+class DecoratorTests(SiteTest):
+ def test_decorator_api_json(self):
+ """ Check the json validation decorator """
+ from pysite.decorators import api_params
+ from pysite.constants import ValidationTypes
+ from schema import Schema
+
+ SCHEMA = Schema([{"user_id": int, "role": int}])
+
+ @api_params(schema=SCHEMA, validation_type=ValidationTypes.json)
+ def try_json_type(data):
+ return data
+
+ try:
+ try_json_type("not json")
+ except Exception as error_message:
+ self.assertEqual(type(error_message), AttributeError)
+
+ def test_decorator_params(self):
+ """ Check the params validation decorator """
+
+ response = self.client.post('/testparams?test=params')
+
+ self.assertEqual(response.status_code, 200)
+ self.assertEqual(response.json, [{'test': 'params'}])
+
+
+class DatabaseTests(SiteTest):
+ """ Test cases for the database module """
+ def test_table_actions(self):
+ import string
+ import secrets
+ from pysite.database import RethinkDB
+
+ alphabet = string.ascii_letters
+ generated_table_name = ''.join(secrets.choice(alphabet) for i in range(8))
+
+ rdb = RethinkDB()
+ # Create table name and expect it to work
+ result = rdb.create_table(generated_table_name)
+ self.assertEquals(result, True)
+
+ # Create the same table name and expect it to already exist
+ result = rdb.create_table(generated_table_name)
+ self.assertEquals(result, False)
+
+ # Drop table and expect it to work
+ result = rdb.drop_table(generated_table_name)
+ self.assertEquals(result, True)
+
+ # Drop the same table and expect it to already be gone
+ result = rdb.drop_table(generated_table_name)
+ self.assertEquals(result, False)
+
+ # This is to get some more code coverage
+ self.assertEquals(rdb.teardown_request('_'), None)
+
+
+class TestWebsocketEcho(SiteTest):
+ """ Test cases for the echo endpoint """
+ def testEcho(self):
+ """ Check rudimentary websockets handlers work """
+ from geventwebsocket.websocket import WebSocket
+ from pysite.views.ws.echo import EchoWebsocket
+ ew = EchoWebsocket(WebSocket)
+ ew.on_open()
+ ew.on_message('message')
+ ew.on_close()
generated by cgit v1.2.3 (git 2.39.1) at 2025-10-18 19:21:18 +0000