142c7ae0d6
Adds the storage interface for the shards storage controller. This controller is meant to be used with the shard management queues admin API interface. The idea mirrors that used in writing the proxy's partition manager. This patch also introduces the distinction between data and control storage drivers. Data storage drivers are those that control core functionality: messages, queues, claims. Control storage drivers are used to manage system functionality from the point of view of an admin - in this case, the registry of shards. The next patch will provide an implementation for mongodb + unit tests. Change-Id: I4f24e84c99689968e60360383b190ce168055e74 Partially-implements: blueprint storage-sharding Partially-Closes: 1241686
54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
# Copyright (c) 2013 Rackspace, 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.
|
|
|
|
from marconi.common import exceptions
|
|
from marconi.queues import bootstrap
|
|
from marconi.queues.storage import pipeline
|
|
from marconi.queues.storage import sharding
|
|
from marconi.queues.storage import sqlite
|
|
from marconi.queues.transport import wsgi
|
|
from marconi.tests import base
|
|
|
|
|
|
class TestBootstrap(base.TestBase):
|
|
|
|
def _bootstrap(self, conf_file):
|
|
conf = self.load_conf(conf_file)
|
|
return bootstrap.Bootstrap(conf)
|
|
|
|
def test_storage_invalid(self):
|
|
boot = self._bootstrap('etc/drivers_storage_invalid.conf')
|
|
self.assertRaises(exceptions.InvalidDriver,
|
|
lambda: boot.storage)
|
|
|
|
def test_storage_sqlite(self):
|
|
bootstrap = self._bootstrap('etc/wsgi_sqlite.conf')
|
|
self.assertIsInstance(bootstrap.storage, pipeline.DataDriver)
|
|
self.assertIsInstance(bootstrap.storage._storage, sqlite.DataDriver)
|
|
|
|
def test_storage_sqlite_sharded(self):
|
|
"""Makes sure we can load the shard driver."""
|
|
bootstrap = self._bootstrap('etc/wsgi_sqlite_sharded.conf')
|
|
self.assertIsInstance(bootstrap.storage._storage, sharding.DataDriver)
|
|
|
|
def test_transport_invalid(self):
|
|
boot = self._bootstrap('etc/drivers_transport_invalid.conf')
|
|
self.assertRaises(exceptions.InvalidDriver,
|
|
lambda: boot.transport)
|
|
|
|
def test_transport_wsgi(self):
|
|
bootstrap = self._bootstrap('etc/wsgi_sqlite.conf')
|
|
self.assertIsInstance(bootstrap.transport, wsgi.Driver)
|