def test_obscure_string(self): result = utils.obscure_string('1234567890') self.assertEquals(result, '******7890') result = utils.obscure_string(None) self.assertTrue(result is None) result = utils.obscure_string('123') self.assertEquals(result, '123') result = utils.obscure_string('abcd') self.assertEquals(result, 'abcd')
def set_environment_variables(self, parameters): """ Set the EC2 specific environment variables. Required values for the environment variables are read from the 'credentials' parameter of the parameters map. (Also see documentation for the BaseAgent class) Args: parameters A dictionary containing the 'credentials' key """ if os.environ.has_key('EC2_JVM_ARGS'): del(os.environ['EC2_JVM_ARGS']) variables = parameters[self.PARAM_CREDENTIALS] for key, value in variables.items(): if value is None: utils.log('None value detected for the credential: {0}.'.format(key)) continue if key.find('KEY') != -1: utils.log('Setting {0} to {1} in our environment.'.format( key, utils.obscure_string(value))) else: utils.log('Setting {0} to {1} in our environment.'.format(key, value)) os.environ[key] = value ec2_keys_dir = os.path.abspath('/etc/appscale/keys/cloud1') os.environ['EC2_PRIVATE_KEY'] = ec2_keys_dir + '/mykey.pem' os.environ['EC2_CERT'] = ec2_keys_dir + '/mycert.pem' utils.log('Setting private key to: {0} and certificate to: {1}'.format( os.environ['EC2_PRIVATE_KEY'], os.environ['EC2_CERT']))