123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059 |
- _NOW = 1151365354
- _TIMEFORMAT = '%b %d %I:%M %p'
- class DummyOptions:
- make_pipes_error = None
- fork_error = None
- execv_error = None
- kill_error = None
- minfds = 5
- loglevel = 20
- def __init__(self):
- self.identifier = 'supervisor'
- self.childlogdir = '/tmp'
- self.uid = 999
- self.logger = self.getLogger()
- self.backofflimit = 10
- self.logfile = '/tmp/logfile'
- self.nocleanup = False
- self.strip_ansi = False
- self.pidhistory = {}
- self.process_group_configs = []
- self.nodaemon = False
- self.socket_map = {}
- self.mood = 1
- self.mustreopen = False
- self.realizeargs = None
- self.fds_cleaned_up = False
- self.rlimit_set = False
- self.setuid_called = False
- self.httpservers_opened = False
- self.signals_set = False
- self.daemonized = False
- self.make_logger_messages = None
- self.autochildlogdir_cleared = False
- self.cleaned_up = False
- self.pidfile_written = False
- self.directory = None
- self.waitpid_return = None, None
- self.kills = {}
- self._signal = None
- self.parent_pipes_closed = None
- self.child_pipes_closed = None
- self.forkpid = 0
- self.pgrp_set = None
- self.duped = {}
- self.written = {}
- self.fds_closed = []
- self._exitcode = None
- self.execve_called = False
- self.execv_args = None
- self.setuid_msg = None
- self.privsdropped = None
- self.logs_reopened = False
- self.environment_processed = False
- self.select_result = [], [], []
- self.select_error = None
- self.write_accept = None
- self.write_error = None
- self.tempfile_name = '/foo/bar'
- self.remove_error = None
- self.removed = []
- self.existing = []
- self.openreturn = None
- self.readfd_result = ''
- self.parse_warnings = []
- self.serverurl = 'http://localhost:9001'
- self.changed_directory = False
- self.chdir_error = None
- self.umaskset = None
- def getLogger(self, *args, **kw):
- logger = DummyLogger()
- logger.handlers = [DummyLogger()]
- logger.args = args, kw
- return logger
- def realize(self, args, **kw):
- self.realizeargs = args
- self.realizekw = kw
- def process_config(self, do_usage=True):
- pass
- def cleanup_fds(self):
- self.fds_cleaned_up = True
- def set_rlimits(self):
- self.rlimits_set = True
- return ['rlimits_set']
- def set_uid(self):
- self.setuid_called = True
- return 'setuid_called'
- def openhttpservers(self, supervisord):
- self.httpservers_opened = True
- def daemonize(self):
- self.daemonized = True
- def setsignals(self):
- self.signals_set = True
- def get_signal(self):
- return self._signal
- def get_socket_map(self):
- return self.socket_map
- def make_logger(self, critical_msgs, warn_msgs, info_msgs):
- self.make_logger_messages = critical_msgs, warn_msgs, info_msgs
- def clear_autochildlogdir(self):
- self.autochildlogdir_cleared = True
- def get_autochildlog_name(self, *ignored):
- return self.tempfile_name
- def cleanup(self):
- self.cleaned_up = True
- def write_pidfile(self):
- self.pidfile_written = True
- def waitpid(self):
- return self.waitpid_return
- def kill(self, pid, sig):
- if self.kill_error:
- raise OSError(self.kill_error)
- self.kills[pid] = sig
- def stat(self, filename):
- import os
- return os.stat(filename)
- def get_path(self):
- return ["/bin", "/usr/bin", "/usr/local/bin"]
- def get_pid(self):
- import os
- return os.getpid()
-
- def check_execv_args(self, filename, argv, st):
- if filename == '/bad/filename':
- from supervisor.options import NotFound
- raise NotFound('bad filename')
- def make_pipes(self, stderr=True):
- if self.make_pipes_error:
- raise OSError(self.make_pipes_error)
- pipes = {}
- pipes['child_stdin'], pipes['stdin'] = (3, 4)
- pipes['stdout'], pipes['child_stdout'] = (5, 6)
- if stderr:
- pipes['stderr'], pipes['child_stderr'] = (7, 8)
- else:
- pipes['stderr'], pipes['child_stderr'] = None, None
- return pipes
- def write(self, fd, chars):
- if self.write_error:
- raise OSError(self.write_error)
- if self.write_accept:
- chars = chars[self.write_accept]
- data = self.written.setdefault(fd, '')
- data += chars
- self.written[fd] = data
- return len(chars)
- def fork(self):
- if self.fork_error:
- raise OSError(self.fork_error)
- return self.forkpid
- def close_fd(self, fd):
- self.fds_closed.append(fd)
- def close_parent_pipes(self, pipes):
- self.parent_pipes_closed = pipes
- def close_child_pipes(self, pipes):
- self.child_pipes_closed = pipes
- def setpgrp(self):
- self.pgrp_set = True
- def dup2(self, frm, to):
- self.duped[frm] = to
- def _exit(self, code):
- self._exitcode = code
- def execve(self, filename, argv, environment):
- self.execve_called = True
- if self.execv_error:
- if self.execv_error == 1:
- raise OSError(self.execv_error)
- else:
- raise RuntimeError(self.execv_error)
- self.execv_args = (filename, argv)
- self.execv_environment = environment
- def dropPrivileges(self, uid):
- if self.setuid_msg:
- return self.setuid_msg
- self.privsdropped = uid
- def readfd(self, fd):
- return self.readfd_result
- def reopenlogs(self):
- self.logs_reopened = True
- def process_environment(self):
- self.environment_processed = True
- def mktempfile(self, prefix, suffix, dir):
- return self.tempfile_name
- def select(self, r, w, x, timeout):
- import select
- if self.select_error:
- raise select.error(self.select_error)
- return self.select_result
- def remove(self, path):
- import os
- if self.remove_error:
- raise os.error(self.remove_error)
- self.removed.append(path)
- def exists(self, path):
- if path in self.existing:
- return True
- return False
- def open(self, name, mode='r'):
- if self.openreturn:
- return self.openreturn
- return open(name, mode)
- def chdir(self, dir):
- if self.chdir_error:
- raise OSError(self.chdir_error)
- self.changed_directory = True
- def setumask(self, mask):
- self.umaskset = mask
- class DummyLogger:
- def __init__(self):
- self.reopened = False
- self.removed = False
- self.closed = False
- self.data = []
- def info(self, msg, **kw):
- if kw:
- msg = msg % kw
- self.data.append(msg)
- warn = debug = critical = trace = error = blather = info
- def log(self, level, msg, **kw):
- if kw:
- msg = msg % kw
- self.data.append(msg)
-
- def reopen(self):
- self.reopened = True
- def close(self):
- self.closed = True
- def remove(self):
- self.removed = True
- def flush(self):
- self.flushed = True
- def getvalue(self):
- return ''.join(self.data)
- class DummySupervisor:
- def __init__(self, options=None, state=None, process_groups=None):
- if options is None:
- self.options = DummyOptions()
- else:
- self.options = options
- if state is None:
- from supervisor.supervisord import SupervisorStates
- self.options.mood = SupervisorStates.RUNNING
- else:
- self.options.mood = state
- if process_groups is None:
- self.process_groups = {}
- else:
- self.process_groups = process_groups
- def get_state(self):
- return self.options.mood
- class DummySocket:
- bind_called = False
- bind_addr = None
- listen_called = False
- listen_backlog = None
- close_called = False
- def __init__(self, fd):
- self.fd = fd
- def fileno(self):
- return self.fd
- def bind(self, addr):
- self.bind_called = True
- self.bind_addr = addr
- def listen(self, backlog):
- self.listen_called = True
- self.listen_backlog = backlog
- def close(self):
- self.close_called = True
- def __str__(self):
- return 'dummy socket'
- class DummySocketConfig:
- def __init__(self, fd):
- self.fd = fd
- def addr(self):
- return 'dummy addr'
- def __eq__(self, other):
- return self.fd == other.fd
- def __ne__(self, other):
- return not self.__eq__(other)
- def create_and_bind(self):
- return DummySocket(self.fd)
- class DummySocketManager:
- def __init__(self, config, **kwargs):
- self._config = config
- def config(self):
- return self._config
- def get_socket(self):
- return DummySocket(self._config.fd)
-
- class DummyProcess:
- # Initial state; overridden by instance variables
- pid = 0 # Subprocess pid; 0 when not running
- laststart = 0 # Last time the subprocess was started; 0 if never
- laststop = 0 # Last time the subprocess was stopped; 0 if never
- delay = 0 # If nonzero, delay starting or killing until this time
- administrative_stop = 0 # true if the process has been stopped by an admin
- system_stop = 0 # true if the process has been stopped by the system
- killing = 0 # flag determining whether we are trying to kill this proc
- backoff = 0 # backoff counter (to backofflimit)
- waitstatus = None
- exitstatus = None
- pipes = None
- rpipes = None
- dispatchers = None
- stdout_logged = ''
- stderr_logged = ''
- spawnerr = None
- stdout_buffer = '' # buffer of characters from child stdout output to log
- stderr_buffer = '' # buffer of characters from child stderr output to log
- stdin_buffer = '' # buffer of characters to send to child process' stdin
- listener_state = None
- group = None
- def __init__(self, config, state=None):
- self.config = config
- self.logsremoved = False
- self.stop_called = False
- self.backoff_secs = None
- self.spawned = False
- if state is None:
- from supervisor.process import ProcessStates
- state = ProcessStates.RUNNING
- self.state = state
- self.error_at_clear = False
- self.killed_with = None
- self.drained = False
- self.stdout_buffer = ''
- self.stderr_buffer = ''
- self.stdout_logged = ''
- self.stderr_logged = ''
- self.stdin_buffer = ''
- self.pipes = {}
- self.rpipes = {}
- self.dispatchers = {}
- self.finished = None
- self.logs_reopened = False
- self.execv_arg_exception = None
- self.input_fd_drained = None
- self.output_fd_drained = None
- self.transitioned = False
- self.write_error = None
- def reopenlogs(self):
- self.logs_reopened = True
- def removelogs(self):
- if self.error_at_clear:
- raise IOError('whatever')
- self.logsremoved = True
- def get_state(self):
- return self.state
- def stop(self):
- self.stop_called = True
- self.killing = False
- from supervisor.process import ProcessStates
- self.state = ProcessStates.STOPPED
- def kill(self, signal):
- self.killed_with = signal
- def spawn(self):
- self.spawned = True
- from supervisor.process import ProcessStates
- self.state = ProcessStates.RUNNING
- def drain(self):
- self.drained = True
- def __cmp__(self, other):
- return cmp(self.config.priority, other.config.priority)
- def readable_fds(self):
- return []
- def record_output(self):
- self.stdout_logged += self.stdout_buffer
- self.stdout_buffer = ''
- self.stderr_logged += self.stderr_buffer
- self.stderr_buffer = ''
- def finish(self, pid, sts):
- self.finished = pid, sts
- def give_up(self):
- from supervisor.process import ProcessStates
- self.state = ProcessStates.FATAL
- def get_execv_args(self):
- if self.execv_arg_exception:
- raise self.execv_arg_exception('whatever')
- import shlex
- commandargs = shlex.split(self.config.command)
- program = commandargs[0]
- return program, commandargs
- def drain_output_fd(self, fd):
- self.output_fd_drained = fd
- def drain_input_fd(self, fd):
- self.input_fd_drained = fd
- def write(self, chars):
- if self.write_error:
- raise OSError(self.write_error)
- self.stdin_buffer += chars
- def transition(self):
- self.transitioned = True
- class DummyPConfig:
- def __init__(self, options, name, command, directory=None, umask=None,
- priority=999, autostart=True,
- autorestart=True, startsecs=10, startretries=999,
- uid=None, stdout_logfile=None, stdout_capture_maxbytes=0,
- stdout_events_enabled=False,
- stdout_logfile_backups=0, stdout_logfile_maxbytes=0,
- stderr_logfile=None, stderr_capture_maxbytes=0,
- stderr_events_enabled=False,
- stderr_logfile_backups=0, stderr_logfile_maxbytes=0,
- redirect_stderr=False,
- stopsignal=None, stopwaitsecs=10, stopasgroup=False, killasgroup=False,
- exitcodes=(0,2), environment=None, serverurl=None):
- self.options = options
- self.name = name
- self.command = command
- self.priority = priority
- self.autostart = autostart
- self.autorestart = autorestart
- self.startsecs = startsecs
- self.startretries = startretries
- self.uid = uid
- self.stdout_logfile = stdout_logfile
- self.stdout_capture_maxbytes = stdout_capture_maxbytes
- self.stdout_events_enabled = stdout_events_enabled
- self.stdout_logfile_backups = stdout_logfile_backups
- self.stdout_logfile_maxbytes = stdout_logfile_maxbytes
- self.stderr_logfile = stderr_logfile
- self.stderr_capture_maxbytes = stderr_capture_maxbytes
- self.stderr_events_enabled = stderr_events_enabled
- self.stderr_logfile_backups = stderr_logfile_backups
- self.stderr_logfile_maxbytes = stderr_logfile_maxbytes
- self.redirect_stderr = redirect_stderr
- if stopsignal is None:
- import signal
- stopsignal = signal.SIGTERM
- self.stopsignal = stopsignal
- self.stopwaitsecs = stopwaitsecs
- self.stopasgroup = stopasgroup
- self.killasgroup = killasgroup
- self.exitcodes = exitcodes
- self.environment = environment
- self.directory = directory
- self.umask = umask
- self.autochildlogs_created = False
- self.serverurl = serverurl
- def create_autochildlogs(self):
- self.autochildlogs_created = True
- def make_process(self, group=None):
- process = DummyProcess(self)
- process.group = group
- return process
- def make_dispatchers(self, proc):
- use_stderr = not self.redirect_stderr
- pipes = self.options.make_pipes(use_stderr)
- stdout_fd,stderr_fd,stdin_fd = (pipes['stdout'],pipes['stderr'],
- pipes['stdin'])
- dispatchers = {}
- if stdout_fd is not None:
- dispatchers[stdout_fd] = DummyDispatcher(readable=True)
- if stderr_fd is not None:
- dispatchers[stderr_fd] = DummyDispatcher(readable=True)
- if stdin_fd is not None:
- dispatchers[stdin_fd] = DummyDispatcher(writable=True)
- return dispatchers, pipes
- def makeExecutable(file, substitutions=None):
- import os
- import sys
- import tempfile
-
- if substitutions is None:
- substitutions = {}
- data = open(file).read()
- last = os.path.split(file)[1]
- substitutions['PYTHON'] = sys.executable
- for key in substitutions.keys():
- data = data.replace('<<%s>>' % key.upper(), substitutions[key])
-
- tmpnam = tempfile.mktemp(prefix=last)
- f = open(tmpnam, 'w')
- f.write(data)
- f.close()
- os.chmod(tmpnam, 0755)
- return tmpnam
- def makeSpew(unkillable=False):
- import os
- here = os.path.dirname(__file__)
- if not unkillable:
- return makeExecutable(os.path.join(here, 'fixtures/spew.py'))
- return makeExecutable(os.path.join(here, 'fixtures/unkillable_spew.py'))
- class DummyMedusaServerLogger:
- def __init__(self):
- self.logged = []
- def log(self, category, msg):
- self.logged.append((category, msg))
- class DummyMedusaServer:
- def __init__(self):
- self.logger = DummyMedusaServerLogger()
- class DummyMedusaChannel:
- def __init__(self):
- self.server = DummyMedusaServer()
- self.producer = None
- def push_with_producer(self, producer):
- self.producer = producer
- def close_when_done(self):
- pass
- class DummyRequest:
- command = 'GET'
- _error = None
- _done = False
- version = '1.0'
- def __init__(self, path, params, query, fragment, env=None):
- self.args = path, params, query, fragment
- self.producers = []
- self.headers = {}
- self.header = []
- self.outgoing = []
- self.channel = DummyMedusaChannel()
- if env is None:
- self.env = {}
- else:
- self.env = env
- def split_uri(self):
- return self.args
- def error(self, code):
- self._error = code
- def push(self, producer):
- self.producers.append(producer)
- def __setitem__(self, header, value):
- self.headers[header] = value
- def has_key(self, header):
- return self.headers.has_key(header)
- def done(self):
- self._done = True
- def build_reply_header(self):
- return ''
- def log(self, *arg, **kw):
- pass
- def cgi_environment(self):
- return self.env
- def get_server_url(self):
- return 'http://example.com'
-
- class DummyRPCInterfaceFactory:
- def __init__(self, supervisord, **config):
- self.supervisord = supervisord
- self.config = config
- class DummyRPCServer:
- def __init__(self):
- self.supervisor = DummySupervisorRPCNamespace()
- self.system = DummySystemRPCNamespace()
- class DummySystemRPCNamespace:
- pass
- class DummySupervisorRPCNamespace:
- _restartable = True
- _restarted = False
- _shutdown = False
- _readlog_error = False
- from supervisor.process import ProcessStates
- all_process_info = [
- {
- 'name':'foo',
- 'group':'foo',
- 'pid':11,
- 'state':ProcessStates.RUNNING,
- 'statename':'RUNNING',
- 'start':_NOW - 100,
- 'stop':0,
- 'spawnerr':'',
- 'now':_NOW,
- 'description':'foo description',
- },
- {
- 'name':'bar',
- 'group':'bar',
- 'pid':12,
- 'state':ProcessStates.FATAL,
- 'statename':'FATAL',
- 'start':_NOW - 100,
- 'stop':_NOW - 50,
- 'spawnerr':'screwed',
- 'now':_NOW,
- 'description':'bar description',
- },
- {
- 'name':'baz_01',
- 'group':'baz',
- 'pid':13,
- 'state':ProcessStates.STOPPED,
- 'statename':'STOPPED',
- 'start':_NOW - 100,
- 'stop':_NOW - 25,
- 'spawnerr':'',
- 'now':_NOW,
- 'description':'baz description',
- },
- ]
- def getAPIVersion(self):
- return '3.0'
- getVersion = getAPIVersion # deprecated
- def getPID(self):
- return 42
- def readProcessStdoutLog(self, name, offset, length):
- from supervisor import xmlrpc
- import xmlrpclib
- if name == 'BAD_NAME':
- raise xmlrpclib.Fault(xmlrpc.Faults.BAD_NAME, 'BAD_NAME')
- elif name == 'FAILED':
- raise xmlrpclib.Fault(xmlrpc.Faults.FAILED, 'FAILED')
- elif name == 'NO_FILE':
- raise xmlrpclib.Fault(xmlrpc.Faults.NO_FILE, 'NO_FILE')
- a = 'output line\n' * 10
- return a[offset:]
- readProcessLog = readProcessStdoutLog
- readProcessStderrLog = readProcessStdoutLog
- def getAllProcessInfo(self):
- return self.all_process_info
- def getProcessInfo(self, name):
- from supervisor import xmlrpc
- import xmlrpclib
- from supervisor.process import ProcessStates
- for i in self.all_process_info:
- if i['name']==name:
- info=i
- return info
- if name == 'BAD_NAME':
- raise xmlrpclib.Fault(xmlrpc.Faults.BAD_NAME, 'BAD_NAME')
- if name == 'FAILED':
- raise xmlrpclib.Fault(xmlrpc.Faults.FAILED, 'FAILED')
- if name == 'NO_FILE':
- raise xmlrpclib.Fault(xmlrpc.Faults.NO_FILE, 'NO_FILE')
- def startProcess(self, name):
- from supervisor import xmlrpc
- from xmlrpclib import Fault
- if name == 'BAD_NAME:BAD_NAME':
- raise Fault(xmlrpc.Faults.BAD_NAME, 'BAD_NAME:BAD_NAME')
- if name == 'BAD_NAME':
- raise Fault(xmlrpc.Faults.BAD_NAME, 'BAD_NAME')
- if name == 'NO_FILE':
- raise Fault(xmlrpc.Faults.NO_FILE, 'NO_FILE')
- if name == 'NOT_EXECUTABLE':
- raise Fault(xmlrpc.Faults.NOT_EXECUTABLE, 'NOT_EXECUTABLE')
- if name == 'ALREADY_STARTED':
- raise Fault(xmlrpc.Faults.ALREADY_STARTED, 'ALREADY_STARTED')
- if name == 'SPAWN_ERROR':
- raise Fault(xmlrpc.Faults.SPAWN_ERROR, 'SPAWN_ERROR')
- return True
- def startProcessGroup(self, name):
- from supervisor import xmlrpc
- return [
- {'name':'foo_00', 'group':'foo',
- 'status': xmlrpc.Faults.SUCCESS,
- 'description': 'OK'},
- {'name':'foo_01', 'group':'foo',
- 'status':xmlrpc.Faults.SUCCESS,
- 'description': 'OK'},
- ]
- def startAllProcesses(self):
- from supervisor import xmlrpc
- return [
- {'name':'foo', 'group':'foo',
- 'status': xmlrpc.Faults.SUCCESS,
- 'description': 'OK'},
- {'name':'foo2', 'group':'foo2',
- 'status':xmlrpc.Faults.SUCCESS,
- 'description': 'OK'},
- {'name':'failed', 'group':'failed_group',
- 'status':xmlrpc.Faults.SPAWN_ERROR,
- 'description':'SPAWN_ERROR'}
- ]
- def stopProcessGroup(self, name):
- from supervisor import xmlrpc
- return [
- {'name':'foo_00', 'group':'foo',
- 'status': xmlrpc.Faults.SUCCESS,
- 'description': 'OK'},
- {'name':'foo_01', 'group':'foo',
- 'status':xmlrpc.Faults.SUCCESS,
- 'description': 'OK'},
- ]
- def stopProcess(self, name):
- from supervisor import xmlrpc
- from xmlrpclib import Fault
- if name == 'BAD_NAME:BAD_NAME':
- raise Fault(xmlrpc.Faults.BAD_NAME, 'BAD_NAME:BAD_NAME')
- if name == 'BAD_NAME':
- raise Fault(xmlrpc.Faults.BAD_NAME, 'BAD_NAME')
- if name == 'NOT_RUNNING':
- raise Fault(xmlrpc.Faults.NOT_RUNNING, 'NOT_RUNNING')
- if name == 'FAILED':
- raise Fault(xmlrpc.Faults.FAILED, 'FAILED')
-
- return True
-
- def stopAllProcesses(self):
- from supervisor import xmlrpc
- return [
- {'name':'foo','group':'foo',
- 'status': xmlrpc.Faults.SUCCESS,
- 'description': 'OK'},
- {'name':'foo2', 'group':'foo2',
- 'status':xmlrpc.Faults.SUCCESS,'description': 'OK'},
- {'name':'failed', 'group':'failed_group',
- 'status':xmlrpc.Faults.BAD_NAME,
- 'description':'FAILED'}
- ]
- def restart(self):
- if self._restartable:
- self._restarted = True
- return
- from xmlrpclib import Fault
- from supervisor import xmlrpc
- raise Fault(xmlrpc.Faults.SHUTDOWN_STATE, '')
- def shutdown(self):
- if self._restartable:
- self._shutdown = True
- return
- from xmlrpclib import Fault
- from supervisor import xmlrpc
- raise Fault(xmlrpc.Faults.SHUTDOWN_STATE, '')
- def reloadConfig(self):
- return [[['added'], ['changed'], ['removed']]]
- def addProcessGroup(self, name):
- from xmlrpclib import Fault
- from supervisor import xmlrpc
- if name == 'ALREADY_ADDED':
- raise Fault(xmlrpc.Faults.ALREADY_ADDED, '')
- if name == 'BAD_NAME':
- raise Fault(xmlrpc.Faults.BAD_NAME, '')
- if hasattr(self, 'processes'):
- self.processes.append(name)
- else:
- self.processes = [name]
- def removeProcessGroup(self, name):
- from xmlrpclib import Fault
- from supervisor import xmlrpc
- if name == 'STILL_RUNNING':
- raise Fault(xmlrpc.Faults.STILL_RUNNING, '')
- if name == 'BAD_NAME':
- raise Fault(xmlrpc.Faults.BAD_NAME, '')
- self.processes.remove(name)
- def clearProcessStdoutLog(self, name):
- from xmlrpclib import Fault
- from supervisor import xmlrpc
- if name == 'BAD_NAME':
- raise Fault(xmlrpc.Faults.BAD_NAME, 'BAD_NAME')
- return True
- clearProcessLog = clearProcessStdoutLog
- clearProcessStderrLog = clearProcessStdoutLog
- clearProcessLogs = clearProcessStdoutLog
- def clearAllProcessLogs(self):
- from supervisor import xmlrpc
- return [
- {'name':'foo', 'group':'foo',
- 'status':xmlrpc.Faults.SUCCESS,
- 'description': 'OK'},
- {'name':'foo2', 'group':'foo2',
- 'status':xmlrpc.Faults.SUCCESS,
- 'description': 'OK'},
- {'name':'failed', 'group':'failed_group',
- 'status':xmlrpc.Faults.FAILED,
- 'description':'FAILED'}
- ]
- def raiseError(self):
- raise ValueError('error')
- def getSupervisorVersion(self):
- return '3000'
- def readLog(self, whence, offset):
- if self._readlog_error:
- from xmlrpclib import Fault
- raise Fault(self._readlog_error, '')
- return 'mainlogdata'
- class DummyPGroupConfig:
- def __init__(self, options, name='whatever', priority=999, pconfigs=None):
- self.options = options
- self.name = name
- self.priority = priority
- if pconfigs is None:
- pconfigs = []
- self.process_configs = pconfigs
- self.after_setuid_called = False
- self.pool_events = []
- self.buffer_size = 10
- def after_setuid(self):
- self.after_setuid_called = True
- def make_group(self):
- return DummyProcessGroup(self)
- def __repr__(self):
- return '<%s instance at %s named %s>' % (self.__class__, id(self),
- self.name)
- class DummyFCGIGroupConfig(DummyPGroupConfig):
- def __init__(self, options, name='whatever', priority=999, pconfigs=None, socket_config=DummySocketConfig(1)):
- DummyPGroupConfig.__init__(self, options, name, priority, pconfigs)
- self.socket_config = socket_config
- class DummyProcessGroup:
- def __init__(self, config):
- self.config = config
- self.transitioned = False
- self.all_stopped = False
- self.dispatchers = {}
- self.unstopped_processes = []
- def transition(self):
- self.transitioned = True
- def stop_all(self):
- self.all_stopped = True
-
- def get_unstopped_processes(self):
- return self.unstopped_processes
- def get_dispatchers(self):
- return self.dispatchers
-
- class DummyFCGIProcessGroup(DummyProcessGroup):
-
- def __init__(self, config):
- DummyProcessGroup.__init__(self, config)
- self.socket_manager = DummySocketManager(config.socket_config)
- class PopulatedDummySupervisor(DummySupervisor):
- def __init__(self, options, group_name, *pconfigs):
- DummySupervisor.__init__(self, options)
- self.process_groups = {}
- processes = {}
- self.group_name = group_name
- gconfig = DummyPGroupConfig(options, group_name, pconfigs=pconfigs)
- pgroup = DummyProcessGroup(gconfig)
- self.process_groups[group_name] = pgroup
- for pconfig in pconfigs:
- process = DummyProcess(pconfig)
- processes[pconfig.name] = process
- pgroup.processes = processes
- def set_procattr(self, process_name, attr_name, val, group_name=None):
- if group_name is None:
- group_name = self.group_name
- process = self.process_groups[group_name].processes[process_name]
- setattr(process, attr_name, val)
- class DummyDispatcher:
- write_event_handled = False
- read_event_handled = False
- error_handled = False
- logs_reopened = False
- logs_removed = False
- closed = False
- flush_error = None
- flushed = False
- def __init__(self, readable=False, writable=False, error=False):
- self._readable = readable
- self._writable = writable
- self._error = error
- self.input_buffer = ''
- if readable:
- # only readable dispatchers should have these methods
- def reopenlogs():
- self.logs_reopened = True
- self.reopenlogs = reopenlogs
- def removelogs():
- self.logs_removed = True
- self.removelogs = removelogs
- def readable(self):
- return self._readable
- def writable(self):
- return self._writable
- def handle_write_event(self):
- if self._error:
- raise self._error
- self.write_event_handled = True
- def handle_read_event(self):
- if self._error:
- raise self._error
- self.read_event_handled = True
- def handle_error(self):
- self.error_handled = True
- def close(self):
- self.closed = True
- def flush(self):
- if self.flush_error:
- raise OSError(self.flush_error)
- self.flushed = True
-
- class DummyStream:
- def __init__(self, error=None):
- self.error = error
- self.closed = False
- self.flushed = False
- self.written = ''
- def close(self):
- if self.error:
- raise self.error
- self.closed = True
- def flush(self):
- self.flushed = True
- def write(self, msg):
- if self.error:
- raise self.error
- self.written +=msg
- def seek(self, num, whence=0):
- pass
- def tell(self):
- return len(self.written)
-
- class DummyEvent:
- def __init__(self, serial='abc'):
- if serial is not None:
- self.serial = serial
- def __str__(self):
- return 'dummy event'
-
- def dummy_handler(event, result):
- pass
- def rejecting_handler(event, result):
- from supervisor.dispatchers import RejectEvent
- raise RejectEvent(result)
- def exception_handler(event, result):
- raise ValueError(result)
- def lstrip(s):
- strings = [x.strip() for x in s.split('\n')]
- return '\n'.join(strings)
|