def test_default_service(self):
"""Testing AvatarServiceRegistry.default_service"""
registry = AvatarServiceRegistry()
registry.register(DummyAvatarService)
registry.set_enabled_services([DummyAvatarService, GravatarService])
self.assertIsNone(registry.default_service)
# Set the default backend to the dummy backend.
registry.set_default_service(DummyAvatarService)
self.assertIsInstance(registry.default_service, DummyAvatarService)
siteconfig = SiteConfiguration.objects.get_current()
self.assertEqual(
siteconfig.get(AvatarServiceRegistry.DEFAULT_SERVICE_KEY),
DummyAvatarService.avatar_service_id)
# Set the default backend to the Gravatar backend.
registry.set_default_service(GravatarService)
self.assertIsInstance(registry.default_service, GravatarService)
siteconfig = SiteConfiguration.objects.get_current()
self.assertEqual(
siteconfig.get(AvatarServiceRegistry.DEFAULT_SERVICE_KEY),
GravatarService.avatar_service_id)
# Remove the default backend.
registry.set_default_service(None)
self.assertIsNone(registry.default_service)
siteconfig = SiteConfiguration.objects.get_current()
self.assertIsNone(
siteconfig.get(AvatarServiceRegistry.DEFAULT_SERVICE_KEY))
def test_default_service(self):
"""Testing AvatarServiceRegistry.default_service"""
registry = AvatarServiceRegistry()
registry.register(DummyAvatarService)
registry.set_enabled_services([DummyAvatarService, GravatarService])
self.assertIsNone(registry.default_service)
# Set the default backend to the dummy backend.
registry.set_default_service(DummyAvatarService)
self.assertIsInstance(registry.default_service, DummyAvatarService)
siteconfig = SiteConfiguration.objects.get_current()
self.assertEqual(
siteconfig.get(AvatarServiceRegistry.DEFAULT_SERVICE_KEY),
DummyAvatarService.avatar_service_id)
# Set the default backend to the Gravatar backend.
registry.set_default_service(GravatarService)
self.assertIsInstance(registry.default_service, GravatarService)
siteconfig = SiteConfiguration.objects.get_current()
self.assertEqual(
siteconfig.get(AvatarServiceRegistry.DEFAULT_SERVICE_KEY),
GravatarService.avatar_service_id)
# Remove the default backend.
registry.set_default_service(None)
self.assertIsNone(registry.default_service)
siteconfig = SiteConfiguration.objects.get_current()
self.assertIsNone(
siteconfig.get(AvatarServiceRegistry.DEFAULT_SERVICE_KEY))
def test_set_enabled_services_invalid_service(self):
"""Testing AvatarServiceRegistry.set_enabled_services with an
unregistered service
"""
registry = AvatarServiceRegistry()
with self.assertRaises(ItemLookupError):
registry.set_enabled_services(
[DummyAvatarService, GravatarService])
self.assertEqual(registry.enabled_services, set())
siteconfig = SiteConfiguration.objects.get_current()
self.assertEqual(
siteconfig.get(AvatarServiceRegistry.ENABLED_SERVICES_KEY), [])
def test_disable_default_from_setter(self):
"""Testing AvatarServiceRegistry.set_enabled_services unsets the
default services when removed
"""
registry = AvatarServiceRegistry()
registry.enable_service(GravatarService)
registry.set_default_service(GravatarService)
registry.set_enabled_services([])
self.assertIsNone(registry.default_service)
siteconfig = SiteConfiguration.objects.get_current()
self.assertEqual(
siteconfig.get(AvatarServiceRegistry.ENABLED_SERVICES_KEY), [])
self.assertIsNone(
siteconfig.get(AvatarServiceRegistry.DEFAULT_SERVICE_KEY))
def test_set_enabled_services_invalid_service(self):
"""Testing AvatarServiceRegistry.set_enabled_services with an
unregistered service
"""
registry = AvatarServiceRegistry()
with self.assertRaises(ItemLookupError):
registry.set_enabled_services([DummyAvatarService,
GravatarService])
self.assertEqual(registry.enabled_services, set())
siteconfig = SiteConfiguration.objects.get_current()
self.assertEqual(
siteconfig.get(AvatarServiceRegistry.ENABLED_SERVICES_KEY),
[])
def test_set_enabled_services(self):
"""Testing AvatarServiceRegistry.set_enabled_services"""
registry = AvatarServiceRegistry()
registry.register(DummyAvatarService)
registry.set_enabled_services([DummyAvatarService, GravatarService])
self.assertEqual(registry.enabled_services,
{DummyAvatarService, GravatarService})
siteconfig = SiteConfiguration.objects.get_current()
self.assertEqual(
set(siteconfig.get(AvatarServiceRegistry.ENABLED_SERVICES_KEY)), {
DummyAvatarService.avatar_service_id,
GravatarService.avatar_service_id,
})
def test_disable_default_from_setter(self):
"""Testing AvatarServiceRegistry.set_enabled_services unsets the
default services when removed
"""
registry = AvatarServiceRegistry()
registry.enable_service(GravatarService)
registry.set_default_service(GravatarService)
registry.set_enabled_services([])
self.assertIsNone(registry.default_service)
siteconfig = SiteConfiguration.objects.get_current()
self.assertEqual(
siteconfig.get(AvatarServiceRegistry.ENABLED_SERVICES_KEY),
[])
self.assertIsNone(
siteconfig.get(AvatarServiceRegistry.DEFAULT_SERVICE_KEY))
def test_set_enabled_services(self):
"""Testing AvatarServiceRegistry.set_enabled_services"""
registry = AvatarServiceRegistry()
registry.register(DummyAvatarService)
registry.set_enabled_services([DummyAvatarService, GravatarService])
self.assertEqual(registry.enabled_services,
{DummyAvatarService, GravatarService})
siteconfig = SiteConfiguration.objects.get_current()
self.assertEqual(
set(siteconfig.get(AvatarServiceRegistry.ENABLED_SERVICES_KEY)),
{
DummyAvatarService.avatar_service_id,
GravatarService.avatar_service_id,
})
def test_set_enabled_services_with_save_false(self):
"""Testing AvatarServiceRegistry.set_enabled_services with save=False
"""
registry = AvatarServiceRegistry()
registry.register(DummyAvatarService)
registry.set_enabled_services([DummyAvatarService, GravatarService],
save=False)
self.assertEqual(registry.enabled_services,
{DummyAvatarService, GravatarService})
# Make sure we've saved this in our cached siteconfig...
siteconfig = SiteConfiguration.objects.get_current()
self.assertEqual(
set(siteconfig.get(AvatarServiceRegistry.ENABLED_SERVICES_KEY)), {
DummyAvatarService.avatar_service_id,
GravatarService.avatar_service_id
})
# ... but not in the database.
siteconfig = SiteConfiguration.objects.get(pk=siteconfig.pk)
self.assertEqual(
siteconfig.get(AvatarServiceRegistry.ENABLED_SERVICES_KEY), [])
def test_set_enabled_services_with_save_false(self):
"""Testing AvatarServiceRegistry.set_enabled_services with save=False
"""
registry = AvatarServiceRegistry()
registry.register(DummyAvatarService)
registry.set_enabled_services([DummyAvatarService, GravatarService],
save=False)
self.assertEqual(registry.enabled_services,
{DummyAvatarService, GravatarService})
# Make sure we've saved this in our cached siteconfig...
siteconfig = SiteConfiguration.objects.get_current()
self.assertEqual(
set(siteconfig.get(AvatarServiceRegistry.ENABLED_SERVICES_KEY)),
{DummyAvatarService.avatar_service_id,
GravatarService.avatar_service_id})
# ... but not in the database.
siteconfig = SiteConfiguration.objects.get(pk=siteconfig.pk)
self.assertEqual(
siteconfig.get(AvatarServiceRegistry.ENABLED_SERVICES_KEY),
[])
