test_everything.py 文件源码

def test_custom_role_limits(self):

        def before_create_app():
            """Create a resource which uses overrides for
            role limits.

            """

            class TestResource(Resource):
                decorators = [
                    limits.limiter.limit(
                        limits.RoleBasedLimits(free='1 per day'),
                    ),
                ]

                def get(self):
                    return {'lol': 'lol!'}

            api.add_resource(TestResource, '/test')

        client = setup_client(before_create_app)

        response = client.get('/v1/test')
        data = json.loads(response.get_data(as_text=True))
        assert 'lol' in data

        response = client.get('/v1/test')
        assert response.status_code == 429