import pytest

from pappyproxy.requestcache import RequestCache, RequestCacheIterator
from pappyproxy.http import Request, Response, get_request
from pappyproxy.util import PappyException

def gen_reqs(n):
    ret = []
    for i in range(1, n+1):
        r = get_request('https://www.kdjasdasdi.sadfasdf')
        r.headers['Test-Id'] = i
        r.reqid = str(i)
        ret.append(r)
    return ret

@pytest.inlineCallbacks
def test_cache_simple():
    reqs = gen_reqs(5)
    cache = RequestCache(5)
    cache.add(reqs[0])
    g = yield cache.get('1')
    assert g == reqs[0]
    
def test_cache_evict():
    reqs = gen_reqs(5)
    cache = RequestCache(3)
    cache.add(reqs[0])
    cache.add(reqs[1])
    cache.add(reqs[2])
    cache.add(reqs[3])
    assert not cache.check(reqs[0].reqid)
    assert cache.check(reqs[1].reqid)
    assert cache.check(reqs[2].reqid)
    assert cache.check(reqs[3].reqid)
    
    # Testing the implementation
    assert reqs[0].reqid not in cache._cached_reqs
    assert reqs[1].reqid in cache._cached_reqs
    assert reqs[2].reqid in cache._cached_reqs
    assert reqs[3].reqid in cache._cached_reqs

@pytest.inlineCallbacks
def test_cache_lru():
    reqs = gen_reqs(5)
    cache = RequestCache(3)
    cache.add(reqs[0])
    cache.add(reqs[1])
    cache.add(reqs[2])
    yield cache.get(reqs[0].reqid)
    cache.add(reqs[3])
    assert cache.check(reqs[0].reqid)
    assert not cache.check(reqs[1].reqid)
    assert cache.check(reqs[2].reqid)
    assert cache.check(reqs[3].reqid)
    
    # Testing the implementation
    assert reqs[0].reqid in cache._cached_reqs
    assert reqs[1].reqid not in cache._cached_reqs
    assert reqs[2].reqid in cache._cached_reqs
    assert reqs[3].reqid in cache._cached_reqs

@pytest.inlineCallbacks
def test_cache_lru_add():
    reqs = gen_reqs(5)
    cache = RequestCache(3)
    cache.add(reqs[0])
    cache.add(reqs[1])
    cache.add(reqs[2])
    yield cache.add(reqs[0])
    cache.add(reqs[3])
    assert cache.check(reqs[0].reqid)
    assert not cache.check(reqs[1].reqid)
    assert cache.check(reqs[2].reqid)
    assert cache.check(reqs[3].reqid)
    
    # Testing the implementation
    assert reqs[0].reqid in cache._cached_reqs
    assert reqs[1].reqid not in cache._cached_reqs
    assert reqs[2].reqid in cache._cached_reqs
    assert reqs[3].reqid in cache._cached_reqs

@pytest.inlineCallbacks
def test_cache_inmem_simple():
    cache = RequestCache(3)
    req = gen_reqs(1)[0]
    req.reqid = None
    cache.add(req)
    assert req.reqid[0] == 'm'
    g = yield cache.get(req.reqid)
    assert req == g

def test_cache_inmem_evict():
    reqs = gen_reqs(5)
    cache = RequestCache(3)
    reqs[0].reqid = None
    reqs[1].reqid = None
    reqs[2].reqid = None
    reqs[3].reqid = None
    cache.add(reqs[0])
    cache.add(reqs[1])
    cache.add(reqs[2])
    cache.add(reqs[3])
    assert not cache.check(reqs[0].reqid)
    assert cache.check(reqs[1].reqid)
    assert cache.check(reqs[2].reqid)
    assert cache.check(reqs[3].reqid)
    
    # Testing the implementation
    assert reqs[0] in cache.inmem_reqs
    assert reqs[1] in cache.inmem_reqs
    assert reqs[2] in cache.inmem_reqs
    assert reqs[3] in cache.inmem_reqs

def test_req_ids():
    reqs = gen_reqs(5)
    cache = RequestCache(3)
    cache.add(reqs[0])
    cache.add(reqs[1])
    cache.add(reqs[2])
    cache.add(reqs[3])
    assert cache.req_ids() == ['4', '3', '2', '1']

def test_req_ids_unmangled():
    reqs = gen_reqs(5)
    cache = RequestCache(3)
    reqs[0].unmangled = reqs[4]
    cache.add(reqs[0])
    cache.add(reqs[4])
    cache.add(reqs[1])
    cache.add(reqs[2])
    cache.add(reqs[3])
    assert cache.req_ids() == ['4', '3', '2', '1']
    assert cache.req_ids(include_unmangled=True) == ['4', '3', '2', '5', '1']