75339f4712
This patch adds a policy, 'download_image', to be enforced when image data is retrieved. It also does some basic refactoring of how policies are enforced. Fixes bug 1038086 Change-Id: Idd844b615d362eae3197e106067c29dba8e3eeda
86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2012 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 json
|
|
import os
|
|
import shutil
|
|
|
|
import stubout
|
|
|
|
from glance.openstack.common import cfg
|
|
from glance import store
|
|
# NOTE(ameade): this import is necessary. Since we override a cfg opt it
|
|
# registers we must have that opt loaded.
|
|
from glance.store import filesystem
|
|
from glance.store import location
|
|
from glance.tests import stubs
|
|
from glance.tests import utils as test_utils
|
|
|
|
CONF = cfg.CONF
|
|
|
|
|
|
class StoreClearingUnitTest(test_utils.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(StoreClearingUnitTest, self).setUp()
|
|
# Ensure stores + locations cleared
|
|
location.SCHEME_TO_CLS_MAP = {}
|
|
store.create_stores()
|
|
|
|
def tearDown(self):
|
|
super(StoreClearingUnitTest, self).tearDown()
|
|
# Ensure stores + locations cleared
|
|
location.SCHEME_TO_CLS_MAP = {}
|
|
|
|
|
|
class IsolatedUnitTest(StoreClearingUnitTest):
|
|
|
|
"""
|
|
Unit test case that establishes a mock environment within
|
|
a testing directory (in isolation)
|
|
"""
|
|
|
|
def setUp(self):
|
|
self.test_id, self.test_dir = test_utils.get_isolated_test_env()
|
|
self.stubs = stubout.StubOutForTesting()
|
|
policy_file = self._copy_data_file('policy.json', self.test_dir)
|
|
self.config(sql_connection='sqlite://',
|
|
verbose=False,
|
|
debug=False,
|
|
default_store='filesystem',
|
|
filesystem_store_datadir=os.path.join(self.test_dir),
|
|
policy_file=policy_file)
|
|
super(IsolatedUnitTest, self).setUp()
|
|
stubs.stub_out_registry_and_store_server(self.stubs, self.test_dir)
|
|
|
|
def _copy_data_file(self, file_name, dst_dir):
|
|
src_file_name = os.path.join('glance/tests/etc', file_name)
|
|
shutil.copy(src_file_name, dst_dir)
|
|
dst_file_name = os.path.join(dst_dir, file_name)
|
|
return dst_file_name
|
|
|
|
def set_policy_rules(self, rules):
|
|
fap = open(CONF.policy_file, 'w')
|
|
fap.write(json.dumps(rules))
|
|
fap.close()
|
|
|
|
def tearDown(self):
|
|
super(IsolatedUnitTest, self).tearDown()
|
|
self.stubs.UnsetAll()
|
|
if os.path.exists(self.test_dir):
|
|
shutil.rmtree(self.test_dir)
|