"""
pycraigslist.tests.test_api.test_limit
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Tests the `limit` parameter in pycraigslist .search and .search_detail methods.
"""

import inspect
import warnings
from pytest import mark

import params
import pycraigslist


def test_zero_limits():
    """Tests number of posts when ignoring `limit` kwarg and when setting `limit`
    kwarg to 0. Both should return full listing."""
    bia = pycraigslist.forsale.bia()
    len_posts = len([post for post in bia.search()])
    expected_len = bia.count
    assert len_posts == expected_len

    len_posts = len([post for post in bia.search(limit=0)])
    expected_len = bia.count
    assert len_posts == expected_len


@mark.parametrize("limit", params.limits.std)
def test_limits(limit):
    """Tests number of posts always equal the search limit, or the maximum
    posts of page if exceeded by limit."""
    apa = pycraigslist.housing.apa()
    len_posts = len([post for post in apa.search(limit=limit)])
    expected_len = apa.count

    if expected_len > len_posts:
        assert len_posts == limit
    else:
        assert len_posts == expected_len


@mark.parametrize("limit", params.limits.std_fringe)
def test_fringe_limits(limit):
    """Tests number of posts always equal the search limit, or the maximum
    posts of page if exceeded by limit. This test captures API behavior when
    using limit values near the fringes of listings pages."""
    apa = pycraigslist.housing.apa()
    len_posts = len([post for post in apa.search(limit=limit)])
    expected_len = apa.count

    if expected_len > len_posts:
        assert len_posts == limit
    else:
        assert len_posts == expected_len


@mark.parametrize("limit", params.limits.detail)
def test_detail_limits(limit):
    """Tests number of detailed posts always equal the search limit, or the maximum
    posts of page if exceeded by limit."""
    cta = pycraigslist.forsale.cta()
    len_posts = len([post for post in cta.search_detail(limit=limit)])
    expected_len = cta.count

    if expected_len > len_posts:
        assert len_posts == limit
    else:
        assert len_posts == expected_len


@mark.parametrize("limit", params.limits.detail_fringe)
def test_fringe_detail_limits(limit):
    """Tests number of detailed posts always equal the search limit, or the maximum
    posts of page if exceeded by limit. This test captures API behavior when using
    limit values near the fringes of listings pages."""
    cta = pycraigslist.forsale.cta()
    len_posts = len([post for post in cta.search_detail(limit=limit)])
    expected_len = cta.count

    try:
        if expected_len > len_posts:
            assert len_posts == limit
        else:
            assert len_posts == expected_len
    except AssertionError:
        # Number of fetched posts DOES NOT match number requested.
        # When failed, the expected limit is usually off by 1 or 2 posts - throw warning instead of failing test.
        warnings.warn(
            UserWarning(
                "encountered AssertionError for function %s -- gathered limit: %d. Adjust if issue persists."
                % (inspect.currentframe().f_code.co_name, len_posts)
            )
        )
        assert True
