def testBadConnectionStrings(self):
with self.assertRaises(ValueError):
storage_interface.USSMetadataManager(
'terrible:connection:1459231232133_string-#$%@',
testgroupid=TESTID)
with self.assertRaises(ValueError):
storage_interface.USSMetadataManager('__init__%password%',
testgroupid=TESTID)
with self.assertRaises(ValueError):
storage_interface.USSMetadataManager('\'printf();\'',
testgroupid=TESTID)
with self.assertRaises(KazooTimeoutError):
storage_interface.USSMetadataManager('123456789101112',
testgroupid=TESTID)
with self.assertRaises(KazooTimeoutError):
storage_interface.USSMetadataManager(
'google.com:2424,gmail.com:14566', testgroupid=TESTID)
def InitializeConnection(options=None):
"""Initializes the wrapper and the connection to the zookeeper servers.
The side effects of this method are to set the global variables 'wrapper' and
'TESTID'.
Args:
options: Options structure with a field per option.
"""
global wrapper, TESTID
if (options and options.verbose) or os.environ.get('INTERUSS_VERBOSE'):
log.setLevel(logging.DEBUG)
log.debug('Initializing USS metadata manager...')
wrapper = storage_interface.USSMetadataManager(
os.getenv('INTERUSS_CONNECTIONSTRING',
options.connectionstring if options else None))
if (options and options.verbose) or os.environ.get('INTERUSS_VERBOSE'):
wrapper.set_verbose()
if (options and options.testid) or os.environ.get('INTERUSS_TESTID'):
TESTID = os.getenv('INTERUSS_TESTID', options.testid if options else None)
log.info('Authorization set to test mode with TESTID=%s' % TESTID)
wrapper.set_testmode(TESTID)
wrapper.delete_testdata(TESTID)
def InitializeConnection(argv):
"""Initializes the wrapper and the connection to the zookeeper servers.
Uses the command line parameters as argv, which can be altered as needed for
testing.
Args:
argv: Command line parameters
Returns:
Host and port to use for the server
"""
global wrapper, TESTID
log.debug('Parsing command line arguments...')
parser = OptionParser(usage='usage: %prog [options]',
version='%prog ' + VERSION)
parser.add_option('-z',
'--zookeeper-servers',
dest='connectionstring',
help='Specific zookeeper server connection string, '
'server:port,server:port...'
'[or env variable INTERUSS_CONNECTIONSTRING]',
metavar='CONNECTIONSTRING')
parser.add_option(
'-s',
'--server',
dest='server',
default=os.getenv('INTERUSS_API_SERVER', '127.0.0.1'),
help='Specific server name to use on this machine for the web services '
'[or env variable INTERUSS_API_SERVER]',
metavar='SERVER')
parser.add_option(
'-p',
'--port',
dest='port',
default=os.getenv('INTERUSS_API_PORT', '5000'),
help='Specific port to use on this machine for the web services '
'[or env variable INTERUSS_API_PORT]',
metavar='PORT')
parser.add_option(
'-v',
'--verbose',
action='store_true',
dest='verbose',
default=False,
help='Verbose (DEBUG) logging [or env variable INTERUSS_VERBOSE]')
parser.add_option(
'-t',
'--testid',
dest='testid',
default=False,
help='Force testing mode with test data located in specific test id '
'[or env variable INTERUSS_TESTID]',
metavar='TESTID')
(options, args) = parser.parse_args(argv)
del args
if options.verbose or os.environ.get('INTERUSS_VERBOSE'):
log.setLevel(logging.DEBUG)
log.debug('Initializing USS metadata manager...')
wrapper = storage_interface.USSMetadataManager(
os.getenv('INTERUSS_CONNECTIONSTRING', options.connectionstring))
if options.verbose or os.environ.get('INTERUSS_VERBOSE'):
wrapper.set_verbose()
if options.testid or os.environ.get('INTERUSS_TESTID'):
TESTID = os.getenv('INTERUSS_TESTID', options.testid)
wrapper.set_testmode(TESTID)
wrapper.delete_testdata(TESTID)
return options.server, options.port
def setUp(self):
# IMPORTANT: Puts us in a test data location
self.mm = storage_interface.USSMetadataManager(
ZK_TEST_CONNECTION_STRING,
testgroupid='InterUSSStorageInterfaceTest')
def setUp(self):
# IMPORTANT: Puts us in a test data location
self.mm = storage_interface.USSMetadataManager(
ZK_TEST_CONNECTION_STRING, testgroupid=TESTID)
self.mm.set_verbose()
