"""
Functional test for Python checker.
"""

import os
from testsupport import *

from snakefood.checker import *
from snakefood.find import ImportVisitor


def visit_source(source, cls):
    mod = compiler.parse(source)
    vis = cls()
    compiler.walk(mod, vis)
    return vis.finalize()

_import_tests = (
    ('import mymod', ['mymod']),
    ('import mod1, mod2, mod3', ['mod1', 'mod2','mod3']),
    ('import mod1 as mymod', ['mymod']),
    ('import mod1 as mymod, mod2 as mymod2', ['mymod', 'mymod2']),
    ('from mod1 import mymod', ['mymod']),
    ('from mod1 import mymod, mymod2', ['mymod', 'mymod2']),
    ('from mod1 import mymod as bli1', ['bli1']),
    ('from mod1 import mymod as bli1, mymod2 as bli2', ['bli1', 'bli2']),
    ('import os.path', ['os.path']),
    )

_names_tests = (
    ('fn = os.path.join("a", "b")', ['os', 'os.path', 'os.path.join']),
    ('a = 1; print b.c ; d.e = 3', ['b', 'b.c', 'd']),
    )

def test_checker_functional():
    for source, expected in _import_tests:
        found, _ = visit_source(source, ImportVisitor)
        actual = [x[2] for x in found if x[2] is not None]
        assert actual == expected, (actual, expected)

    for source, expected in _names_tests:
        dotted, simple = visit_source(source, NamesVisitor)
        actual = [x[0] for x in dotted]
        assert actual == expected, (actual, expected)




def test_checker_expected():
    checkdir = join(data, 'checker')
    pytocheck = [join(checkdir, fn) for fn in
                 os.listdir(checkdir) if fn.endswith('.py')]

    for fn in pytocheck:
        print 'Testing checker on: %s' % fn
        compare_expect(None, fn.replace('.py', '.expect'),
                       'sfood-checker', fn, filterdir=(data, 'ROOT'))