Example #1
0
 def maybe_detach(self, argv, dopts=None):
     dopts = ['-D', '--detach'] if not dopts else dopts
     if any(arg in argv for arg in dopts):
         argv = [v for v in argv if v not in dopts]
         # will never return
         detached_celeryd(self.app).execute_from_commandline(argv)
         raise SystemExit(0)
Example #2
0
 def maybe_detach(self, argv, dopts=['-D', '--detach']):
     argv = list(sys.argv) if argv is None else argv
     if any(arg in argv for arg in dopts):
         argv = [arg for arg in argv if arg not in dopts]
         # never returns
         detached_celeryd().execute_from_commandline(argv)
         raise SystemExit(0)
Example #3
0
 def maybe_detach(self, argv, dopts=['-D', '--detach']):
     argv = list(sys.argv) if argv is None else argv
     if any(arg in argv for arg in dopts):
         argv = [arg for arg in argv if arg not in dopts]
         # never returns
         detached_celeryd().execute_from_commandline(argv)
         raise SystemExit(0)
Example #4
0
    def test_parser(self):
        x = detached_celeryd(self.app)
        p = x.Parser('celeryd_detach')
        options, values = p.parse_args([
            '--logfile=foo', '--fake', '--enable',
            'a', 'b', '-c1', '-d', '2',
        ])
        self.assertEqual(options.logfile, 'foo')
        self.assertEqual(values, ['a', 'b'])
        self.assertEqual(p.leftovers, ['--enable', '-c1', '-d', '2'])
        options, values = p.parse_args([
            '--fake', '--enable',
            '--pidfile=/var/pid/foo.pid',
            'a', 'b', '-c1', '-d', '2',
        ])
        self.assertEqual(options.pidfile, '/var/pid/foo.pid')

        with override_stdouts():
            with self.assertRaises(SystemExit):
                p.parse_args(['--logfile'])
            p.get_option('--logfile').nargs = 2
            with self.assertRaises(SystemExit):
                p.parse_args(['--logfile=a'])
            with self.assertRaises(SystemExit):
                p.parse_args(['--fake=abc'])

        assert p.get_option('--logfile').nargs == 2
        p.parse_args(['--logfile=a', 'b'])
        p.get_option('--logfile').nargs = 1
    def test_parser(self):
        x = detached_celeryd(self.app)
        p = x.create_parser('celeryd_detach')
        options, leftovers = p.parse_known_args([
            '--logfile=foo', '--fake', '--enable',
            'a', 'b', '-c1', '-d', '2',
        ])
        assert options.logfile == 'foo'
        assert leftovers, ['--enable', '-c1', '-d' == '2']
        options, leftovers = p.parse_known_args([
            '--fake', '--enable',
            '--pidfile=/var/pid/foo.pid',
            'a', 'b', '-c1', '-d', '2',
        ])
        assert options.pidfile == '/var/pid/foo.pid'

        with mock.stdouts():
            with pytest.raises(SystemExit):
                p.parse_args(['--logfile'])
            p._option_string_actions['--logfile'].nargs = 2
            with pytest.raises(SystemExit):
                p.parse_args(['--logfile=a'])
            with pytest.raises(SystemExit):
                p.parse_args(['--fake=abc'])

        assert p._option_string_actions['--logfile'].nargs == 2
        p.parse_args(['--logfile', 'a', 'b'])
Example #6
0
    def test_parser(self):
        x = detached_celeryd(self.app)
        p = x.create_parser('celeryd_detach')
        options, leftovers = p.parse_known_args([
            '--logfile=foo', '--fake', '--enable',
            'a', 'b', '-c1', '-d', '2',
        ])
        assert options.logfile == 'foo'
        assert leftovers, ['--enable', '-c1', '-d' == '2']
        options, leftovers = p.parse_known_args([
            '--fake', '--enable',
            '--pidfile=/var/pid/foo.pid',
            'a', 'b', '-c1', '-d', '2',
        ])
        assert options.pidfile == '/var/pid/foo.pid'

        with mock.stdouts():
            with pytest.raises(SystemExit):
                p.parse_args(['--logfile'])
            p._option_string_actions['--logfile'].nargs = 2
            with pytest.raises(SystemExit):
                p.parse_args(['--logfile=a'])
            with pytest.raises(SystemExit):
                p.parse_args(['--fake=abc'])

        assert p._option_string_actions['--logfile'].nargs == 2
        p.parse_args(['--logfile', 'a', 'b'])
Example #7
0
 def test_parse_options(self):
     x = detached_celeryd(app=self.app)
     o, v, l = x.parse_options('cd', self.argv)
     self.assertEqual(o.logfile, '/var/log')
     self.assertEqual(l, ['--autoscale=10,2', '-c', '1',
                          '-lDEBUG', '--logfile=/var/log',
                          '--pidfile=celeryd.pid'])
     x.parse_options('cd', [])  # no args
 def test_parse_options(self):
     x = detached_celeryd()
     o, v, l = x.parse_options('cd', self.argv)
     self.assertEqual(o.logfile, '/var/log')
     self.assertEqual(l, ['--autoscale=10,2', '-c', '1',
                          '-lDEBUG', '--logfile=/var/log',
                          '--pidfile=celeryd.pid'])
     x.parse_options('cd', [])  # no args
Example #9
0
 def test_parse_options(self):
     x = detached_celeryd()
     o, v, l = x.parse_options("cd", self.argv)
     self.assertEqual(o.logfile, "/var/log")
     self.assertEqual(l, ["--autoscale=10,2", "-c", "1",
                          "-lDEBUG", "--logfile=/var/log",
                          "--pidfile=celeryd.pid"])
     x.parse_options("cd", [])  # no args
Example #10
0
 def test_execute_from_commandline(self, detach, exit):
     x = detached_celeryd()
     x.execute_from_commandline(self.argv)
     self.assertTrue(exit.called)
     detach.assert_called_with(path=x.execv_path, uid=None, gid=None,
         umask=0, fake=False, logfile='/var/log', pidfile='celeryd.pid',
         argv=['-m', 'celery', 'worker', '-c', '1', '-lDEBUG',
               '--logfile=/var/log', '--pidfile=celeryd.pid',
               '--', '.disable_rate_limits=1'],
     )
Example #11
0
 def test_parse_options(self):
     x = detached_celeryd(app=self.app)
     o, v, l = x.parse_options('cd', self.argv)
     assert o.logfile == '/var/log'
     assert l == [
         '--foobar=10,2', '-c', '1',
         '-lDEBUG', '--logfile=/var/log',
         '--pidfile=celeryd.pid',
     ]
     x.parse_options('cd', [])  # no args
Example #12
0
 def test_parse_options(self):
     x = detached_celeryd(app=self.app)
     _, argv = x._split_command_line_config(self.argv)
     o, l = x.parse_options('cd', argv)
     assert o.logfile == '/var/log'
     assert l == [
         '--foobar=10,2', '-c', '1',
         '-lDEBUG', '--logfile=/var/log',
         '--pidfile=celeryd.pid',
     ]
     x.parse_options('cd', [])  # no args
 def test_parse_options(self):
     x = detached_celeryd(app=self.app)
     _, argv = x._split_command_line_config(self.argv)
     o, l = x.parse_options('cd', argv)
     assert o.logfile == '/var/log'
     assert l == [
         '--foobar=10,2', '-c', '1',
         '-lDEBUG', '--logfile=/var/log',
         '--pidfile=celeryd.pid',
     ]
     x.parse_options('cd', [])  # no args
Example #14
0
 def test_execute_from_commandline(self, detach, exit):
     x = detached_celeryd()
     x.execute_from_commandline(self.argv)
     self.assertTrue(exit.called)
     detach.assert_called_with(path=x.execv_path, uid=None, gid=None,
         umask=0, working_directory=None, fake=False,
         logfile='/var/log', pidfile='celeryd.pid',
         argv=['-m', 'celery.bin.celeryd', '-c', '1', '-lDEBUG',
               '--logfile=/var/log', '--pidfile=celeryd.pid',
               '--', '.disable_rate_limits=1'],
     )
Example #15
0
 def test_execute_from_commandline(self, detach, exit):
     x = detached_celeryd(app=self.app)
     x.execute_from_commandline(self.argv)
     exit.assert_called()
     detach.assert_called_with(
         path=x.execv_path, uid=None, gid=None,
         umask=None, fake=False, logfile='/var/log', pidfile='celeryd.pid',
         workdir=None, executable=None, hostname=None,
         argv=x.execv_argv + [
             '-c', '1', '-lDEBUG',
             '--logfile=/var/log', '--pidfile=celeryd.pid',
             '--', '.disable_rate_limits=1'
         ],
         app=self.app,
     )
 def test_execute_from_commandline(self, detach, exit):
     x = detached_celeryd(app=self.app)
     x.execute_from_commandline(self.argv)
     exit.assert_called()
     detach.assert_called_with(
         path=x.execv_path, uid=None, gid=None,
         umask=None, fake=False, logfile='/var/log', pidfile='celeryd.pid',
         workdir=None, executable=None, hostname=None,
         argv=x.execv_argv + [
             '-c', '1', '-lDEBUG',
             '--logfile=/var/log', '--pidfile=celeryd.pid',
             '--', '.disable_rate_limits=1'
         ],
         app=self.app,
     )
Example #17
0
 def test_execute_from_commandline(self, detach, exit):
     x = detached_celeryd(app=self.app)
     x.execute_from_commandline(self.argv)
     self.assertTrue(exit.called)
     detach.assert_called_with(
         path=x.execv_path,
         uid=None,
         gid=None,
         umask=None,
         fake=False,
         logfile="/var/log",
         pidfile="celeryd.pid",
         working_directory=None,
         argv=x.execv_argv
         + ["-c", "1", "-lDEBUG", "--logfile=/var/log", "--pidfile=celeryd.pid", "--", ".disable_rate_limits=1"],
         app=self.app,
     )
Example #18
0
    def test_parser(self):
        x = detached_celeryd(self.app)
        p = x.Parser("celeryd_detach")
        options, values = p.parse_args(["--logfile=foo", "--fake", "--enable", "a", "b", "-c1", "-d", "2"])
        self.assertEqual(options.logfile, "foo")
        self.assertEqual(values, ["a", "b"])
        self.assertEqual(p.leftovers, ["--enable", "-c1", "-d", "2"])

        with override_stdouts():
            with self.assertRaises(SystemExit):
                p.parse_args(["--logfile"])
            p.get_option("--logfile").nargs = 2
            with self.assertRaises(SystemExit):
                p.parse_args(["--logfile=a"])
            with self.assertRaises(SystemExit):
                p.parse_args(["--fake=abc"])

        assert p.get_option("--logfile").nargs == 2
        p.parse_args(["--logfile=a", "b"])
        p.get_option("--logfile").nargs = 1
Example #19
0
    def test_parser(self):
        x = detached_celeryd()
        p = x.Parser("celeryd_detach")
        options, values = p.parse_args(['--logfile=foo', "--fake", "--enable",
                                        "a", "b", "-c1", "-d", "2"])
        self.assertEqual(options.logfile, "foo")
        self.assertEqual(values, ["a", "b"])
        self.assertEqual(p.leftovers, ["--enable", "-c1", "-d", "2"])

        with override_stdouts():
            with self.assertRaises(SystemExit):
                p.parse_args(["--logfile"])
            p.get_option("--logfile").nargs = 2
            with self.assertRaises(SystemExit):
                p.parse_args(["--logfile=a"])
            with self.assertRaises(SystemExit):
                p.parse_args(["--fake=abc"])

        assert p.get_option("--logfile").nargs == 2
        p.parse_args(["--logfile=a", "b"])
        p.get_option("--logfile").nargs = 1
Example #20
0
    def test_parser(self):
        x = detached_celeryd(self.app)
        p = x.create_parser("celeryd_detach")
        options, values = p.parse_args(["--logfile=foo", "--fake", "--enable", "a", "b", "-c1", "-d", "2"])
        print(p.option_list)
        print("O: %r V: %r" % (vars(options), values))
        self.assertEqual(options.logfile, "foo")
        self.assertEqual(values, ["a", "b"])
        self.assertEqual(p.leftovers, ["--enable", "-c1", "-d", "2"])
        options, values = p.parse_args(["--fake", "--enable", "--pidfile=/var/pid/foo.pid", "a", "b", "-c1", "-d", "2"])
        self.assertEqual(options.pidfile, "/var/pid/foo.pid")

        with override_stdouts():
            with self.assertRaises(SystemExit):
                p.parse_args(["--logfile"])
            p.get_option("--logfile").nargs = 2
            with self.assertRaises(SystemExit):
                p.parse_args(["--logfile=a"])
            with self.assertRaises(SystemExit):
                p.parse_args(["--fake=abc"])

        assert p.get_option("--logfile").nargs == 2
        p.parse_args(["--logfile=a", "b"])
        p.get_option("--logfile").nargs = 1
Example #21
0
 def maybe_detach(self, argv, dopts=["-D", "--detach"]):
     if any(arg in argv for arg in dopts):
         argv = [v for v in argv if v not in dopts]
         # never returns
         detached_celeryd().execute_from_commandline(argv)
         raise SystemExit(0)
Example #22
0
 def maybe_detach(self, argv, dopts=['-D', '--detach']):
     if any(arg in argv for arg in dopts):
         argv = [v for v in argv if v not in dopts]
         # will never return
         detached_celeryd(self.app).execute_from_commandline(argv)
         raise SystemExit(0)
Example #23
0
 def maybe_detach(self, argv, dopts=['-D', '--detach']):
     if any(arg in argv for arg in dopts):
         argv = [v for v in argv if v not in dopts]
         # never returns
         detached_celeryd().execute_from_commandline(argv)
         raise SystemExit(0)