Untitled

mail@pastecode.io avatarunknown
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()