mirror of
https://github.com/Casvt/MIND.git
synced 2026-02-19 11:54:46 -05:00
Fixed API tests
This commit is contained in:
@@ -3,7 +3,7 @@ import unittest
|
||||
from flask import Blueprint
|
||||
|
||||
from backend.custom_exceptions import *
|
||||
from frontend.api import api, auth, error_handler, return_api
|
||||
from frontend.api import api, return_api
|
||||
|
||||
class Test_API(unittest.TestCase):
|
||||
def test_blueprint(self):
|
||||
@@ -22,30 +22,4 @@ class Test_API(unittest.TestCase):
|
||||
self.assertEqual(result[1], case['code'])
|
||||
else:
|
||||
self.assertEqual(result[1], 200)
|
||||
|
||||
def test_auth(self):
|
||||
method = lambda x: x
|
||||
result = auth(method)
|
||||
self.assertEqual(result.__name__, method.__name__)
|
||||
|
||||
def _raise_exception(self, e, *args):
|
||||
raise e(*args)
|
||||
|
||||
def test_error_handler(self):
|
||||
result = error_handler(self._raise_exception)
|
||||
self.assertEqual(result.__name__, self._raise_exception.__name__)
|
||||
self.assertEqual(result(UsernameTaken), return_api(**UsernameTaken.api_response))
|
||||
self.assertEqual(result(UsernameInvalid), return_api(**UsernameInvalid.api_response))
|
||||
self.assertEqual(result(UserNotFound), return_api(**UserNotFound.api_response))
|
||||
self.assertEqual(result(AccessUnauthorized), return_api(**AccessUnauthorized.api_response))
|
||||
self.assertEqual(result(ReminderNotFound), return_api(**ReminderNotFound.api_response))
|
||||
self.assertEqual(result(NotificationServiceNotFound), return_api(**NotificationServiceNotFound.api_response))
|
||||
self.assertEqual(result(InvalidTime), return_api(**InvalidTime.api_response))
|
||||
self.assertEqual(result(NotificationServiceInUse, 'test'), return_api(**NotificationServiceInUse('test').api_response))
|
||||
self.assertEqual(result(KeyNotFound, 'test'), return_api(**KeyNotFound('test').api_response))
|
||||
self.assertEqual(result(InvalidKeyValue, 'test', 'value'), return_api(**InvalidKeyValue('test', 'value').api_response))
|
||||
with self.assertRaises(TypeError):
|
||||
result(TypeError)
|
||||
with self.assertRaises(KeyError):
|
||||
result(KeyError)
|
||||
|
||||
Reference in New Issue
Block a user