"""Tests for Validation."""

from unittest.mock import MagicMock, patch

from flask import Request, Response

from account.constants import header
from account.validation import validation


def run_with_reject_grass_headers(headers):
    """Submit request with headers and get response."""
    request = MagicMock(spec=Request, headers=headers)
    success_response = MagicMock(spec=Response, status_code=200)

    def inner():
        return success_response

    with patch.object(validation, 'request', request):
        return validation.reject_grass_headers(inner)()


def test_reject_grass_headers():
    """Test that 400 return when grass request made."""
    result = run_with_reject_grass_headers(
        {header.GRASS_ACCOUNT_ID: 1234, header.GRASS_ACCOUNT_TYPE: 'subaccount'}
    )
    assert result.status_code == 400


def test_reject_grass_headers_passing():
    """Test that inner function called when not grass."""
    result = run_with_reject_grass_headers({'foo': 'bar'})
    assert result.status_code == 200