12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067 |
- _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:
- level = None
- 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 addHandler(self, handler):
- pass
- 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
- def set_terminator(self, terminator):
- 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)
|