test_cmd.py

import os
import unittest

from click.testing import CliRunner
from pyq.pyq import main


def pjoin(*path):
    return os.path.join(os.path.dirname(__file__), *path)


class TestASTMatchEngine(unittest.TestCase):
    def setUp(self):
        self.runner = CliRunner()

        # chdir to testfiles/cmd
        self.currentdir = os.getcwd()
        os.chdir(pjoin('testfiles', 'cmd'))

    def tearDown(self):
        # restore cwd
        os.chdir(self.currentdir)

    def invoke(self, *args):
        return self.runner.invoke(*args, catch_exceptions=False)

    def test_noargs(self):
        result = self.invoke(main, [])

        self.assertNotEqual(result.exit_code, 0)
        self.assertIn('Missing argument "selector"', result.output_bytes)

    def test_nodir(self):
        result = self.invoke(main, ['def'])
        output = result.output_bytes.splitlines()

        self.assertEqual(result.exit_code, 0)
        self.assertEqual(len(output), 4)
        self.assertEqual(output[0], 'cmd.py:7  def foo(self):')
        self.assertEqual(output[1], 'cmd.py:11  def baz(arg1, arg2):')
        self.assertEqual(output[2], 'file2.py:1  def hello():')
        self.assertEqual(output[3], '.test_hidden_dir/foo.py:1  def bar():')

    def test_notpyfile(self):
        result = self.invoke(main, ['def', 'notpyfile.txt'])

        self.assertEqual(result.exit_code, 0)

    def test_file(self):
        result = self.invoke(main, ['> def', 'cmd.py'])
        output = result.output_bytes.splitlines()

        self.assertEqual(result.exit_code, 0)
        self.assertEqual(output[0], 'cmd.py:11  def baz(arg1, arg2):')

    def test_wildcard(self):
        result = self.invoke(main, ['def', 'cmd.py', 'file2.py',
                                           'notpyfile.txt', 'nofile.unknown'])
        output = result.output_bytes.splitlines()

        self.assertEqual(result.exit_code, 0)
        self.assertEqual(len(output), 3)
        self.assertEqual(output[0], 'cmd.py:7  def foo(self):')
        self.assertEqual(output[1], 'cmd.py:11  def baz(arg1, arg2):')
        self.assertEqual(output[2], 'file2.py:1  def hello():')

    def test_print_filenames(self):
        result = self.invoke(main, ['-l', 'def'])
        output = result.output_bytes.splitlines()

        self.assertEqual(result.exit_code, 0)
        self.assertEqual(output[0], 'cmd.py')
        self.assertEqual(output[1], 'file2.py')

    def test_ignoredir(self):
        r = self.invoke(main, ['-l', 'class'])
        output = r.output_bytes.splitlines()

        self.assertEqual(r.exit_code, 0)
        self.assertTrue(any('ignoredir' in p for p in output))

        r = self.invoke(main, ['-l', 'class', '--ignore-dir', 'ignoredir'])
        output = r.output_bytes.splitlines()

        self.assertEqual(r.exit_code, 0)
        self.assertFalse(any('ignoredir' in p for p in output))

    def test_ignoredir_norecurse(self):
        r = self.invoke(main, ['-l', 'class', '--ignore-dir', 'ignoredir2'])
        output = r.output_bytes.splitlines()

        self.assertEqual(r.exit_code, 0)
        self.assertFalse(any('ignoredir2' in p for p in output))

        r = self.invoke(main, ['-l', 'class', '--ignore-dir', 'ignoredir2',
                               '-n'])
        output = r.output_bytes.splitlines()

        self.assertEqual(r.exit_code, 0)
        self.assertTrue(any('ignoredir2' in p for p in output))

    def test_expand_matches(self):
        r1 = self.invoke(main, ['call', 'cmd.py'])
        r2 = self.invoke(main, ['-e', 'call', 'cmd.py'])
        r3 = self.invoke(main, ['-el', 'call', 'cmd.py'])

        output1 = r1.output_bytes.splitlines()
        output2 = r2.output_bytes.splitlines()
        output3 = r3.output_bytes.splitlines()

        self.assertEqual(r1.exit_code, 0)
        self.assertEqual(len(output1), 1)
        self.assertEqual(output1[0], 'cmd.py:15  foo() | bar()')

        self.assertEqual(r2.exit_code, 0)
        self.assertEqual(len(output2), 2)
        self.assertEqual(output2[0], 'cmd.py:15:0  foo() | bar()')
        self.assertEqual(output2[1], 'cmd.py:15:8  foo() | bar()')

        self.assertEqual(r3.exit_code, 0)
        self.assertEqual(len(output3), 1)
        self.assertEqual(output3[0], 'cmd.py')


if __name__ == '__main__':
    unittest.main()