iframe-proxy | Sunbelt Computer Software

265 lines (200 loc) · 7.11 KB
#   Copyright 2013 Nebula Inc.
#   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 json
from unittest import mock
from keystoneauth1 import fixture
import requests
AUTH_TOKEN = "foobar"
AUTH_URL = "http://0.0.0.0"
USERNAME = "itchy"
PASSWORD = "scratchy"
PROJECT_NAME = "poochie"
REGION_NAME = "richie"
INTERFACE = "catchy"
VERSION = "3"
TEST_RESPONSE_DICT = fixture.V2Token(token_id=AUTH_TOKEN,
                                     user_name=USERNAME)
_s = TEST_RESPONSE_DICT.add_service('identity', name='keystone')
_s.add_endpoint(AUTH_URL + ':5000/v2.0')
_s = TEST_RESPONSE_DICT.add_service('network', name='neutron')
_s.add_endpoint(AUTH_URL + ':9696')
_s = TEST_RESPONSE_DICT.add_service('compute', name='nova')
_s.add_endpoint(AUTH_URL + ':8774/v2.1')
_s = TEST_RESPONSE_DICT.add_service('image', name='glance')
_s.add_endpoint(AUTH_URL + ':9292')
_s = TEST_RESPONSE_DICT.add_service('object', name='swift')
_s.add_endpoint(AUTH_URL + ':8080/v1')
TEST_RESPONSE_DICT_V3 = fixture.V3Token(user_name=USERNAME)
TEST_RESPONSE_DICT_V3.set_project_scope()
TEST_VERSIONS = fixture.DiscoveryList(href=AUTH_URL)
def to_unicode_dict(catalog_dict):
    """Converts dict to unicode dict
    if isinstance(catalog_dict, dict):
        return {to_unicode_dict(key): to_unicode_dict(value)
                for key, value in catalog_dict.items()}
    elif isinstance(catalog_dict, list):
        return [to_unicode_dict(element) for element in catalog_dict]
    elif isinstance(catalog_dict, str):
        return catalog_dict + u""
        return catalog_dict
class FakeStdout(object):
    def __init__(self):
        self.content = []
    def write(self, text):
        self.content.append(text)
    def make_string(self):
        result = ''
        for line in self.content:
            result = result + line
        return result
class FakeLog(object):
    def __init__(self):
        self.messages = {}
    def debug(self, msg):
        self.messages['debug'] = msg
    def info(self, msg):
        self.messages['info'] = msg
    def warning(self, msg):
        self.messages['warning'] = msg
    def error(self, msg):
        self.messages['error'] = msg
    def critical(self, msg):
        self.messages['critical'] = msg
class FakeApp(object):
    def __init__(self, _stdout, _log):
        self.stdout = _stdout
        self.client_manager = None
        self.api_version = {}
        self.stdin = sys.stdin
        self.stdout = _stdout or sys.stdout
        self.stderr = sys.stderr
        self.log = _log
class FakeOptions(object):
    def __init__(self, **kwargs):
        self.os_beta_command = False
class FakeClient(object):
    def __init__(self, **kwargs):
        self.endpoint = kwargs['endpoint']
        self.token = kwargs['token']
class FakeClientManager(object):
    _api_version = {
        'image': '2',
    def __init__(self):
        self.compute = None
        self.identity = None
        self.image = None
        self.object_store = None
        self.volume = None
        self.network = None
        self.session = None
        self.auth_ref = None
        self.auth_plugin_name = None
        self.network_endpoint_enabled = True
        self.compute_endpoint_enabled = True
        self.volume_endpoint_enabled = True
    def get_configuration(self):
        return {
            'auth': {
                'username': USERNAME,
                'password': PASSWORD,
                'token': AUTH_TOKEN,
            'region': REGION_NAME,
            'identity_api_version': VERSION,
    def is_network_endpoint_enabled(self):
        return self.network_endpoint_enabled
    def is_compute_endpoint_enabled(self):
        return self.compute_endpoint_enabled
    def is_volume_endpoint_enabled(self, client):
        return self.volume_endpoint_enabled
class FakeModule(object):
    def __init__(self, name, version):
        self.name = name
        self.__version__ = version
        # Workaround for openstacksdk case
        self.version = mock.Mock()
        self.version.__version__ = version
class FakeResource(object):
    def __init__(self, manager=None, info=None, loaded=False, methods=None):
        """Set attributes and methods for a resource.
        :param manager:
            The resource manager
        :param Dictionary info:
            A dictionary with all attributes
        :param bool loaded:
            True if the resource is loaded in memory
        :param Dictionary methods:
            A dictionary with all methods
        """
        info = info or {}
        methods = methods or {}
        self.__name__ = type(self).__name__
        self.manager = manager
        self._info = info
        self._add_details(info)
        self._add_methods(methods)
        self._loaded = loaded
    def _add_details(self, info):
        for (k, v) in info.items():
            setattr(self, k, v)
    def _add_methods(self, methods):
        """Fake methods with MagicMock objects.
        For each <@key, @value> pairs in methods, add an callable MagicMock
        object named @key as an attribute, and set the mock's return_value to
        @value. When users access the attribute with (), @value will be
        returned, which looks like a function call.
        """
        for (name, ret) in methods.items():
            method = mock.Mock(return_value=ret)
            setattr(self, name, method)
    def __repr__(self):
        reprkeys = sorted(k for k in self.__dict__.keys() if k[0] != '_' and
                          k != 'manager')
        info = ", ".join("%s=%s" % (k, getattr(self, k)) for k in reprkeys)
        return "<%s %s>" % (self.__class__.__name__, info)
    def keys(self):
        return self._info.keys()
    def to_dict(self):
        return self._info
    @property
    def info(self):
        return self._info
    def __getitem__(self, item):
        return self._info.get(item)
    def get(self, item, default=None):
        return self._info.get(item, default)
    def pop(self, key, default_value=None):
        return self.info.pop(key, default_value)
class FakeResponse(requests.Response):
    def __init__(self, headers=None, status_code=200,
                 data=None, encoding=None):
        super(FakeResponse, self).__init__()
        headers = headers or {}
        self.status_code = status_code
        self.headers.update(headers)
        self._content = json.dumps(data)
        if not isinstance(self._content, bytes):
            self._content = self._content.encode()
class FakeModel(dict):
    def __getattr__(self, key):
        try:
            return self[key]
        except KeyError:
            raise AttributeError(key)
Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

fakes.py

Latest commit

History

fakes.py

File metadata and controls
