Commit d84b3ec8 authored by Julien Muchembled's avatar Julien Muchembled

Add script to quickly setup a NEO cluster for testing purpose

git-svn-id: https://svn.erp5.org/repos/neo/trunk@2768 71dcc9de-d417-0410-9af5-da40c76e7ee4
parent 0a20ca1f
#!/usr/bin/env python
##############################################################################
#
# Copyright (c) 2011 Nexedi SARL and Contributors. All Rights Reserved.
# Julien Muchembled <jm@nexedi.com>
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsibility of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# guarantees and support are strongly advised to contract a Free Software
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
##############################################################################
import inspect, random, signal, sys
from optparse import OptionParser
from neo.lib import logger, logging
from neo.tests import functional
def main():
args, _, _, defaults = inspect.getargspec(functional.NEOCluster.__init__)
option_list = zip(args[-len(defaults):], defaults)
parser = OptionParser(usage="%prog [options] [db...]",
description="Quickly setup a simple NEO cluster for testing purpose.")
parser.add_option('--seed', help="settings like node ports/uuids and"
" cluster name are random: pass any string to initialize the RNG")
defaults = {}
for option, default in sorted(option_list):
kw = {}
if type(default) is bool:
kw['action'] = "store_true"
defaults[option] = False
elif default is not None:
defaults[option] = default
if isinstance(default, int):
kw['type'] = "int"
parser.add_option('--' + option, **kw)
parser.set_defaults(**defaults)
options, args = parser.parse_args()
if options.verbose:
logger.PACKET_LOGGER.enable(True)
if options.seed:
functional.random = random.Random(options.seed)
cluster = functional.NEOCluster(args, **dict((x, getattr(options, x))
for x, _ in option_list))
try:
cluster.start()
logging.info("Cluster running ...")
signal.pause()
finally:
cluster.stop()
if __name__ == "__main__":
sys.exit(main())
...@@ -311,7 +311,7 @@ class NEOCluster(object): ...@@ -311,7 +311,7 @@ class NEOCluster(object):
NEOProcess(command, uuid, arguments)) NEOProcess(command, uuid, arguments))
def __allocateUUID(self): def __allocateUUID(self):
uuid = os.urandom(16) uuid = ('%032x' % random.getrandbits(128)).decode('hex')
self.uuid_set.add(uuid) self.uuid_set.add(uuid)
return uuid return uuid
......
...@@ -41,6 +41,7 @@ setup( ...@@ -41,6 +41,7 @@ setup(
'neomigrate=neo.scripts.neomigrate:main', 'neomigrate=neo.scripts.neomigrate:main',
'neostorage=neo.scripts.neostorage:main', 'neostorage=neo.scripts.neostorage:main',
'neotestrunner=neo.scripts.runner:main', 'neotestrunner=neo.scripts.runner:main',
'neosimple=neo.scripts.simple:main',
], ],
}, },
# Raah!!! I wish I could write something like: # Raah!!! I wish I could write something like:
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment