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()