TestRequireCookies

Complexity

Total Complexity

3

Size/Duplication

Total Lines	8
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

1 Method

Rating	Name	Duplication	Size	Complexity
A	test_invalid_require_cookies_decorator()	0	5	3

"""Test jinja filters."""

from flask import Flask
import pytest

from flask_extras import FlaskExtras
from flask_extras import decorators

app = Flask('test_flask_jsondash')
app.debug = True
app.secret_key = 'Foo'
FlaskExtras(app)


@app.route('/xhr-custom')
@decorators.xhr_only(status_code=400)
def foo_xhr_custom():
    return ''


@app.route('/xhr')
@decorators.xhr_only()
def foo_xhr():
    return ''


@app.route('/args')
@decorators.require_args(params=['foo', 'bar'])
def foo_args():
    return ''


@app.route('/cookies')
@decorators.require_cookies(['foo'])
def foo_cookies():
    return ''


@app.route('/headers')
@decorators.require_headers(headers=['X-Foo'])
def foo_headers():
    return ''

client = app.test_client()


class TestXhr:
    """Test class for function."""

    def test_invalid_xhr_decorator(self):
        """Test expected failure of function."""
        with app.app_context():
            res = client.get('/xhr')
            assert res.status_code == 415

    def test_invalid_xhr_custom_decorator(self):
        """Test expected success of function."""
        with app.app_context():
            res = client.get('/xhr-custom')
            assert res.status_code == 400


class TestRequireArgs:
    """Test class for function."""

    def test_invalid_require_args_decorator(self):
        """Test expected failure of function."""
        with pytest.raises(ValueError):
            with app.app_context():
                client.get('/args')

    def test_valid_require_args_decorator(self):
        """Test expected success of function."""
        with app.app_context():
            res = client.get('/args?foo=1&bar=1')
            assert res.status_code == 200


class TestRequireCookies:
    """Test class for function."""

    def test_invalid_require_cookies_decorator(self):
        """Test expected failure of function."""
        with pytest.raises(ValueError):
            with app.app_context():
                client.get('/cookies')


class TestRequireHeaders:
    """Test class for function."""

    def test_invalid_require_headers_decorator(self):
        """Test expected failure of function."""
        with pytest.raises(ValueError):
            with app.app_context():
                client.get('/headers')

    def test_valid_require_headers_decorator(self):
        """Test expected success of function."""
        with app.app_context():
            res = client.get('/headers', headers={'X-Foo': 'Foo'})
            assert res.status_code == 200