def test_get_middleware_by_uri(self):
uri = '%s://' % TestUnregisteredMiddleware.__protocol__
self.assertRaises(
Middleware.Error, Middleware.get_middleware_by_uri, uri)
uri = '%s://' % TestRegisteredMiddleware.__protocol__
middleware = Middleware.get_middleware_by_uri(uri)
self.assertEqual(type(middleware), TestRegisteredMiddleware)
self.assertEqual(middleware.data_scope, DEFAULT_DATA_SCOPE)
uri = '%s%s%s://' % (
TestRegisteredWithDataTypeMiddleware.__protocol__,
SCHEME_SEPARATOR,
TestRegisteredWithDataTypeMiddleware.__datatype__)
middleware = Middleware.get_middleware_by_uri(uri)
self.assertEqual(
type(middleware), TestRegisteredWithDataTypeMiddleware)
self.assertEqual(middleware.data_scope, DEFAULT_DATA_SCOPE)
self.assertEqual(middleware.data_type,
TestRegisteredWithDataTypeMiddleware.__datatype__)
data_scope = 'test'
uri = '%s%s%s%s%s://' % (
TestRegisteredWithDataTypeMiddleware.__protocol__,
SCHEME_SEPARATOR,
TestRegisteredWithDataTypeMiddleware.__datatype__,
SCHEME_SEPARATOR,
data_scope)
middleware = Middleware.get_middleware_by_uri(uri)
self.assertEqual(
type(middleware), TestRegisteredWithDataTypeMiddleware)
self.assertEqual(middleware.data_scope, data_scope)
self.assertEqual(middleware.data_type,
TestRegisteredWithDataTypeMiddleware.__datatype__)
def test_resolve_middleware_by_uri(self):
uri = '%s://' % TestUnregisteredMiddleware.__protocol__
self.assertRaises(
Middleware.Error, Middleware.resolve_middleware_by_uri, uri)
uri = '%s://' % TestRegisteredMiddleware.__protocol__
middleware_class = Middleware.resolve_middleware_by_uri(uri)
self.assertEqual(middleware_class, TestRegisteredMiddleware)
uri = '%s%s%s://' % (
TestRegisteredWithDataTypeMiddleware.__protocol__,
SCHEME_SEPARATOR,
TestRegisteredWithDataTypeMiddleware.__datatype__)
middleware_class = Middleware.resolve_middleware_by_uri(uri)
self.assertEqual(
middleware_class, TestRegisteredWithDataTypeMiddleware)
def test_resolve_middleware(self):
protocol = TestUnregisteredMiddleware.__protocol__
self.assertRaises(
Middleware.Error, Middleware.resolve_middleware, protocol=protocol)
protocol = TestRegisteredMiddleware.__protocol__
middleware_class = Middleware.resolve_middleware(protocol=protocol)
self.assertEqual(middleware_class, TestRegisteredMiddleware)
data_type = None
middleware_class = Middleware.resolve_middleware(
protocol=protocol, data_type=data_type)
self.assertEqual(middleware_class, TestRegisteredMiddleware)
data_type = TestRegisteredWithDataTypeMiddleware.__datatype__
middleware_class = Middleware.resolve_middleware(
protocol=protocol, data_type=data_type)
self.assertEqual(
middleware_class, TestRegisteredWithDataTypeMiddleware)
def test_get_middleware(self):
protocol = TestUnregisteredMiddleware.__protocol__
self.assertRaises(
Middleware.Error, Middleware.get_middleware, protocol=protocol)
protocol = TestRegisteredMiddleware.__protocol__
middleware = Middleware.get_middleware(protocol=protocol)
self.assertEqual(type(middleware), TestRegisteredMiddleware)
self.assertEqual(middleware.data_scope, DEFAULT_DATA_SCOPE)
data_type = None
middleware = Middleware.get_middleware(
protocol=protocol, data_type=data_type)
self.assertEqual(type(middleware), TestRegisteredMiddleware)
self.assertEqual(middleware.data_scope, DEFAULT_DATA_SCOPE)
data_scope = 'test'
data_type = TestRegisteredWithDataTypeMiddleware.__datatype__
middleware = Middleware.get_middleware(
protocol=protocol, data_type=data_type, data_scope=data_scope)
self.assertEqual(middleware.data_scope, data_scope)
self.assertEqual(
type(middleware), TestRegisteredWithDataTypeMiddleware)
