def test_get_service(self):
""" get service """
class IFoo(Interface):
pass
@provides(IFoo)
class Foo(HasTraits):
pass
# Register a couple of services.
foo = Foo()
self.service_registry.register_service(IFoo, foo)
goo = Foo()
self.service_registry.register_service(IFoo, goo)
# Look up one of them!
service = self.service_registry.get_service(IFoo)
self.assertTrue(foo is service or goo is service)
class IBar(Interface):
pass
# Lookup a non-existent service.
service = self.service_registry.get_service(IBar)
self.assertEqual(None, service)
def test_minimize_and_maximize(self):
""" minimize and maximize """
class IFoo(Interface):
price = Int
@provides(IFoo)
class Foo(HasTraits):
price = Int
# Register some objects with various prices.
x = Foo(price=10)
y = Foo(price=5)
z = Foo(price=100)
for foo in [x, y, z]:
self.service_registry.register_service(IFoo, foo)
# Find the service with the lowest price.
service = self.service_registry.get_service(IFoo, minimize='price')
self.assertNotEqual(None, service)
self.assertEqual(Foo, type(service))
self.assertEqual(y, service)
# Find the service with the highest price.
service = self.service_registry.get_service(IFoo, maximize='price')
self.assertNotEqual(None, service)
self.assertEqual(Foo, type(service))
self.assertEqual(z, service)
def test_get_services(self):
""" get services """
class IFoo(Interface):
pass
@provides(IFoo)
class Foo(HasTraits):
pass
# Register two services.
foo = Foo()
self.service_registry.register_service(IFoo, foo)
foo = Foo()
self.service_registry.register_service(IFoo, foo)
# Look it up again.
services = self.service_registry.get_services(IFoo)
self.assertEqual(2, len(services))
class IBar(Interface):
pass
# Lookup a non-existent service.
services = self.service_registry.get_services(IBar)
self.assertEqual([], services)
def test_unregister_service(self):
""" unregister service """
class IFoo(Interface):
price = Int
class Foo(HasTraits):
implements(IFoo)
price = Int
# Register two services.
#
# This one shows how the object's attributes are used when evaluating
# a query.
foo = Foo(price=100)
foo_id = self.service_registry.register_service(IFoo, foo)
# This one shows how properties can be specified that *take precedence*
# over the object's attributes when evaluating a query.
goo = Foo(price=10)
goo_id = self.service_registry.register_service(
IFoo, goo, {'price' : 200}
)
# Create a query that doesn't match any registered object.
service = self.service_registry.get_service(IFoo, 'price < 100')
self.assertEqual(None, service)
# Create a query that matches one of the registered objects.
service = self.service_registry.get_service(IFoo, 'price <= 100')
self.assertEqual(foo, service)
# Create a query that matches both registered objects.
service = self.service_registry.get_service(IFoo, 'price >= 100')
self.assert_(foo is service or goo is service)
#### Now do some unregistering! ####
# Unregister 'foo'.
self.service_registry.unregister_service(foo_id)
# This query should no longer match any of the registered objects.
service = self.service_registry.get_service(IFoo, 'price <= 100')
self.assertEqual(None, service)
# Unregister 'goo'.
self.service_registry.unregister_service(goo_id)
# This query should no longer match any of the registered objects.
service = self.service_registry.get_service(IFoo, 'price >= 100')
self.assertEqual(None, service)
# Try to unregister a non-existent service.
self.failUnlessRaises(
ValueError, self.service_registry.unregister_service, -1
)
return
def test_get_and_set_service_properties(self):
""" get and set service properties """
class IFoo(Interface):
price = Int
class Foo(HasTraits):
implements(IFoo)
price = Int
# Register two services.
#
# This one has no properties.
foo = Foo(price=100)
foo_id = self.service_registry.register_service(IFoo, foo)
# This one has properties.
goo = Foo(price=10)
goo_id = self.service_registry.register_service(
IFoo, goo, {'price' : 200}
)
# Get the properties.
foo_properties = self.service_registry.get_service_properties(foo_id)
self.assertEqual({}, foo_properties)
goo_properties = self.service_registry.get_service_properties(goo_id)
self.assertEqual(200, goo_properties['price'])
# Update the properties.
foo_properties['price'] = 300
goo_properties['price'] = 500
# Set the properties.
self.service_registry.set_service_properties(foo_id, foo_properties)
self.service_registry.set_service_properties(goo_id, goo_properties)
# Get the properties again.
foo_properties = self.service_registry.get_service_properties(foo_id)
self.assertEqual(300, foo_properties['price'])
goo_properties = self.service_registry.get_service_properties(goo_id)
self.assertEqual(500, goo_properties['price'])
# Try to get the properties of a non-existent service.
self.failUnlessRaises(
ValueError, self.service_registry.get_service_properties, -1
)
# Try to set the properties of a non-existent service.
self.failUnlessRaises(
ValueError, self.service_registry.set_service_properties, -1, {}
)
return
def test_get_and_set_service_properties(self):
""" get and set service properties """
class IFoo(Interface):
price = Int
@provides(IFoo)
class Foo(HasTraits):
price = Int
# Register two services.
#
# This one has no properties.
foo = Foo(price=100)
foo_id = self.service_registry.register_service(IFoo, foo)
# This one has properties.
goo = Foo(price=10)
goo_id = self.service_registry.register_service(
IFoo, goo, {"price": 200}
)
# Get the properties.
foo_properties = self.service_registry.get_service_properties(foo_id)
self.assertEqual({}, foo_properties)
goo_properties = self.service_registry.get_service_properties(goo_id)
self.assertEqual(200, goo_properties["price"])
# Update the properties.
foo_properties["price"] = 300
goo_properties["price"] = 500
# Set the properties.
self.service_registry.set_service_properties(foo_id, foo_properties)
self.service_registry.set_service_properties(goo_id, goo_properties)
# Get the properties again.
foo_properties = self.service_registry.get_service_properties(foo_id)
self.assertEqual(300, foo_properties["price"])
goo_properties = self.service_registry.get_service_properties(goo_id)
self.assertEqual(500, goo_properties["price"])
# Try to get the properties of a non-existent service.
with self.assertRaises(ValueError):
self.service_registry.get_service_properties(-1)
# Try to set the properties of a non-existent service.
with self.assertRaises(ValueError):
self.service_registry.set_service_properties(-1, {})
def test_get_services_with_strings(self):
""" get services with strings """
from envisage.tests.foo import Foo
# Register a couple of services using a string protocol name.
protocol_name = 'envisage.tests.foo.IFoo'
self.service_registry.register_service(protocol_name, Foo())
self.service_registry.register_service(protocol_name, Foo())
# Look them up using the same string!
services = self.service_registry.get_services(protocol_name)
self.assertEqual(2, len(services))
def foo_factory(self, **properties):
""" A factory for foos. """
from envisage.tests.foo import Foo
self.foo = Foo()
return self.foo
def test_get_services_with_query(self):
""" get services with query """
class IFoo(Interface):
price = Int
class Foo(HasTraits):
implements(IFoo)
price = Int
# Register two services.
#
# This one shows how the object's attributes are used when evaluating
# a query.
foo = Foo(price=100)
self.service_registry.register_service(IFoo, foo)
# This one shows how properties can be specified that *take precedence*
# over the object's attributes when evaluating a query.
goo = Foo(price=10)
self.service_registry.register_service(IFoo, goo, {'price' : 200})
# Create a query that doesn't match any registered object.
services = self.service_registry.get_services(IFoo, 'color == "red"')
self.assertEqual([], services)
# Create a query that matches one of the registered objects.
services = self.service_registry.get_services(IFoo, 'price <= 100')
self.assertEqual([foo], services)
# Create a query that matches both registered objects.
services = self.service_registry.get_services(IFoo, 'price >= 100')
self.assert_(foo in services)
self.assert_(goo in services)
self.assertEqual(2, len(services))
class IBar(Interface):
pass
# Lookup a non-existent service.
services = self.service_registry.get_services(IBar, 'price <= 100')
self.assertEqual([], services)
return
def test_should_get_required_service(self):
class Foo(HasTraits):
price = Int
foo = Foo()
# Register a service factory.
self.service_registry.register_service(Foo, foo)
service = self.service_registry.get_required_service(Foo)
self.assertIs(foo, service)
def test_get_service_with_query(self):
""" get service with query """
class IFoo(Interface):
price = Int
@provides(IFoo)
class Foo(HasTraits):
price = Int
# Register two services.
#
# This one shows how the object's attributes are used when evaluating
# a query.
foo = Foo(price=100)
self.service_registry.register_service(IFoo, foo)
# This one shows how properties can be specified that *take precedence*
# over the object's attributes when evaluating a query.
goo = Foo(price=10)
self.service_registry.register_service(IFoo, goo, {"price": 200})
# Create a query that doesn't match any registered object.
service = self.service_registry.get_service(IFoo, "price < 100")
self.assertEqual(None, service)
# Create a query that matches one of the registered objects.
service = self.service_registry.get_service(IFoo, "price <= 100")
self.assertEqual(foo, service)
# Create a query that matches both registered objects.
service = self.service_registry.get_service(IFoo, "price >= 100")
self.assertTrue(foo is service or goo is service)
class IBar(Interface):
pass
# Lookup a non-existent service.
service = self.service_registry.get_service(IBar, "price <= 100")
self.assertEqual(None, service)
def foo_factory(**properties):
""" A factory for foos. """
return Foo(**properties)
