#!/usr/bin/env python from __future__ import print_function import unittest from ruffus.cmdline import handle_verbose import ruffus """ test_cmdline.py """ import os import sys # add grandparent to search path for testing grandparent_dir = os.path.abspath( os.path.join(os.path.dirname(__file__), "..", "..")) sys.path.insert(0, grandparent_dir) # module name = script name without extension module_name = os.path.splitext(os.path.basename(__file__))[0] # funky code to import by file name parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "..")) cmdline = ruffus.cmdline # mock for command line options class t_options(object): def __str__(self): return self.__dict__ def __repr__(self): return str(self.__dict__) class Test_cmdline(unittest.TestCase): # def setUp(self): # if sys.hexversion < 0x03000000: # self.assertRaisesRegex = self.assertRaisesRegexp # ___________________________________________________________________________ # # test_something() # ___________________________________________________________________________ # def test_something(self): # s = StringIO() # cleanup_tmpdir() # pipeline_printout(s, [test_regex_task], verbose=5, wrap_width = 10000, pipeline= "main") # self.assertTrue(re.search('Missing files.*\[tmp_test_regex_error_messages/a_name.tmp1, tmp_test_regex_error_messages/a_name.tmp2', s.getvalue(), re.DOTALL)) # self.assertIn("Warning: File match failure: File 'tmp_test_regex_error_messages/a_name.tmp1' does not match regex", s.getvalue()) # self.assertRaisesRegex(fatal_error_input_file_does_not_match, # "File '.*?' does not match regex\('.*?'\) and pattern '.*?':\n.*unknown group name", # pipeline_printout, # s, [test_regex_misspelt_capture_error_task], # verbose = 3) # ___________________________________________________________________________ # # cleanup # ___________________________________________________________________________ # def tearDown(self): # pass # shutil.rmtree(workdir) # # # _____________________________________________________________________________________ # # test_verbose # _____________________________________________________________________________________ def test_verbose(self): """ --verbose on its own increases the verbosity by one --verbose NNN (re)sets the verbosity to NNN whatever the previous state """ # options.verbose defined by user to be None options = t_options() setattr(options, "verbose", None) handle_verbose(options) self.assertTrue(options.verbose == None) self.assertTrue(options.verbose_abbreviated_path is None) # options.verbose defined by user to be 0 options = t_options() setattr(options, "verbose", 0) handle_verbose(options) self.assertTrue(options.verbose == 0) self.assertTrue(options.verbose_abbreviated_path is None) # options.verbose defined by user to be "6" options = t_options() setattr(options, "verbose", "6") handle_verbose(options) self.assertTrue(options.verbose == 6) self.assertTrue(options.verbose_abbreviated_path == None) # options.verbose defined by user to be 6 options = t_options() setattr(options, "verbose", 6) handle_verbose(options) self.assertTrue(options.verbose == 6) self.assertTrue(options.verbose_abbreviated_path == None) # options.verbose defined by user to "+" options = t_options() setattr(options, "verbose", "+") handle_verbose(options) self.assertTrue(options.verbose == 1) self.assertTrue(options.verbose_abbreviated_path == None) # --verbose not set options = t_options() setattr(options, "verbose", []) handle_verbose(options) self.assertTrue(options.verbose == 0) self.assertTrue(options.verbose_abbreviated_path == None) # --verbose options = t_options() setattr(options, "verbose", ["+"]) handle_verbose(options) self.assertTrue(options.verbose == 1) self.assertTrue(options.verbose_abbreviated_path == None) # --verbose --verbose 5 --verbose options = t_options() setattr(options, "verbose", ["+", "5", "+"]) handle_verbose(options) self.assertTrue(options.verbose == 6) self.assertTrue(options.verbose_abbreviated_path == None) # --verbose --verbose 5 --verbose --verbose 4 # last value overrides the 5 options = t_options() setattr(options, "verbose", ["+", "5", "+", "4"]) handle_verbose(options) self.assertTrue(options.verbose == 4) self.assertTrue(options.verbose_abbreviated_path == None) # _____________________________________________________________________________________ # # test_verbose_abbreviated_path # _____________________________________________________________________________________ def test_verbose_abbreviated_path(self): """ --verbose NNN:MMM sets the verbose_abbreviated_path to MMM """ # # do not override users' verbose_abbreviated_path # options = t_options() # take verbose_abbreviated_path setattr(options, "verbose", ["+", "5", "+", "4:3"]) handle_verbose(options) self.assertTrue(options.verbose == 4) self.assertTrue(options.verbose_abbreviated_path == 3) # do not override users' verbose_abbreviated_path setattr(options, "verbose", ["+", "5", "+", "7:5"]) handle_verbose(options) self.assertTrue(options.verbose == 7) self.assertTrue(options.verbose_abbreviated_path == 3) options = t_options() # take verbose_abbreviated_path setattr(options, "verbose", ["+", "5:3", "+", "7:5", "+"]) handle_verbose(options) self.assertTrue(options.verbose == 8) self.assertTrue(options.verbose_abbreviated_path == 5) # _____________________________________________________________________________________ # # test_argparse # _____________________________________________________________________________________ def test_argparse(self): """ Same as above but setting up options using ruffus.cmdline.get_argparse --verbose on its own increases the verbosity by one --verbose NNN (re)sets the verbosity to NNN whatever the previous state --verbose NNN:MMM sets the verbose_abbreviated_path to MMM """ parser = cmdline.get_argparse( description='WHAT DOES THIS PIPELINE DO?') import sys sys.argv = ["test", "--verbose", "--verbose=2"] options = parser.parse_args() handle_verbose(options) self.assertTrue(options.verbose == 2) self.assertTrue(options.verbose_abbreviated_path == None) sys.argv = ["test", "--verbose", "--verbose=3", "--verbose"] options = parser.parse_args() handle_verbose(options) self.assertTrue(options.verbose == 4) self.assertTrue(options.verbose_abbreviated_path == None) sys.argv = ["test", "--verbose", "--verbose=5:3", "--verbose"] options = parser.parse_args() handle_verbose(options) self.assertTrue(options.verbose == 6) self.assertTrue(options.verbose_abbreviated_path == 3) sys.argv = ["test", "--verbose", "--verbose=5:3", "--verbose", "--verbose=7", "--verbose"] options = parser.parse_args() handle_verbose(options) self.assertTrue(options.verbose == 8) self.assertTrue(options.verbose_abbreviated_path == 3) sys.argv = ["test", "--verbose", "--verbose=5:3", "--verbose", "--verbose=7:5", "--verbose"] options = parser.parse_args() handle_verbose(options) self.assertTrue(options.verbose == 8) self.assertTrue(options.verbose_abbreviated_path == 5) # _____________________________________________________________________________________ # # test_optparse # _____________________________________________________________________________________ def test_optparse(self): """ Same as above but setting up options using ruffus.cmdline.get_opt-parse --verbose on its own increases the verbosity by one --verbose NNN (re)sets the verbosity to NNN whatever the previous state --verbose NNN:MMM sets the verbose_abbreviated_path to MMM """ parser = cmdline.get_optparse(usage='WHAT DOES THIS PIPELINE DO?') sys.argv = ["test", "--verbose", "--verbose=2"] (options, remaining_args) = parser.parse_args() handle_verbose(options) self.assertTrue(options.verbose == 2) self.assertTrue(options.verbose_abbreviated_path == None) sys.argv = ["test", "--verbose", "--verbose=3", "--verbose"] (options, remaining_args) = parser.parse_args() handle_verbose(options) self.assertTrue(options.verbose == 4) self.assertTrue(options.verbose_abbreviated_path == None) sys.argv = ["test", "--verbose", "--verbose=5:3", "--verbose"] (options, remaining_args) = parser.parse_args() handle_verbose(options) self.assertTrue(options.verbose == 6) self.assertTrue(options.verbose_abbreviated_path == 3) sys.argv = ["test", "--verbose", "--verbose=5:3", "--verbose", "--verbose=7", "--verbose"] (options, remaining_args) = parser.parse_args() handle_verbose(options) self.assertTrue(options.verbose == 8) self.assertTrue(options.verbose_abbreviated_path == 3) sys.argv = ["test", "--verbose", "--verbose=5:3", "--verbose", "--verbose=7:5", "--verbose"] (options, remaining_args) = parser.parse_args() handle_verbose(options) self.assertTrue(options.verbose == 8) self.assertTrue(options.verbose_abbreviated_path == 5) # # Necessary to protect the "entry point" of the program under windows. # see: http://docs.python.org/library/multiprocessing.html#multiprocessing-programming # if __name__ == '__main__': #pipeline_printout(sys.stdout, [test_product_task], verbose = 3, pipeline= "main") unittest.main()