class TestCli(unittest.TestCase):
def setUp(self):
self.cli = Cli(IoMock())
def testCliPrintsBoard(self):
self.cli.prompt_board(BoardMock())
output = self.cli.io.output
self.assertIn('1', output)
def testCliAsksForMove(self):
expected = 'Please enter your move: '
self.cli.ask_move()
output = self.cli.io.output
self.assertEqual(expected, output)
def testCliPromptsGoodBye(self):
expected = 'Good bye!'
self.cli.prompt_good_bye()
output = self.cli.io.output
self.assertEqual(expected, output)
def testCliPromptsWelcome(self):
expected = 'Welcome!'
self.cli.prompt_welcome()
output = self.cli.io.output
self.assertEqual(expected, output)
def testReturnsUserInput(self):
expected = '5'
user_input = self.cli.ask_move()
self.assertEqual(expected, user_input)
def main():
colorama.init()
try:
cli = Cli()
if cli.flag_exists('version', 'v'):
version()
del cli
return
if cli.flag_exists('help', 'p'):
show_help()
del cli
return
command = cli.get_arg(1, 'command',
['resize', 'trim', 'crop', 'convert'])
path = cli.get_arg(2, 'path')
imp = Imp(path)
# Flags.
w = cli.get_flag_value('width', 'w')
h = cli.get_flag_value('height', 'h')
t = cli.get_flag_value('top', 't')
l = cli.get_flag_value('left', 'l')
e = cli.get_flag_value('ext', 'e')
o = cli.get_flag_value('out', 'o')
if o != None:
imp.set_path(o)
if e != None:
imp.set_extension(e)
if command == 'resize':
imp.resize(w, h)
imp.save()
log(f'Image resized with success in "{imp.get_path()}"!')
elif command == 'crop':
imp.crop(l, t, w, h)
imp.save()
log(f'Image cropped with success in "{imp.get_path()}"!')
elif command == 'trim':
imp.trim()
imp.save()
log(f'Image trimmed with success in "{imp.get_path()}"!')
elif command == 'convert':
imp.convert(e)
imp.save()
log(f'Image converted with success in "{imp.get_path()}"!')
if cli in locals():
del cli
if imp in locals():
del imp
except Exception as err:
show_help()
error(err)
def test_something(self):
cli = Cli()
def setUp(self): self.cli = Cli(IoMock())
from src.cli import Cli cli = Cli() cli.test_movement()
#! /usr/bin/env python3
from src.cli import Cli
if __name__ == '__main__':
cli = Cli()
cli.run()
