78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
import unittest
|
|
|
|
import django.utils.copycompat as copy
|
|
from django.utils.functional import SimpleLazyObject
|
|
|
|
class _ComplexObject(object):
|
|
def __init__(self, name):
|
|
self.name = name
|
|
|
|
def __eq__(self, other):
|
|
return self.name == other.name
|
|
|
|
def __hash__(self):
|
|
return hash(self.name)
|
|
|
|
def __str__(self):
|
|
return "I am _ComplexObject(%r)" % self.name
|
|
|
|
def __unicode__(self):
|
|
return unicode(self.name)
|
|
|
|
def __repr__(self):
|
|
return "_ComplexObject(%r)" % self.name
|
|
|
|
complex_object = lambda: _ComplexObject("joe")
|
|
|
|
class TestUtilsSimpleLazyObject(unittest.TestCase):
|
|
"""
|
|
Tests for SimpleLazyObject
|
|
"""
|
|
# Note that concrete use cases for SimpleLazyObject are also found in the
|
|
# auth context processor tests (unless the implementation of that function
|
|
# is changed).
|
|
|
|
def test_equality(self):
|
|
self.assertEqual(complex_object(), SimpleLazyObject(complex_object))
|
|
self.assertEqual(SimpleLazyObject(complex_object), complex_object())
|
|
|
|
def test_hash(self):
|
|
# hash() equality would not be true for many objects, but it should be
|
|
# for _ComplexObject
|
|
self.assertEqual(hash(complex_object()),
|
|
hash(SimpleLazyObject(complex_object)))
|
|
|
|
def test_repr(self):
|
|
# For debugging, it will really confuse things if there is no clue that
|
|
# SimpleLazyObject is actually a proxy object. So we don't
|
|
# proxy __repr__
|
|
self.assertTrue("SimpleLazyObject" in repr(SimpleLazyObject(complex_object)))
|
|
|
|
def test_str(self):
|
|
self.assertEqual("I am _ComplexObject('joe')", str(SimpleLazyObject(complex_object)))
|
|
|
|
def test_unicode(self):
|
|
self.assertEqual(u"joe", unicode(SimpleLazyObject(complex_object)))
|
|
|
|
def test_class(self):
|
|
# This is important for classes that use __class__ in things like
|
|
# equality tests.
|
|
self.assertEqual(_ComplexObject, SimpleLazyObject(complex_object).__class__)
|
|
|
|
def test_deepcopy(self):
|
|
# Check that we *can* do deep copy, and that it returns the right
|
|
# objects.
|
|
|
|
# First, for an unevaluated SimpleLazyObject
|
|
s = SimpleLazyObject(complex_object)
|
|
assert s._wrapped is None
|
|
s2 = copy.deepcopy(s)
|
|
assert s._wrapped is None # something has gone wrong is s is evaluated
|
|
self.assertEqual(s2, complex_object())
|
|
|
|
# Second, for an evaluated SimpleLazyObject
|
|
name = s.name # evaluate
|
|
assert s._wrapped is not None
|
|
s3 = copy.deepcopy(s)
|
|
self.assertEqual(s3, complex_object())
|