class SawsTest(unittest.TestCase):
@mock.patch('saws.resources.print')
def setUp(self, mock_print):
self.file_name = os.path.expanduser('~') + '/' + '.saws.log'
self.saws = Saws()
mock_print.assert_called_with('Loaded resources from cache')
self.DOCS_HOME_URL = 'http://docs.aws.amazon.com/cli/latest/reference/index.html'
@mock.patch('saws.saws.click')
def test_log_exception(self, mock_click):
exception_message = 'test_log_exception'
e = Exception(exception_message)
try:
raise e
except Exception:
# Traceback needs to have an active exception as described in:
# http://bugs.python.org/issue23003
self.saws.log_exception(e, traceback, echo=True)
mock_click.secho.assert_called_with(str(e), fg='red')
assert os.path.isfile(self.file_name)
with open(self.file_name, 'r') as fp:
for line in fp:
pass
assert exception_message in line
def test_set_get_color(self):
self.saws.set_color(True)
assert self.saws.get_color()
self.saws.set_color(False)
assert not self.saws.get_color()
def test_get_set_fuzzy_match(self):
self.saws.set_fuzzy_match(True)
assert self.saws.get_fuzzy_match()
self.saws.set_fuzzy_match(False)
assert not self.saws.get_fuzzy_match()
def test_get_set_shortcut_match(self):
self.saws.set_shortcut_match(True)
assert self.saws.get_shortcut_match()
self.saws.set_shortcut_match(False)
assert not self.saws.get_shortcut_match()
@mock.patch('saws.saws.webbrowser')
def test_handle_docs(self, mock_webbrowser):
EC2_URL = 'http://docs.aws.amazon.com/cli/latest/reference/ec2/index.html'
EC2_DESC_INSTANCES_URL = 'http://docs.aws.amazon.com/cli/latest/reference/ec2/describe-instances.html'
assert not self.saws.handle_docs('')
assert not self.saws.handle_docs('foo bar')
assert self.saws.handle_docs('', from_fkey=True)
mock_webbrowser.open.assert_called_with(self.DOCS_HOME_URL)
assert self.saws.handle_docs('baz', from_fkey=True)
mock_webbrowser.open.assert_called_with(self.DOCS_HOME_URL)
assert self.saws.handle_docs('aws ec2', from_fkey=True)
mock_webbrowser.open.assert_called_with(EC2_URL)
assert self.saws.handle_docs('aws ec2 docs', from_fkey=False)
mock_webbrowser.open.assert_called_with(EC2_URL)
assert self.saws.handle_docs('aws ec2 describe-instances',
from_fkey=True)
mock_webbrowser.open.assert_called_with(EC2_DESC_INSTANCES_URL)
assert self.saws.handle_docs('aws ec2 describe-instances docs',
from_fkey=False)
mock_webbrowser.open.assert_called_with(EC2_DESC_INSTANCES_URL)
@mock.patch('saws.saws.os')
def test_handle_cd(self, mock_os):
assert not self.saws.handle_cd('aws')
assert self.saws.handle_cd('cd ')
assert self.saws.handle_cd('cd foo')
mock_os.chdir.assert_called_with('foo')
def test_colorize_output(self):
self.saws.set_color(False)
assert self.saws.colorize_output(AwsCommands.AWS_COMMAND) == \
AwsCommands.AWS_COMMAND
self.saws.set_color(True)
assert self.saws.colorize_output(AwsCommands.AWS_CONFIGURE) == \
AwsCommands.AWS_CONFIGURE
assert self.saws.colorize_output(AwsCommands.AWS_HELP) == \
AwsCommands.AWS_HELP
EC2_LS_CMD = 'aws ec2 ls'
assert self.saws.colorize_output(EC2_LS_CMD) == \
EC2_LS_CMD + self.saws.PYGMENTS_CMD
@mock.patch('saws.saws.subprocess')
@mock.patch('saws.saws.webbrowser')
def test_process_command_docs(self, mock_webbrowser, mock_subprocess):
self.saws.process_command('aws docs')
mock_webbrowser.open.assert_called_with(self.DOCS_HOME_URL)
mock_subprocess.call.assert_not_called()
@mock.patch('saws.saws.subprocess')
def test_process_command_cd(self, mock_subprocess):
self.saws.process_command('cd .')
mock_subprocess.call.assert_not_called()
@mock.patch('saws.saws.subprocess')
def test_process_command(self, mock_subprocess):
self.saws.set_color(False)
INVAL_CMD = 'foo'
self.saws.process_command(INVAL_CMD)
mock_subprocess.call.assert_called_with(INVAL_CMD,
shell=True)
self.saws.process_command(AwsCommands.AWS_COMMAND)
mock_subprocess.call.assert_called_with(AwsCommands.AWS_COMMAND,
shell=True)
self.saws.set_color(True)
colorized_command = AwsCommands.AWS_COMMAND + self.saws.PYGMENTS_CMD
self.saws.process_command(AwsCommands.AWS_COMMAND)
mock_subprocess.call.assert_called_with(colorized_command,
shell=True)
class SawsTest(unittest.TestCase):
def setUp(self):
self.file_name = os.path.expanduser('~') + '/' + '.saws.log'
self.saws = Saws(refresh_resources=False)
self.DOCS_HOME_URL = \
'http://docs.aws.amazon.com/cli/latest/reference/index.html'
@mock.patch('saws.saws.click')
def test_log_exception(self, mock_click):
exception_message = 'test_log_exception'
e = Exception(exception_message)
try:
raise e
except Exception:
# Traceback needs to have an active exception as described in:
# http://bugs.python.org/issue23003
self.saws.log_exception(e, traceback, echo=True)
mock_click.secho.assert_called_with(str(e), fg='red')
assert os.path.isfile(self.file_name)
with open(self.file_name, 'r') as fp:
for line in fp:
pass
assert exception_message in line
def test_set_get_color(self):
self.saws.set_color(True)
assert self.saws.get_color()
self.saws.set_color(False)
assert not self.saws.get_color()
def test_get_set_fuzzy_match(self):
self.saws.set_fuzzy_match(True)
assert self.saws.get_fuzzy_match()
self.saws.set_fuzzy_match(False)
assert not self.saws.get_fuzzy_match()
def test_get_set_shortcut_match(self):
self.saws.set_shortcut_match(True)
assert self.saws.get_shortcut_match()
self.saws.set_shortcut_match(False)
assert not self.saws.get_shortcut_match()
@mock.patch('saws.saws.webbrowser')
def test_handle_docs(self, mock_webbrowser):
EC2_URL = \
'http://docs.aws.amazon.com/cli/latest/reference/ec2/index.html'
EC2_DESC_INSTANCES_URL = \
'http://docs.aws.amazon.com/cli/latest/reference/ec2/describe-instances.html' # NOQA
assert not self.saws.handle_docs('')
assert not self.saws.handle_docs('foo bar')
assert self.saws.handle_docs('',
from_fkey=True)
mock_webbrowser.open.assert_called_with(self.DOCS_HOME_URL)
assert self.saws.handle_docs('baz',
from_fkey=True)
mock_webbrowser.open.assert_called_with(self.DOCS_HOME_URL)
assert self.saws.handle_docs('aws ec2',
from_fkey=True)
mock_webbrowser.open.assert_called_with(EC2_URL)
assert self.saws.handle_docs('aws ec2 docs',
from_fkey=False)
mock_webbrowser.open.assert_called_with(EC2_URL)
assert self.saws.handle_docs('aws ec2 describe-instances',
from_fkey=True)
mock_webbrowser.open.assert_called_with(EC2_DESC_INSTANCES_URL)
assert self.saws.handle_docs('aws ec2 describe-instances docs',
from_fkey=False)
mock_webbrowser.open.assert_called_with(EC2_DESC_INSTANCES_URL)
@mock.patch('saws.saws.os')
def test_handle_cd(self, mock_os):
assert not self.saws._handle_cd('aws')
assert self.saws._handle_cd('cd ')
assert self.saws._handle_cd('cd foo')
mock_os.chdir.assert_called_with('foo')
def test_colorize_output(self):
self.saws.set_color(False)
assert self.saws._colorize_output(AwsCommands.AWS_COMMAND) == \
AwsCommands.AWS_COMMAND
self.saws.set_color(True)
assert self.saws._colorize_output(AwsCommands.AWS_CONFIGURE) == \
AwsCommands.AWS_CONFIGURE
assert self.saws._colorize_output(AwsCommands.AWS_HELP) == \
AwsCommands.AWS_HELP
EC2_LS_CMD = 'aws ec2 ls'
assert self.saws._colorize_output(EC2_LS_CMD) == \
EC2_LS_CMD + self.saws.PYGMENTS_CMD
@mock.patch('saws.saws.subprocess')
@mock.patch('saws.saws.webbrowser')
def test_process_command_docs(self, mock_webbrowser, mock_subprocess):
self.saws._process_command('aws docs')
mock_webbrowser.open.assert_called_with(self.DOCS_HOME_URL)
mock_subprocess.call.assert_not_called()
@mock.patch('saws.saws.subprocess')
def test_process_command_cd(self, mock_subprocess):
self.saws._process_command('cd .')
mock_subprocess.call.assert_not_called()
@mock.patch('saws.saws.subprocess')
def test_process_command(self, mock_subprocess):
self.saws.set_color(False)
INVAL_CMD = 'foo'
self.saws._process_command(INVAL_CMD)
mock_subprocess.call.assert_called_with(INVAL_CMD,
shell=True)
self.saws._process_command(AwsCommands.AWS_COMMAND)
mock_subprocess.call.assert_called_with(AwsCommands.AWS_COMMAND,
shell=True)
self.saws.set_color(True)
colorized_command = AwsCommands.AWS_COMMAND + self.saws.PYGMENTS_CMD
self.saws._process_command(AwsCommands.AWS_COMMAND)
mock_subprocess.call.assert_called_with(colorized_command,
shell=True)
def test_handle_keyboard_interrupt(self):
e = KeyboardInterrupt('')
# TODO: Mock calls to renderer.clear and input_processor.feed
self.saws._handle_keyboard_interrupt(e, platform='Darwin')
with self.assertRaises(KeyboardInterrupt):
self.saws._handle_keyboard_interrupt(e, platform='Windows')
