def testGetOffsetLink(self):
mock_entity = MockEntity(6, 16)
mock_entity.layer.auto_managed = False
self.mox.StubOutWithMock(pages.template.django.template, 'resolve_variable')
self.mox.StubOutWithMock(entities, '_GetEntitySiblings')
self.mox.StubOutWithMock(entities, '_GetFlyToLink')
dummy_siblings_list = [5, 6, 7, 8, 9, 10]
pages.template.django.template.resolve_variable(
'_cache', 'dummy-context').AndReturn('dummy-cache')
pages.template.django.template.resolve_variable(
'_link_template', 'dummy-context').AndReturn('dummy-template')
entities._GetEntitySiblings(mock_entity, 'dummy-cache').AndReturn(
dummy_siblings_list)
entities._GetFlyToLink(8, 'balloon', 'dummy-template').AndReturn(
'dummy-link1')
pages.template.django.template.resolve_variable(
'_cache', 'dummy-context').AndReturn('dummy-cache')
pages.template.django.template.resolve_variable(
'_link_template', 'dummy-context').AndReturn('dummy-template')
entities._GetEntitySiblings(mock_entity, 'dummy-cache').AndReturn(
dummy_siblings_list)
entities._GetFlyToLink(10, 'balloon', 'dummy-template').AndReturn(
'dummy-link2')
self.mox.ReplayAll()
link = entities._GetOffsetLink('dummy-context', mock_entity, 2, 'balloon')
self.assertEqual(link, 'dummy-link1')
link = entities._GetOffsetLink('dummy-context', mock_entity, -2, 'balloon')
self.assertEqual(link, 'dummy-link2')
mock_entity.layer.auto_managed = True
self.assertRaises(ValueError, entities._GetOffsetLink,
'dummy-context', mock_entity, -2, 'balloon')
def testGetFlyToLinkWithGoodData(self): self.assertEqual(entities._GetFlyToLink(42, 'abc'), '#id42;abc') self.assertEqual(entities._GetFlyToLink(42, 'abc', 'def'), 'def') link_template = 'a' + settings.BALLOON_LINK_PLACEHOLDER + 'b' self.assertEqual(entities._GetFlyToLink(42, 'abc', link_template), 'aid42b')
