예제 #1
0
 def validate_options(self):
     origin = options.allow_origin
     if origin is not None:
         try:
             re.compile(origin)
         except Exception:
             msg = 'Invalid `allow_origin` regex r"{}"'
             raise Command.UsageError(msg.format(origin))
예제 #2
0
    def test_run_raises_UsageError(self):
        cb = Mock()
        c = Command(on_usage_error=cb)
        c.verify_args = Mock()
        c.run = Mock()
        exc = c.run.side_effect = c.UsageError('foo', status=3)

        self.assertEqual(c(), exc.status)
        cb.assert_called_with(exc)
        c.verify_args.assert_called_with(())