# test_getopt.py

# David Goodger <dgoodger@bigfoot.com> 2000-08-19

from test.test_support import verbose, run_doctest, run_unittest

import unittest

import getopt

import os

sentinel = object()

class GetoptTests(unittest.TestCase):

    def setUp(self):

        self.old_posixly_correct = os.environ.get("POSIXLY_CORRECT", sentinel)

        if self.old_posixly_correct is not sentinel:

            del os.environ["POSIXLY_CORRECT"]

    def tearDown(self):

        if self.old_posixly_correct is sentinel:

            os.environ.pop("POSIXLY_CORRECT", None)

        else:

            os.environ["POSIXLY_CORRECT"] = self.old_posixly_correct

    def assertError(self, *args, **kwargs):

        self.assertRaises(getopt.GetoptError, *args, **kwargs)

    def test_short_has_arg(self):

        self.failUnless(getopt.short_has_arg('a', 'a:'))

        self.failIf(getopt.short_has_arg('a', 'a'))

        self.assertError(getopt.short_has_arg, 'a', 'b')

    def test_long_has_args(self):

        has_arg, option = getopt.long_has_args('abc', ['abc='])

        self.failUnless(has_arg)

        self.assertEqual(option, 'abc')

        has_arg, option = getopt.long_has_args('abc', ['abc'])

        self.failIf(has_arg)

        self.assertEqual(option, 'abc')

        has_arg, option = getopt.long_has_args('abc', ['abcd'])

        self.failIf(has_arg)

        self.assertEqual(option, 'abcd')

        self.assertError(getopt.long_has_args, 'abc', ['def'])

        self.assertError(getopt.long_has_args, 'abc', [])

        self.assertError(getopt.long_has_args, 'abc', ['abcd','abcde'])

    def test_do_shorts(self):

        opts, args = getopt.do_shorts([], 'a', 'a', [])

        self.assertEqual(opts, [('-a', '')])

        self.assertEqual(args, [])

        opts, args = getopt.do_shorts([], 'a1', 'a:', [])

        self.assertEqual(opts, [('-a', '1')])

        self.assertEqual(args, [])

        #opts, args = getopt.do_shorts([], 'a=1', 'a:', [])

        #self.assertEqual(opts, [('-a', '1')])

        #self.assertEqual(args, [])

        opts, args = getopt.do_shorts([], 'a', 'a:', ['1'])

        self.assertEqual(opts, [('-a', '1')])

        self.assertEqual(args, [])

        opts, args = getopt.do_shorts([], 'a', 'a:', ['1', '2'])

        self.assertEqual(opts, [('-a', '1')])

        self.assertEqual(args, ['2'])

        self.assertError(getopt.do_shorts, [], 'a1', 'a', [])

        self.assertError(getopt.do_shorts, [], 'a', 'a:', [])

    def test_do_longs(self):

        opts, args = getopt.do_longs([], 'abc', ['abc'], [])

        self.assertEqual(opts, [('--abc', '')])

        self.assertEqual(args, [])

        opts, args = getopt.do_longs([], 'abc=1', ['abc='], [])

        self.assertEqual(opts, [('--abc', '1')])

        self.assertEqual(args, [])

        opts, args = getopt.do_longs([], 'abc=1', ['abcd='], [])

        self.assertEqual(opts, [('--abcd', '1')])

        self.assertEqual(args, [])

        opts, args = getopt.do_longs([], 'abc', ['ab', 'abc', 'abcd'], [])

        self.assertEqual(opts, [('--abc', '')])

        self.assertEqual(args, [])

        # Much like the preceding, except with a non-alpha character ("-") in

        # option name that precedes "="; failed in

        # http://python.org/sf/126863

        opts, args = getopt.do_longs([], 'foo=42', ['foo-bar', 'foo=',], [])

        self.assertEqual(opts, [('--foo', '42')])

        self.assertEqual(args, [])

        self.assertError(getopt.do_longs, [], 'abc=1', ['abc'], [])

        self.assertError(getopt.do_longs, [], 'abc', ['abc='], [])

    def test_getopt(self):

        # note: the empty string between '-a' and '--beta' is significant:

        # it simulates an empty string option argument ('-a ""') on the

        # command line.

        cmdline = ['-a', '1', '-b', '--alpha=2', '--beta', '-a', '3', '-a',

                   '', '--beta', 'arg1', 'arg2']

        opts, args = getopt.getopt(cmdline, 'a:b', ['alpha=', 'beta'])

        self.assertEqual(opts, [('-a', '1'), ('-b', ''),

                                ('--alpha', '2'), ('--beta', ''),

                                ('-a', '3'), ('-a', ''), ('--beta', '')])

        # Note ambiguity of ('-b', '') and ('-a', '') above. This must be

        # accounted for in the code that calls getopt().

        self.assertEqual(args, ['arg1', 'arg2'])

        self.assertError(getopt.getopt, cmdline, 'a:b', ['alpha', 'beta'])

    def test_gnu_getopt(self):

        # Test handling of GNU style scanning mode.

        cmdline = ['-a', 'arg1', '-b', '1', '--alpha', '--beta=2']

        # GNU style

        opts, args = getopt.gnu_getopt(cmdline, 'ab:', ['alpha', 'beta='])

        self.assertEqual(args, ['arg1'])

        self.assertEqual(opts, [('-a', ''), ('-b', '1'),

                                ('--alpha', ''), ('--beta', '2')])

        # Posix style via +

        opts, args = getopt.gnu_getopt(cmdline, '+ab:', ['alpha', 'beta='])

        self.assertEqual(opts, [('-a', '')])

        self.assertEqual(args, ['arg1', '-b', '1', '--alpha', '--beta=2'])

        # Posix style via POSIXLY_CORRECT

        os.environ["POSIXLY_CORRECT"] = "1"

        opts, args = getopt.gnu_getopt(cmdline, 'ab:', ['alpha', 'beta='])

        self.assertEqual(opts, [('-a', '')])

        self.assertEqual(args, ['arg1', '-b', '1', '--alpha', '--beta=2'])

    def test_libref_examples(self):

        s = """

        Examples from the Library Reference:  Doc/lib/libgetopt.tex

        An example using only Unix style options:

        >>> import getopt

        >>> args = '-a -b -cfoo -d bar a1 a2'.split()

        >>> args

        ['-a', '-b', '-cfoo', '-d', 'bar', 'a1', 'a2']

        >>> optlist, args = getopt.getopt(args, 'abc:d:')

        >>> optlist

        [('-a', ''), ('-b', ''), ('-c', 'foo'), ('-d', 'bar')]

        >>> args

        ['a1', 'a2']

        Using long option names is equally easy:

        >>> s = '--condition=foo --testing --output-file abc.def -x a1 a2'

        >>> args = s.split()

        >>> args

        ['--condition=foo', '--testing', '--output-file', 'abc.def', '-x', 'a1', 'a2']

        >>> optlist, args = getopt.getopt(args, 'x', [

        ...     'condition=', 'output-file=', 'testing'])

        >>> optlist

        [('--condition', 'foo'), ('--testing', ''), ('--output-file', 'abc.def'), ('-x', '')]

        >>> args

        ['a1', 'a2']

        """

        import types

        m = types.ModuleType("libreftest", s)

        run_doctest(m, verbose)

def test_main():

    run_unittest(GetoptTests)

if __name__ == "__main__":

    test_main()