147 lines
4.5 KiB
Python
147 lines
4.5 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2010-2011 OpenStack LLC.
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import ConfigParser
|
|
from hashlib import md5
|
|
import nose.plugins.skip
|
|
import os
|
|
import unittest2
|
|
|
|
NOVA_DATA = {}
|
|
GLANCE_DATA = {}
|
|
SWIFT_DATA = {}
|
|
RABBITMQ_DATA = {}
|
|
CONFIG_DATA = {}
|
|
KEYSTONE_DATA = {}
|
|
|
|
|
|
class skip_test(object):
|
|
"""Decorator that skips a test."""
|
|
def __init__(self, msg):
|
|
self.message = msg
|
|
|
|
def __call__(self, func):
|
|
def _skipper(*args, **kw):
|
|
"""Wrapped skipper function."""
|
|
raise nose.SkipTest(self.message)
|
|
_skipper.__name__ = func.__name__
|
|
_skipper.__doc__ = func.__doc__
|
|
return _skipper
|
|
|
|
|
|
class skip_if(object):
|
|
"""Decorator that skips a test."""
|
|
def __init__(self, condition, msg):
|
|
self.condition = condition
|
|
self.message = msg
|
|
|
|
def __call__(self, func):
|
|
def _skipper(*args, **kw):
|
|
"""Wrapped skipper function."""
|
|
if self.condition:
|
|
raise nose.SkipTest(self.message)
|
|
func(*args, **kw)
|
|
_skipper.__name__ = func.__name__
|
|
_skipper.__doc__ = func.__doc__
|
|
return _skipper
|
|
|
|
|
|
class skip_unless(object):
|
|
"""Decorator that skips a test."""
|
|
def __init__(self, condition, msg):
|
|
self.condition = condition
|
|
self.message = msg
|
|
|
|
def __call__(self, func):
|
|
def _skipper(*args, **kw):
|
|
"""Wrapped skipper function."""
|
|
if not self.condition:
|
|
raise nose.SkipTest(self.message)
|
|
func(*args, **kw)
|
|
_skipper.__name__ = func.__name__
|
|
_skipper.__doc__ = func.__doc__
|
|
return _skipper
|
|
|
|
|
|
class FunctionalTest(unittest2.TestCase):
|
|
def setUp(self):
|
|
global GLANCE_DATA, NOVA_DATA, SWIFT_DATA, RABBITMQ_DATA
|
|
global KEYSTONE_DATA, CONFIG_DATA
|
|
# Define config dict
|
|
self.config = CONFIG_DATA
|
|
# Define service specific dicts
|
|
self.glance = GLANCE_DATA
|
|
self.nova = NOVA_DATA
|
|
self.swift = SWIFT_DATA
|
|
self.rabbitmq = RABBITMQ_DATA
|
|
self.keystone = KEYSTONE_DATA
|
|
|
|
self._parse_defaults_file()
|
|
|
|
# Swift Setup
|
|
if 'swift' in self.config:
|
|
self.swift = self.config['swift']
|
|
self.swift['ver'] = 'v1.0' # need to find a better way to get this
|
|
|
|
# Glance Setup
|
|
self.glance = self.config['glance']
|
|
|
|
if 'nova' in self.config:
|
|
self.nova = self.config['nova']
|
|
self.nova['ver'] = self.config['nova']['apiver']
|
|
|
|
if 'keystone' in self.config:
|
|
self.keystone = self.config['keystone']
|
|
self.keystone['pass'] = self.config['keystone']['password']
|
|
|
|
def _md5sum_file(self, path):
|
|
md5sum = md5()
|
|
with open(path, 'rb') as file:
|
|
for chunk in iter(lambda: file.read(8192), ''):
|
|
md5sum.update(chunk)
|
|
return md5sum.hexdigest()
|
|
|
|
def _read_in_chunks(self, infile, chunk_size=1024 * 64):
|
|
file_data = open(infile, "rb")
|
|
while True:
|
|
# chunk = file_data.read(chunk_size).encode('base64')
|
|
chunk = file_data.read(chunk_size)
|
|
if chunk:
|
|
yield chunk
|
|
else:
|
|
return
|
|
file_data.close()
|
|
|
|
def _parse_defaults_file(self):
|
|
cfg = os.path.abspath(os.path.join(os.path.dirname(__file__),
|
|
os.path.pardir, os.path.pardir,
|
|
"etc", "config.ini"))
|
|
if os.path.exists(cfg):
|
|
self._build_config(cfg)
|
|
else:
|
|
raise Exception("Cannot read %s" % cfg)
|
|
|
|
def _build_config(self, config_file):
|
|
parser = ConfigParser.ConfigParser()
|
|
parser.read(config_file)
|
|
|
|
for section in parser.sections():
|
|
self.config[section] = {}
|
|
for value in parser.options(section):
|
|
self.config[section][value] = parser.get(section, value)
|
|
# print "%s = %s" % (value, parser.get(section, value))
|