import pytest

from tests.factories.rule import QueryRuleFactory
from tests.factories.segment import SegmentFactory


@pytest.mark.django_db
def test_request_query_rule(client, site):
    segment = SegmentFactory(name='Query')
    QueryRuleFactory(
        parameter="query",
        value="value",
        segment=segment,
    )

    response = client.get('/?query=value')
    assert response.status_code == 200

    assert any(
        item['encoded_name'] == 'query' for item in client.session['segments'])


@pytest.mark.django_db
def test_request_only_one_query_rule(client, site):
    segment = SegmentFactory(name='Query')
    QueryRuleFactory(
        parameter="query",
        value="value",
        segment=segment
    )

    response = client.get('/?test=test&query=value')
    assert response.status_code == 200
    assert any(
        item['encoded_name'] == 'query' for item in client.session['segments'])


@pytest.mark.django_db
def test_request_multiple_queries(client, site):
    segment = SegmentFactory(name='Multiple queries')
    QueryRuleFactory(
        parameter="test",
        value="test",
        segment=segment
    )

    QueryRuleFactory(
        parameter="query",
        value="value",
        segment=segment,
    )

    response = client.get('/?test=test&query=value')
    assert response.status_code == 200
    assert any(
        item['encoded_name'] == 'multiple-queries'
        for item in client.session['segments']
    )


@pytest.mark.django_db
def test_request_persistent_segmenting(client, site):
    segment = SegmentFactory(name='Persistent', persistent=True)
    QueryRuleFactory(
        parameter="test",
        value="test",
        segment=segment
    )

    response = client.get('/?test=test')
    assert response.status_code == 200

    assert any(
        item['encoded_name'] == 'persistent'
        for item in client.session['segments'])

    response = client.get('/')
    assert response.status_code == 200
    assert any(
        item['encoded_name'] == 'persistent'
        for item in client.session['segments'])


@pytest.mark.django_db
def test_request_non_persistent_segmenting(client, site):
    segment = SegmentFactory(name='Non Persistent')
    QueryRuleFactory(
        parameter="test",
        value="test",
        segment=segment
    )

    response = client.get('/?test=test')
    assert response.status_code == 200
    assert any(
        item['encoded_name'] == 'non-persistent'
        for item in client.session['segments'])

    response = client.get('/')
    assert response.status_code == 200

    assert not any(
        item['encoded_name'] == 'non-persistent'
        for item in client.session['segments'])


@pytest.mark.django_db
def test_request_match_any_segmenting(client, site):
    segment = SegmentFactory(name='Match any', match_any=True)
    QueryRuleFactory(
        parameter='test',
        value='test',
        segment=segment,
    )
    QueryRuleFactory(
        parameter='test2',
        value='test2',
        segment=segment
    )

    response = client.get('/?test=test')
    assert response.status_code == 200

    assert any(
        item['encoded_name'] == 'match-any'
        for item in client.session['segments'])