Rev 164: Fix up the tests for compute_total_size in http://bazaar.launchpad.net/~meliae-dev/meliae/trunk

[John Arbash Meinel]

At http://bazaar.launchpad.net/~meliae-dev/meliae/trunk

------------------------------------------------------------
revno: 164
revision-id: john at arbash-meinel.com-20100729144035-tlibn0y4qhlg2qyz
parent: john at arbash-meinel.com-20100729143729-glo62wr5im8e1ggf
committer: John Arbash Meinel <john at arbash-meinel.com>
branch nick: trunk
timestamp: Thu 2010-07-29 09:40:35 -0500
message:
  Fix up the tests for compute_total_size
-------------- next part --------------
=== modified file 'meliae/tests/test_loader.py'
--- a/meliae/tests/test_loader.py	2010-07-28 20:27:28 +0000
+++ b/meliae/tests/test_loader.py	2010-07-29 14:40:35 +0000
@@ -281,25 +281,20 @@
 
     def test_compute_total_size(self):
         manager = loader.load(_example_dump, show_prog=False)
-        manager.compute_total_size()
         objs = manager.objs
+        manager.compute_total_size(objs[1])
         self.assertEqual(261, objs[1].total_size)
-        self.assertEqual(197, objs[2].total_size)
-        self.assertEqual(68, objs[3].total_size)
-        self.assertEqual(12, objs[4].total_size)
-        self.assertEqual(12, objs[5].total_size)
-        self.assertEqual(29, objs[6].total_size)
-        self.assertEqual(44, objs[7].total_size)
-        self.assertEqual(257, objs[8].total_size)
 
     def test_compute_total_size_missing_ref(self):
         lines = list(_example_dump)
         # 999 isn't in the dump, not sure how we get these in real life, but
-        # they exist
-        lines.append('{"address": 8, "type": "tuple", "size": 16, "len": 1'
+        # they exist. we should live with references that can't be resolved.
+        lines[-1] = ('{"address": 8, "type": "tuple", "size": 16, "len": 1'
                      ', "refs": [999]}')
         manager = loader.load(lines, show_prog=False)
-        manager.compute_total_size()
+        obj = manager[8]
+        manager.compute_total_size(obj)
+        self.assertEqual(16, obj.total_size)
 
     def test_remove_expensive_references(self):
         lines = list(_example_dump)