def test_do_command_no_args(self): """ Test executing a command without arguments. """ command = "ls" args = None method_args = [command] response = "foobar" self.mox.StubOutWithMock(RestrictedCommandHandler, "make_syscall") RestrictedCommandHandler.make_syscall(method_args).AndReturn(response) self.mox.ReplayAll() restricted_handler = RestrictedCommandHandler() self.assertEquals(response, restricted_handler.do_command(command, args))
def test_do_command_popen_raises(self): """ Ensure proper behavior when Popen raises an assertion upon command execution. """ command = "ls" args = "-al" method_args = [command] method_args.extend(args) self.mox.StubOutWithMock(RestrictedCommandHandler, "make_syscall") RestrictedCommandHandler.make_syscall(method_args).AndRaise( OSError(2, None, 'File not found')) self.mox.ReplayAll() restricted_handler = RestrictedCommandHandler() self.assertEqual(type(""), type(restricted_handler.do_command(command, args)))