class TrackObjectTestCase(unittest.TestCase):
def setUp(self):
self.tracker = ClassTracker()
def tearDown(self):
self.tracker.detach_all()
def test_track_object(self):
"""Test object registration.
"""
foo = Foo()
bar = Bar()
self.tracker.track_object(foo)
self.tracker.track_object(bar)
self.assert_(id(foo) in self.tracker.objects)
self.assert_(id(bar) in self.tracker.objects)
self.assert_('Foo' in self.tracker.index)
self.assert_('Bar' in self.tracker.index)
self.assertEqual(self.tracker.objects[id(foo)].ref(),foo)
self.assertEqual(self.tracker.objects[id(bar)].ref(),bar)
def test_type_errors(self):
"""Test intrackable objects.
"""
i = 42
j = 'Foobar'
k = [i,j]
l = {i: j}
self.assertRaises(TypeError, self.tracker.track_object, i)
self.assertRaises(TypeError, self.tracker.track_object, j)
self.assertRaises(TypeError, self.tracker.track_object, k)
self.assertRaises(TypeError, self.tracker.track_object, l)
self.assert_(id(i) not in self.tracker.objects)
self.assert_(id(j) not in self.tracker.objects)
self.assert_(id(k) not in self.tracker.objects)
self.assert_(id(l) not in self.tracker.objects)
def test_track_by_name(self):
"""Test registering objects by name.
"""
foo = Foo()
self.tracker.track_object(foo, name='Foobar')
self.assert_('Foobar' in self.tracker.index )
self.assertEqual(self.tracker.index['Foobar'][0].ref(),foo)
def test_keep(self):
"""Test lifetime of tracked objects.
"""
foo = Foo()
bar = Bar()
self.tracker.track_object(foo, keep=1)
self.tracker.track_object(bar)
idfoo = id(foo)
idbar = id(bar)
del foo
del bar
self.assert_(self.tracker.objects[idfoo].ref() is not None)
self.assert_(self.tracker.objects[idbar].ref() is None)
def test_mixed_tracking(self):
"""Test mixed instance and class tracking.
"""
foo = Foo()
self.tracker.track_object(foo)
self.tracker.create_snapshot()
self.tracker.track_class(Foo)
objs = []
for _ in range(10):
objs.append(Foo())
self.tracker.create_snapshot()
def test_recurse(self):
"""Test recursive sizing and saving of referents.
"""
foo = Foo()
self.tracker.track_object(foo, resolution_level=1)
self.tracker.create_snapshot()
fp = self.tracker.objects[id(foo)].snapshots[-1]
refs = fp[1].refs
dref = [r for r in refs if r.name == '__dict__']
self.assertEqual(len(dref),1)
dref = dref[0]
self.assert_(dref.size > 0, dref.size)
self.assert_(dref.flat > 0, dref.flat)
self.assertEqual(dref.refs,())
# Test track_change and more fine-grained resolution
self.tracker.track_change(foo, resolution_level=2)
self.tracker.create_snapshot()
fp = self.tracker.objects[id(foo)].snapshots[-1]
refs = fp[1].refs
dref = [r for r in refs if r.name == '__dict__']
self.assertEqual(len(dref),1)
dref = dref[0]
namerefs = [r.name for r in dref.refs]
self.assert_('[K] foo' in namerefs, namerefs)
self.assert_("[V] foo: 'foo'" in namerefs, namerefs)
class TrackObjectTestCase(unittest.TestCase):
def setUp(self):
self.tracker = ClassTracker()
def tearDown(self):
self.tracker.detach_all()
def test_track_object(self):
"""Test object registration.
"""
foo = Foo()
bar = Bar()
self.tracker.track_object(foo)
self.tracker.track_object(bar)
self.assertTrue(id(foo) in self.tracker.objects)
self.assertTrue(id(bar) in self.tracker.objects)
self.assertTrue('Foo' in self.tracker.index)
self.assertTrue('Bar' in self.tracker.index)
self.assertEqual(self.tracker.objects[id(foo)].ref(), foo)
self.assertEqual(self.tracker.objects[id(bar)].ref(), bar)
def test_type_errors(self):
"""Test intrackable objects.
"""
i = 42
j = 'Foobar'
k = [i, j]
l = {i: j}
self.assertRaises(TypeError, self.tracker.track_object, i)
self.assertRaises(TypeError, self.tracker.track_object, j)
self.assertRaises(TypeError, self.tracker.track_object, k)
self.assertRaises(TypeError, self.tracker.track_object, l)
self.assertTrue(id(i) not in self.tracker.objects)
self.assertTrue(id(j) not in self.tracker.objects)
self.assertTrue(id(k) not in self.tracker.objects)
self.assertTrue(id(l) not in self.tracker.objects)
def test_track_by_name(self):
"""Test registering objects by name.
"""
foo = Foo()
self.tracker.track_object(foo, name='Foobar')
self.assertTrue('Foobar' in self.tracker.index)
self.assertEqual(self.tracker.index['Foobar'][0].ref(), foo)
def test_keep(self):
"""Test lifetime of tracked objects.
"""
foo = Foo()
bar = Bar()
self.tracker.track_object(foo, keep=1)
self.tracker.track_object(bar)
idfoo = id(foo)
idbar = id(bar)
del foo
del bar
self.assertTrue(self.tracker.objects[idfoo].ref() is not None)
self.assertTrue(self.tracker.objects[idbar].ref() is None)
def test_mixed_tracking(self):
"""Test mixed instance and class tracking.
"""
foo = Foo()
self.tracker.track_object(foo)
self.tracker.create_snapshot()
self.tracker.track_class(Foo)
objs = []
for _ in range(10):
objs.append(Foo())
self.tracker.create_snapshot()
def test_recurse(self):
"""Test recursive sizing and saving of referents.
"""
foo = Foo()
self.tracker.track_object(foo, resolution_level=1)
self.tracker.create_snapshot()
fp = self.tracker.objects[id(foo)].snapshots[-1]
refs = fp[1].refs
dref = [r for r in refs if r.name == '__dict__']
self.assertEqual(len(dref), 1)
dref = dref[0]
self.assertTrue(dref.size > 0, dref.size)
self.assertTrue(dref.flat > 0, dref.flat)
self.assertEqual(dref.refs, ())
# Test track_change and more fine-grained resolution
self.tracker.track_change(foo, resolution_level=2)
self.tracker.create_snapshot()
fp = self.tracker.objects[id(foo)].snapshots[-1]
refs = fp[1].refs
dref = [r for r in refs if r.name == '__dict__']
self.assertEqual(len(dref), 1)
dref = dref[0]
namerefs = [r.name for r in dref.refs]
self.assertTrue('[K] foo' in namerefs, namerefs)
self.assertTrue("[V] foo: 'foo'" in namerefs, namerefs)
