def test_parse_body_help_existing_command(self): """ Test the help function for an existing command. """ command = "foobar" args = "some command arguments" hlp = "command help" command_set = [(command, args, hlp)] self.mox.StubOutWithMock(RestrictedCommandHandler, "__init__") self.mox.StubOutWithMock(configuration.commands, "restricted_set") RestrictedCommandHandler.__init__() configuration.commands.restricted_set().AndReturn(command_set) configuration.commands.restricted_set().AndReturn(command_set) configuration.commands.restricted_set().AndReturn(command_set) configuration.commands.restricted_set().AndReturn(command_set) self.mox.ReplayAll() restricted_handler = RestrictedCommandHandler() self.assertEquals("foobar - command help", restricted_handler.parse_body("help foobar")) self.assertEquals("foobar - command help", restricted_handler.parse_body("help foobar")) self.assertEquals("foobar - command help", restricted_handler.parse_body("help foobar ")) self.assertEquals("foobar - command help", restricted_handler.parse_body("help foobar "))
def test_parse_body_empty_body(self): """ Ensure proper behavior on a None command. """ self.mox.StubOutWithMock(RestrictedCommandHandler, "__init__") RestrictedCommandHandler.__init__() self.mox.ReplayAll() restricted_handler = RestrictedCommandHandler() self.assertEquals(None, restricted_handler.parse_body(None))
def test_parse_body_disallowed_command(self): """ Test for a disallowed command with some arguments. """ command = "foobar" args = "some command arguments" hlp = "command help" command_set = [(command, args, hlp)] self.mox.StubOutWithMock(RestrictedCommandHandler, "__init__") self.mox.StubOutWithMock(configuration.commands, "restricted_set") RestrictedCommandHandler.__init__() configuration.commands.restricted_set().AndReturn(command_set) self.mox.ReplayAll() restricted_handler = RestrictedCommandHandler() self.assertEquals(None, restricted_handler.parse_body("disallowed command"))
def test_parse_body_command_in_set_args(self): """ Test for an allowed command with arguments. """ command = "foobar" args = "some command arguments" hlp = "command help" command_set = [(command, args, hlp)] response = "a response" self.mox.StubOutWithMock(RestrictedCommandHandler, "__init__") self.mox.StubOutWithMock(RestrictedCommandHandler, "do_command") self.mox.StubOutWithMock(configuration.commands, "restricted_set") RestrictedCommandHandler.__init__() configuration.commands.restricted_set().AndReturn(command_set) RestrictedCommandHandler.do_command(command, args).AndReturn(response) self.mox.ReplayAll() restricted_handler = RestrictedCommandHandler() self.assertEquals(response, restricted_handler.parse_body(command))