# vim:fileencoding=utf-8
# License: BSD
# Copyright: 2015, Kovid Goyal <kovid at kovidgoyal.net>

linter = require('../tools/lint.mjs')

async def lint(code):
    return await linter.lint_code(code, {'filename':'<test>', 'report':def():
                                   pass
})

async def err_test(code, ident, line, name, other_line):
    msgs = await lint(code)
    assrt.ok(len(msgs) > 0, 'failed to find error in: ' + code)
    assrt.equal(len(msgs), 1, 'found ' + (len(msgs) - 1) + ' extra errors in: ' + code)
    assrt.equal(msgs[0].ident, ident, 'incorrect ident: ' + msgs[0].ident + ' for: ' + code)
    if line != undefined:
        assrt.equal(msgs[0].start_line, line, 'incorrect line: ' + msgs[0].start_line + ' for: ' + code)
    if name != undefined:
        assrt.equal(msgs[0].name, name, 'incorrect name: ' + msgs[0].name + ' for: ' + code)
    if other_line != undefined:
        assrt.equal(msgs[0].other_line, other_line, 'incorrect other_line: ' + msgs[0].other_line + ' for: ' + code)

async def ok_test(code):
    msgs = await lint(code)
    assrt.equal(msgs.length, 0, 'Got unexpected msg: ' + (msgs[0] or {}).message + ' for code: ' + code)

async def run_tests():
    # Imports
    await err_test('import foo', 'unused-import', 1, 'foo')
    await err_test('import foo.boo', 'unused-import', 1, 'foo')
    await err_test('import foo as boo', 'unused-import', 1, 'boo')
    await err_test('from x import foo', 'unused-import', 1, 'foo')
    await err_test('from x import foo as boo', 'unused-import', 1, 'boo')
    await ok_test('import foo\nfoo')

    # Function arguments
    await ok_test('def f(a):\n return a')
    await ok_test('def f(a=1):\n return a')
    await ok_test('def f(*a):\n return a')
    await ok_test('def f(**a):\n return a')
    await ok_test('def f(a):\n return 1')

    # Extended slices
    await ok_test('a = []; a[::2], a[1:-1]')

    # Top level unused ignored
    await ok_test('a=1\ndef f():\n pass')

    # Destructuring assignment
    await ok_test('def f():\n a, b = 1, 2; a, b')
    await ok_test('def f():\n a = 1; b, c = a, 2; return b, c')
    await ok_test('for x, (y, z) in [ [1, [2, 3]] ]:\n x + y + z')
    await err_test('def f():\n a, b = 1, 1; return a', 'unused-local', 2, 'b')

    # Compound assignment
    await err_test('a += 1', 'undef', 1, 'a')
    await ok_test('def f():\n a = 1; a += 1')

    # Unused bindings
    await err_test('def f():\n a=1', 'unused-local', 2, 'a')
    await err_test('def f():\n def a():\n  pass', 'unused-local', 2, 'a')

    # Undefined references
    await err_test('f()', 'undef', 1, 'f')
    await err_test('a', 'undef', 1, 'a')
    await err_test('def f():\n a=1; return a\na', 'undef', 3, 'a')

    # Comprehensions
    await ok_test('[x for x in [1,2]]')
    await ok_test('[1 for x in [1,2] if x]')
    await ok_test('[1 for x in [1,2]]')
    await ok_test('def f():\n l=[1,2];[x for x in l]')
    await ok_test('def f():\n l=[1,2];{x:True for x in l}')
    await err_test('def f():\n [x for x in [1,2]]; return x', 'undef', 2, 'x')

    # Loops
    await ok_test('def f():\n for x in "":\n  pass\n return x')
    await ok_test('def f():\n for x in "":\n  x += 1\n')
    await ok_test('def f():\n for x, y in "":\n  pass\n return x, y')
    await ok_test('for v"var i = 0; i < 1; i++":\n print(i)')
    await err_test('def f():\n a = 1\n for a in "":\n  a', 'loop-shadowed', 3, 'a', 2)

    # Semi-colons
    await err_test('a=1;;a', 'extra-semicolon', 1, ';')
    await err_test('a=1;', 'eol-semicolon', 1, ';')
    await ok_test('a = """\nhello;\n"""')
    await ok_test("a = '''\nhello;\n'''")

    # Builtins
    for k in 'String Symbol this self window Map arguments print len range dir undefined'.split(' '):
        await ok_test(k)

    # noqa
    await ok_test('f() # noqa')
    await ok_test('f() # noqa:undef')
    await err_test('f() # noqa:xxx', 'undef')

    # Named func in branch
    await err_test('if 1:\n def f():\n  pass', 'func-in-branch', 2, 'f')
    await err_test('if 1:\n pass\nelse:\n def f():\n  pass', 'func-in-branch', 4, 'f')
    await err_test('try:\n def f():\n  pass\nexcept:\n pass', 'func-in-branch', 2, 'f')
    await ok_test('if 1:\n a = def():\n  pass')

    # Syntax errors
    await err_test('def f(:\n pass', 'syntax-err', 1)

    # Functions
    await ok_test('def f():\n pass\nf()')

    # Non-locals
    await err_test('def a():\n x = 1\n def b():\n  nonlocal x\n b()', 'unused-local', 2, 'x')
    await ok_test('def a():\n x = 1\n def b():\n  nonlocal x\n  x\n b()')
    await ok_test('def a():\n x = 1\n def b():\n  nonlocal x\n  x\n  x = 2\n b()')
    await ok_test('def a():\n x = 1\n def ():\n  nonlocal x\n  x = 1\n  x')
    await ok_test('nonlocal a\na()')

    # Define after use
    await err_test('def g():\n f()\n f()\n def f():\n  pass', 'def-after-use', 3, 'f')

    # Decorators
    await ok_test('from x import y\n@y\ndef f():\n pass')

    # try/except
    await ok_test('try:\n 1\nexcept Error as e:\n e')

    # Classes
    await ok_test('''
class A:

    h = j = 1
    x = h + j
    if True:
        k = 1
    else:
        k = 2

    def __init__(self, a):
        self.a = a

    def unused(self):
        pass
    other = unused

class B(A):

    def __init__(self):
        A.__init__(self, 'b')
''')
    await err_test('class A:\n def a(self):\n  a()', 'undef', 3, 'a')
    await err_test('class A:\n def a(self):\n  pass\n def a(self):\n  pass', 'dup-method', 4, 'a')

    # Object literals
    await err_test('{"a":1, "a":2}', 'dup-key', 1, 'a')

    # keyword arg values
    await ok_test('def f():\n a=1\n f(b=a)')
    await ok_test('def f():\n a={}\n f(**a)')

    # with statement
    await ok_test('''
def f():
    a = 1
    with a as b:
        b()
''')
    await err_test('with a:\n pass', 'undef', '1', 'a')

__test_async_done__ = run_tests()