from wagtail.search.backends import get_search_backend

from tests.conftest import EventPageFactory, GenericPageFactory


SEARCH_QUERY = """
query Search($query: String) {
  results: search(query: $query) {
    __typename
    ... on PageInterface {
      title
    }
  }
}
"""


def _index(page):
    # Wagtail's post_save signal enqueues indexing via django-tasks, which isn't
    # drained synchronously in tests. Call the backend directly so the page is
    # findable through the live search code path.
    get_search_backend().add(page)


def _titles_for(body, typename):
    return [r["title"] for r in body["data"]["results"] if r["__typename"] == typename]


def test_search_returns_live_generic_page(home_page, graphql_post):
    page = GenericPageFactory(
        parent=home_page,
        title="PublishedGenericSearchToken",
        slug="published-generic-search",
    )
    _index(page)

    response, body = graphql_post(SEARCH_QUERY, {"query": "PublishedGenericSearchToken"})

    assert response.status_code == 200
    assert "errors" not in body, body
    assert "PublishedGenericSearchToken" in _titles_for(body, "GenericPage")


def test_search_excludes_draft_generic_page(home_page, graphql_post):
    page = GenericPageFactory(
        parent=home_page,
        title="DraftGenericSearchToken",
        slug="draft-generic-search",
        live=False,
    )
    _index(page)

    response, body = graphql_post(SEARCH_QUERY, {"query": "DraftGenericSearchToken"})

    assert response.status_code == 200
    assert "errors" not in body, body
    assert "DraftGenericSearchToken" not in _titles_for(body, "GenericPage")


def test_search_returns_live_event_page(home_page, event_index, graphql_post):
    page = EventPageFactory(
        parent=event_index,
        title="PublishedEventSearchToken",
        slug="published-event-search",
    )
    _index(page)

    response, body = graphql_post(SEARCH_QUERY, {"query": "PublishedEventSearchToken"})

    assert response.status_code == 200
    assert "errors" not in body, body
    assert "PublishedEventSearchToken" in _titles_for(body, "EventPage")


def test_search_excludes_draft_event_page(home_page, event_index, graphql_post):
    page = EventPageFactory(
        parent=event_index,
        title="DraftEventSearchToken",
        slug="draft-event-search",
        live=False,
    )
    _index(page)

    response, body = graphql_post(SEARCH_QUERY, {"query": "DraftEventSearchToken"})

    assert response.status_code == 200
    assert "errors" not in body, body
    assert "DraftEventSearchToken" not in _titles_for(body, "EventPage")