"""Tests for the lll script in the Tools/script directory."""

import os

import tempfile

from test import support

from test.test_tools import skip_if_missing, import_tool

import unittest

skip_if_missing()

class lllTests(unittest.TestCase):

def setUp(self):

self.lll = import_tool('lll')

def test_lll_multiple_dirs(self):

with tempfile.TemporaryDirectory() as dir1, \

tempfile.TemporaryDirectory() as dir2:

fn1 = os.path.join(dir1, 'foo1')

fn2 = os.path.join(dir2, 'foo2')

for fn, dir in (fn1, dir1), (fn2, dir2):

open(fn, 'w').close()

os.symlink(fn, os.path.join(dir, 'symlink'))

with support.captured_stdout() as output:

self.lll.main([dir1, dir2])

self.assertEqual(output.getvalue(),

f'{dir1}:\n'

f'symlink -> {fn1}\n'

f'\n'

f'{dir2}:\n'

f'symlink -> {fn2}\n'

)

if __name__ == '__main__':

unittest.main()

Fix the argument handling in Tools/scripts/lll.py.

full = os.path.join(dirname, name)

if os.path.islink(full):

print(name, '->', os.readlink(full))

def main():

args = sys.argv[1:]

def main(args):

if not args: args = [os.curdir]

first = 1

for arg in args:

if len(args) > 1:

if not first: print()

first = 0

print(arg + ':')

lll(arg)

lll(arg)

if __name__ == '__main__':

main()

main(sys.argv[1:])