Untitled
plain_text
18 days ago
1.4 kB
1
Indexable
Never
import json import unittest import tinyAPI import resy.user.api.error as multiple_list_error from unittest import mock from resy.user.api.MultipleList import MultipleList from tinyAPI.base.services.unit_testing import TransactionalDataStoreTestCase from resy.dispatcher import app from resy.user.api.UserInfo import UserInfo class TestCase(TransactionalDataStoreTestCase): def set_up(self): self.path = '/3/user/lists' app.config['TESTING'] = True self.app = app.test_client() self.user_id = 1 self.auth_token = UserInfo(self.user_id).get_session() self.headers = { 'Authorization': 'ResyAPI api_key="T7mNMYsBboQLyJEwFFK1ltZvKjNg2EeJ"', 'X-Resy-Auth-Token': self.auth_token, } @mock.patch('resy.user.api.MultipleList.get_user_list_count', side_effect=multiple_list_error.ListLimitExceededError) def test_400_list_limit_exceeded_error(self, mock_get_user_list_count): response = self.app.post( self.path, headers=self.headers, content_type='application/json', data=json.dumps({ 'name': 'My Favorite Venues', }) ) assert False if __name__ == '__main__': try: unittest.main() finally: tinyAPI.dsh().rollback(True) tinyAPI.dsh().close()