class TrackClassTestCase(unittest.TestCase):
def setUp(self):
self.tracker = ClassTracker()
def tearDown(self):
self.tracker.stop_periodic_snapshots()
self.tracker.clear()
def test_type_errors(self):
"""Test invalid parameters for class tracking.
"""
i = 42
j = 'Foobar'
k = [i,j]
l = {i: j}
foo = Foo()
bar = Bar()
self.assertRaises(TypeError, self.tracker.track_class, i)
self.assertRaises(TypeError, self.tracker.track_class, j)
self.assertRaises(TypeError, self.tracker.track_class, k)
self.assertRaises(TypeError, self.tracker.track_class, l)
self.assertRaises(TypeError, self.tracker.track_class, foo)
self.assertRaises(TypeError, self.tracker.track_class, bar)
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_class(self):
"""Test tracking objects through classes.
"""
self.tracker.track_class(Foo)
self.tracker.track_class(Bar)
self.tracker.track_class(Empty)
self.tracker.track_class(Foo)
foo = Foo()
bar = Bar()
empty = Empty()
self.assert_(id(foo) in self.tracker.objects)
self.assert_(id(bar) in self.tracker.objects)
self.assert_(id(empty) in self.tracker.objects)
def test_track_class_new(self):
"""Test tracking new style classes.
"""
self.tracker.track_class(FooNew)
self.tracker.track_class(BarNew)
foo = FooNew()
bar = BarNew()
self.assert_(id(foo) in self.tracker.objects)
self.assert_(id(bar) in self.tracker.objects)
def test_track_by_name(self):
"""Test registering objects by name.
"""
self.tracker.track_class(Foo, name='Foobar')
foo = Foo()
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.
"""
self.tracker.track_class(Foo, keep=1)
self.tracker.track_class(Bar)
foo = Foo()
bar = 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_class_history(self):
"""Test instance history of tracked class.
"""
self.tracker.track_class(Foo, name='Foo')
f1 = Foo()
f2 = Foo()
f3 = Foo()
del f1
del f2
f4 = Foo()
del f3
del f4
instances = [cnt for _, cnt in self.tracker.history['Foo']]
self.assertEqual(instances, [1, 2, 3, 2, 1, 2, 1, 0])
def test_trace(self):
"""Test instantiation tracing of tracked objects.
"""
from inspect import stack
self.tracker.track_class(Foo, trace=True)
self.tracker.track_class(BarNew, trace=True)
foo = Foo()
bar = BarNew()
idfoo = id(foo)
idbar = id(bar)
trace = []
st = stack()
try:
for fr in st:
trace.insert(0, fr[1:])
finally:
del st
self.assertEqual(self.tracker.objects[idfoo].trace[-1][3][0].strip(),"foo = Foo()")
self.assertEqual(self.tracker.objects[idfoo].trace[:-1],trace[:-1], trace)
self.assertEqual(self.tracker.objects[idbar].trace[:-1],trace[:-1], trace)
def test_detach(self):
"""Test detaching from tracked classes.
"""
self.tracker.track_class(Foo)
self.tracker.track_class(Bar)
foo = Foo()
bar = Bar()
self.assert_(id(foo) in self.tracker.objects)
self.assert_(id(bar) in self.tracker.objects)
self.tracker.detach_class(Foo)
self.tracker.detach_class(Bar)
foo2 = Foo()
bar2 = Bar()
self.assert_(id(foo2) not in self.tracker.objects)
self.assert_(id(bar2) not in self.tracker.objects)
self.assertRaises(KeyError, self.tracker.detach_class, Foo)
def test_change_name(self):
"""Test modifying name.
"""
self.tracker.track_class(Foo, name='Foobar')
self.tracker.track_class(Foo, name='Baz')
foo = Foo()
self.assert_('Foobar' not in self.tracker.index)
self.assert_('Baz' in self.tracker.index)
self.assertEqual(self.tracker.index['Baz'][0].ref(),foo)
class TrackClassTestCase(unittest.TestCase):
def setUp(self):
self.tracker = ClassTracker()
def tearDown(self):
self.tracker.stop_periodic_snapshots()
self.tracker.clear()
def test_type_errors(self):
"""Test invalid parameters for class tracking.
"""
i = 42
j = 'Foobar'
k = [i, j]
l = {i: j}
foo = Foo()
bar = Bar()
self.assertRaises(TypeError, self.tracker.track_class, i)
self.assertRaises(TypeError, self.tracker.track_class, j)
self.assertRaises(TypeError, self.tracker.track_class, k)
self.assertRaises(TypeError, self.tracker.track_class, l)
self.assertRaises(TypeError, self.tracker.track_class, foo)
self.assertRaises(TypeError, self.tracker.track_class, bar)
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_class(self):
"""Test tracking objects through classes.
"""
self.tracker.track_class(Foo)
self.tracker.track_class(Bar)
self.tracker.track_class(Empty)
self.tracker.track_class(Foo)
foo = Foo()
bar = Bar()
empty = Empty()
self.assertTrue(id(foo) in self.tracker.objects)
self.assertTrue(id(bar) in self.tracker.objects)
self.assertTrue(id(empty) in self.tracker.objects)
def test_track_class_new(self):
"""Test tracking new style classes.
"""
self.tracker.track_class(FooNew)
self.tracker.track_class(BarNew)
foo = FooNew()
bar = BarNew()
self.assertTrue(id(foo) in self.tracker.objects)
self.assertTrue(id(bar) in self.tracker.objects)
def test_track_by_name(self):
"""Test registering objects by name.
"""
self.tracker.track_class(Foo, name='Foobar')
foo = Foo()
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.
"""
self.tracker.track_class(Foo, keep=1)
self.tracker.track_class(Bar)
foo = Foo()
bar = 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_class_history(self):
"""Test instance history of tracked class.
"""
self.tracker.track_class(Foo, name='Foo')
f1 = Foo()
f2 = Foo()
f3 = Foo()
del f1
del f2
f4 = Foo()
del f3
del f4
instances = [cnt for _, cnt in self.tracker.history['Foo']]
self.assertEqual(instances, [1, 2, 3, 2, 1, 2, 1, 0])
def test_trace(self):
"""Test instantiation tracing of tracked objects.
"""
from inspect import stack
self.tracker.track_class(Foo, trace=True)
self.tracker.track_class(BarNew, trace=True)
foo = Foo()
bar = BarNew()
idfoo = id(foo)
idbar = id(bar)
trace = []
st = stack()
try:
for fr in st:
trace.insert(0, fr[1:])
finally:
del st
self.assertEqual(self.tracker.objects[idfoo].trace[-1][3][0].strip(),
"foo = Foo()")
self.assertEqual(self.tracker.objects[idfoo].trace[:-1], trace[:-1],
trace)
self.assertEqual(self.tracker.objects[idbar].trace[:-1], trace[:-1],
trace)
def test_detach(self):
"""Test detaching from tracked classes.
"""
self.tracker.track_class(Foo)
self.tracker.track_class(Bar)
foo = Foo()
bar = Bar()
self.assertTrue(id(foo) in self.tracker.objects)
self.assertTrue(id(bar) in self.tracker.objects)
self.tracker.detach_class(Foo)
self.tracker.detach_class(Bar)
foo2 = Foo()
bar2 = Bar()
self.assertTrue(id(foo2) not in self.tracker.objects)
self.assertTrue(id(bar2) not in self.tracker.objects)
self.assertRaises(KeyError, self.tracker.detach_class, Foo)
def test_change_name(self):
"""Test modifying name.
"""
self.tracker.track_class(Foo, name='Foobar')
self.tracker.track_class(Foo, name='Baz')
foo = Foo()
self.assertTrue('Foobar' not in self.tracker.index)
self.assertTrue('Baz' in self.tracker.index)
self.assertEqual(self.tracker.index['Baz'][0].ref(), foo)
