class TestArgParsing(unittest.TestCase): """ Tests for parsing arguments for the `rollback_split_course` management command """ def setUp(self): self.command = Command() def test_no_args(self): errstring = "rollback_split_course requires at least one argument" with self.assertRaisesRegexp(CommandError, errstring): self.command.handle() def test_invalid_locator(self): errstring = "Invalid locator string !?!" with self.assertRaisesRegexp(CommandError, errstring): self.command.handle("!?!")