def test_cmd_status(self): argv = self._get_argv('status') main(argv=argv, out=self.out) output_lines = self.out.getvalue().splitlines() self.assertTrue(output_lines[0].startswith( 'Instance:i-1111 (name-i-1111) stopped')) self.assertTrue(output_lines[1].startswith( 'Instance:i-2222 (name-i-2222) stopped'))
def test_cmd_now(self): self.out = StringIO() argv = self._get_argv('now') main(argv=argv, out=self.out) output_lines = self.out.getvalue().splitlines() self.assertTrue(output_lines[0].startswith( 'Instance:i-1111 (name-i-1111) running')) self.assertTrue(output_lines[1].startswith( 'Instance:i-2222 (name-i-2222) running')) instance_ids = self.mock_method.call_args[0][0] self.mock_method.return_value = get_instances_mock(instance_ids) self.out = StringIO() argv = self._get_argv('now') main(argv=argv, out=self.out) output_lines = self.out.getvalue().splitlines() self.assertTrue(output_lines[0].startswith( 'Instance:i-2222 (name-i-2222) running')) self.assertTrue(output_lines[1].startswith( 'Instance:i-1111 (name-i-1111) running'))
def test_cmd_restart(self): argv = self._get_argv('restart') main(argv=argv, out=self.out) output_lines = self.out.getvalue().splitlines() self.assertTrue(output_lines[0].startswith('Instance:i-1111 running')) self.assertTrue(output_lines[1].startswith('Instance:i-2222 running'))