Gooey/gooey/source_parser_unittest.py

'''
Created on Feb 2, 2014

@author: Chris

TODO:
  - test no argparse module
  - test argparse in main
  - test argparse in try/catch
  -

'''

import os
import ast
import unittest

import source_parser


basic_pyfile = '''
import os

def say_jello():
  print "Jello!"

def main():
  print "hello!"
  parser = ArgumentParser(description='Example Argparse Program', formatter_class=RawDescriptionHelpFormatter)
  parser.add_argument("filename", help="filename")
  parser.add_argument("-r", "--recursive", dest="recurse", action="store_true",
                      help="recurse into subfolders [default: %(default)s]")
  parser.add_argument("-v", "--verbose", dest="verbose", action="count",
                      help="set verbosity level [default: %(default)s]")
  parser.add_argument("-i", "--include", action="append",
                      help="only include paths matching this regex pattern. Note: exclude is given preference over include. [default: %(default)s]",
                      metavar="RE")
  parser.add_argument("-m", "--mycoolargument", help="mycoolargument")
  parser.add_argument("-e", "--exclude", dest="exclude",
                      help="exclude paths matching this regex pattern. [default: %(default)s]", metavar="RE")
  parser.add_argument('-V', '--version', action='version')
  parser.add_argument('-T', '--tester', choices=['yes', 'no'])
  parser.add_argument(dest="paths", help="paths to folder(s) with source file(s) [default: %(default)s]",
                      metavar="path", nargs='+')

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



class TestSourceParser(unittest.TestCase):
  PATH = os.path.join(os.path.dirname(__file__), 'mockapplications')

  def module_path(self, name):
    return os.path.join(self.PATH, name)

  def setUp(self):
    self._mockapp = self.module_path('mockapplications.py')
    self._module_with_noargparse = self.module_path('module_with_no_argparse.py')
    self._module_with_arparse_in_try = self.module_path('example_argparse_souce_in_try.py')
    self._module_with_argparse_in_main = self.module_path('example_argparse_souce_in_main.py')

  def test_should_throw_parser_exception_if_no_argparse_found_in_module(self):
    with self.assertRaises(source_parser.ParserError):
      source_parser.parse_source_file(self._module_with_noargparse)


  def test_find_main(self):
    example_source = '''
def main():  pass
    '''
    nodes = ast.parse(example_source)
    main_node = source_parser.find_main(nodes)
    self.assertEqual('main', main_node.name)


  def test_find_main_throws_exception_if_not_found(self):
    example_source = '''
def some_cool_function_that_is_not_main():  pass
    '''
    with self.assertRaises(source_parser.ParserError):
      nodes = ast.parse(example_source)
      main_node = source_parser.find_main(nodes)
      self.assertEqual('main', main_node.name)


  def test_find_try_blocks_finds_all_tryblock_styles(self):
    example_source = '''
try: a = 1
except: pass

try: pass
finally: pass

try: pass
except: pass
else: pass
    '''
    nodes = ast.parse(example_source)
    try_blocks = source_parser.find_try_blocks(nodes)
    self.assertEqual(3, len(try_blocks))


  def test_find_try_blocks_returns_empty_if_no_blocks_present(self):
    example_source = 'def main(): pass'
    nodes = ast.parse(example_source)
    result = source_parser.find_try_blocks(nodes)
    self.assertEqual(list(), result)

  def test_find_argparse_located_object_when_imported_by_direct_name(self):
    example_source = '''
def main():
  parser = ArgumentParser(description='Example Argparse Program', formatter_class=RawDescriptionHelpFormatter)
    '''
    nodes = ast.parse(example_source)
    main_node = source_parser.find_main(nodes)
    self.assertEqual('main', main_node.name)
    containing_block = source_parser.find_block_containing_argparse([main_node])
    self.assertTrue(containing_block is not None)

  def test_find_argparse_located_object_when_access_through_module_dot_notation(self):
    example_source = '''
def main():
  parser = argparse.ArgumentParser(description='Example Argparse Program', formatter_class=RawDescriptionHelpFormatter)
    '''
    nodes = ast.parse(example_source)
    main_node = source_parser.find_main(nodes)
    self.assertEqual('main', main_node.name)
    containing_block = source_parser.find_block_containing_argparse([main_node])
    self.assertTrue(containing_block is not None)

  def test_find_argparse_locates_assignment_stmnt_in_main(self):
    nodes = ast.parse(source_parser._openfile(self._module_with_argparse_in_main))
    main_node = source_parser.find_main(nodes)
    self.assertEqual('main', main_node.name)
    containing_block = source_parser.find_block_containing_argparse([main_node])
    self.assertTrue(containing_block is not None)
    self.assertEqual('main', containing_block.name)


  def test_find_argparse_locates_assignment_stmnt_in_try_block(self):
    nodes = ast.parse(source_parser._openfile(self._module_with_arparse_in_try))
    main_node = source_parser.find_main(nodes)
    self.assertEqual('main', main_node.name)
    try_nodes = source_parser.find_try_blocks(main_node)
    self.assertTrue(len(try_nodes) > 0)
    containing_block = source_parser.find_block_containing_argparse([main_node] + try_nodes)
    self.assertEqual(ast.TryExcept, type(containing_block))


  def test_find_argparse_throws_exception_if_not_found(self):
    with self.assertRaises(source_parser.ParserError):
      nodes = ast.parse(source_parser._openfile(self._module_with_noargparse))
      main_node = source_parser.find_main(nodes)
      self.assertEqual('main', main_node.name)
      try_nodes = source_parser.find_try_blocks(main_node)
      containing_block = source_parser.find_block_containing_argparse([main_node] + try_nodes)


  def test_has_instantiator_returns_true_if_object_found(self):
    source = '''
parser = ArgumentParser(description='Example Argparse Program', formatter_class=RawDescriptionHelpFormatter)
parser.add_argument("filename", help="filename")
    '''
    nodes = ast.parse(source)
    self.assertTrue(source_parser.has_instantiator(nodes.body[0], 'ArgumentParser'))


  def test_has_instantiator_returns_false_if_object_not_found(self):
    source = '''
parser = NopeParser(description='Example Argparse Program', formatter_class=RawDescriptionHelpFormatter)
parser.add_argument("filename", help="filename")
    '''
    nodes = ast.parse(source)
    self.assertFalse(source_parser.has_instantiator(nodes.body[0], 'ArgumentParser'))

  def test_has_assignment_returns_true_if_object_found(self):
    source = '''
parser = ArgumentParser(description='Example Argparse Program', formatter_class=RawDescriptionHelpFormatter)
parser.add_argument("filename", help="filename")
    '''
    nodes = ast.parse(source)
    self.assertTrue(source_parser.has_assignment(nodes.body[1], 'add_argument'))

  def test_has_assignment_returns_false_if_object_not_found(self):
    source = '''
parser = ArgumentParser(description='Example Argparse Program', formatter_class=RawDescriptionHelpFormatter)
parser.add_argument("filename", help="filename")
    '''
    nodes = ast.parse(source)
    self.assertFalse(source_parser.has_instantiator(nodes.body[1], 'add_argument'))

  def test_parser_identifies_import_module(self):
    source = '''
import os
import itertools
from os import path
    '''
    import _ast
    nodes = ast.parse(source)
    module_imports = source_parser.get_nodes_by_instance_type(nodes, _ast.Import)
    self.assertEqual(2, len(module_imports))

  def test_parser_identifies_import_from(self):
    source = '''
import os
import itertools
from os import path
from gooey.gooey_decorator import Gooey
    '''
    import _ast
    nodes = ast.parse(source)
    from_imports = source_parser.get_nodes_by_instance_type(nodes, _ast.ImportFrom)
    self.assertEqual(2, len(from_imports))


  def test_get_indent_return_indent_amount_for_tabs_and_spaces(self):
    spaced_lines = ["def main"," def main","  def main","    def main"]
    expected_indent = ["", " ", "  ", "    "]
    for line, expected in zip(spaced_lines, expected_indent):
      self.assertEqual(expected, source_parser.get_indent(line))

  # def test_parse_source_file__file_with_argparse_in_main__succesfully_finds_and_returns_ast_obejcts(self):
  #   ast_objects = source_parser.parse_source_file(self._module_with_argparse_in_main)
  #   for obj in ast_objects:
  #     self.assertTrue(type(obj) in (ast.Assign, ast.Expr))
  #
  # def test_parse_source_file__file_with_argparse_in_try_block__succesfully_finds_and_returns_ast_obejcts(self):
  #   ast_objects = source_parser.parse_source_file(self._module_with_arparse_in_try)
  #   for obj in ast_objects:
  #     self.assertTrue(type(obj) in (ast.Assign, ast.Expr))


if __name__ == "__main__":
  #import sys;sys.argv = ['', 'Test.testName']
  unittest.main()