Source

pypy-effect-analysis / lib_pypy / ctypes_config_cache / test / test_cache.py

Full commit
import py
import sys, os
from pypy.tool.udir import udir

dirpath = py.path.local(__file__).dirpath().dirpath()


def run(filename, outputname):
    filepath = dirpath.join(filename)
    tmpdir = udir.ensure('testcache-' + os.path.splitext(filename)[0],
                         dir=True)
    tmpdir.join('dumpcache.py').write(dirpath.join('dumpcache.py').read())
    path = sys.path[:]
    try:
        sys.path.insert(0, str(tmpdir))
        execfile(str(filepath), {})
    finally:
        sys.path[:] = path
        sys.modules.pop('dumpcache', None)
    #
    outputpath = tmpdir.join(outputname)
    assert outputpath.check(exists=1)
    modname = os.path.splitext(outputname)[0]
    try:
        sys.path.insert(0, str(tmpdir))
        d = {}
        execfile(str(outputpath), d)
    finally:
        sys.path[:] = path
    return d


def test_syslog():
    d = run('syslog.ctc.py', '_syslog_cache.py')
    assert 'LOG_NOTICE' in d

def test_resource():
    d = run('resource.ctc.py', '_resource_cache.py')
    assert 'RLIM_NLIMITS' in d

def test_pyexpat():
    d = run('pyexpat.ctc.py', '_pyexpat_cache.py')
    assert 'XML_COMBINED_VERSION' in d

def test_locale():
    d = run('locale.ctc.py', '_locale_cache.py')
    assert 'LC_ALL' in d
    assert 'CHAR_MAX' in d