TeamspeakStats/tsstats/tests/test_cache.py

# -*- coding: utf-8 -*-
import os
import shutil
from pickle import UnpicklingError

import pytest

from tsstats.cache import Cache, CachedLog, _calculate_hash
from tsstats.log import parse_logs
from tsstats.tests.test_log import testlog_path


@pytest.fixture
def cache_path(request):
    cache_path = 'tsstats/tests/res/tsstats.cache.{}'.format(
        request.function.__name__
    )

    def clean():
        if os.path.exists(cache_path):
            os.remove(cache_path)

    request.addfinalizer(clean)
    yield cache_path


@pytest.fixture
def cache(cache_path):
    return Cache(cache_path)


# UNIT
def test_cache_add_get_remove_iter_len(cache):
    expected_cachedlog = CachedLog(
        testlog_path, _calculate_hash(testlog_path), []
    )
    cache[testlog_path] = []
    assert cache.store[testlog_path] == expected_cachedlog
    assert cache[testlog_path] == expected_cachedlog
    assert next(iter(cache)) == testlog_path
    assert len(cache) == 1
    del cache[testlog_path]
    assert len(cache) == 0


def test_cache_read_write(cache):
    cache[testlog_path] = []
    cache.write()
    cache2 = Cache.read(cache.path)
    assert cache2[testlog_path] == CachedLog(
        testlog_path, _calculate_hash(testlog_path), []
    )


def test_cache_needs_parsing(cache, tmpdir):
    tmplog_path = str(tmpdir.mkdir('cache').join('test.log'))
    # copy logfile to temporary location
    shutil.copy(testlog_path, tmplog_path)

    assert cache.needs_parsing(tmplog_path)
    cache[tmplog_path] = []
    assert not cache.needs_parsing(tmplog_path)
    with open(tmplog_path, 'a') as f:
        f.writelines(['content'])
    assert cache.needs_parsing(tmplog_path)


def test_cache_read_broken_file():
    cache = Cache.read('tsstats/__init__.py')
    assert isinstance(cache, Cache)


# INTEGRATION
def test_cache_integration(cache_path):
    first_run = list(parse_logs(
        testlog_path, online_dc=False, cache_path=cache_path
    ))
    second_run = list(parse_logs(
        testlog_path, online_dc=False, cache_path=cache_path
    ))
    assert first_run == second_run


def test_same_result_without_cache(cache_path):
    first_run = list(parse_logs(
        testlog_path, online_dc=False
    ))
    second_run = list(parse_logs(
        testlog_path, online_dc=False, cache_path=cache_path
    ))
    third_run = list(parse_logs(
        testlog_path, online_dc=False, cache_path=cache_path
    ))
    assert first_run == second_run
    assert first_run == third_run
    assert second_run == third_run