# coding: utf-8
"""Tests for the compilerop module."""
# -----------------------------------------------------------------------------
#  Copyright (C) 2010-2011 The IPython Development Team.
#
#  Distributed under the terms of the BSD License.
#
#  The full license is in the file COPYING.txt, distributed with this software.
# -----------------------------------------------------------------------------

# -----------------------------------------------------------------------------
# Imports
# -----------------------------------------------------------------------------

# Stdlib imports
import linecache
import sys

# Our own imports
from IPython.core import compilerop

# -----------------------------------------------------------------------------
# Test functions
# -----------------------------------------------------------------------------


def test_code_name():
    code = "x=1"
    name = compilerop.code_name(code)
    assert name.startswith("<ipython-input-0")


def test_code_name2():
    code = "x=1"
    name = compilerop.code_name(code, 9)
    assert name.startswith("<ipython-input-9")


def test_cache():
    """Test the compiler correctly compiles and caches inputs"""
    cp = compilerop.CachingCompiler()
    ncache = len(linecache.cache)
    cp.cache("x=1")
    assert len(linecache.cache) > ncache


def test_proper_default_encoding():
    # Check we're in a proper Python 2 environment (some imports, such
    # as GTK, can change the default encoding, which can hide bugs.)
    assert sys.getdefaultencoding() == "utf-8"


def test_cache_unicode():
    cp = compilerop.CachingCompiler()
    ncache = len(linecache.cache)
    cp.cache("t = 'žćčšđ'")
    assert len(linecache.cache) > ncache


def test_compiler_check_cache():
    """Test the compiler properly manages the cache."""
    # Rather simple-minded tests that just exercise the API
    cp = compilerop.CachingCompiler()
    cp.cache("x=1", 99)
    # Ensure now that after clearing the cache, our entries survive
    linecache.checkcache()
    assert any(
        k.startswith("<ipython-input-99") for k in linecache.cache
    ), "Entry for input-99 missing from linecache"
