zaqar/tests/unit/test_bootstrap.py
Flavio Percoco a828ec96cd Isolate tests a bit more
This patch isolates mongodb tests and adds new `shortcut-methods` to
base clases in order to make it easier for subclasses to access conf
instances, override them or even hook them.

Change-Id: I8fe05bf383198244616499adb60480859f1ec7c6
2013-11-21 14:38:11 +01:00

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 errors
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):
self.conf = self.load_conf(conf_file)
return bootstrap.Bootstrap(self.conf)
def test_storage_invalid(self):
bootstrap = self._bootstrap('drivers_storage_invalid.conf')
self.assertRaises(errors.InvalidDriver,
lambda: bootstrap.storage)
def test_storage_sqlite(self):
bootstrap = self._bootstrap('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('wsgi_sqlite_sharded.conf')
self.assertIsInstance(bootstrap.storage._storage, sharding.DataDriver)
def test_transport_invalid(self):
bootstrap = self._bootstrap('drivers_transport_invalid.conf')
self.assertRaises(errors.InvalidDriver,
lambda: bootstrap.transport)
def test_transport_wsgi(self):
bootstrap = self._bootstrap('wsgi_sqlite.conf')
self.assertIsInstance(bootstrap.transport, wsgi.Driver)