fix typo statement

trolldbois · Oct 8, 2015 · 41c90a4 · 41c90a4
1 parent a692693
commit 41c90a4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/haystack/reverse/test_utils.py b/test/haystack/reverse/test_utils.py
@@ -9,6 +9,7 @@
 
 from haystack.reverse import utils
 from haystack.reverse import context
+from haystack import dump_loader
 
 
 class TestBasicFunctions(unittest.TestCase):
@@ -23,7 +24,11 @@ def test_closestFloorValue(self):
         with self.assertRaises(ValueError):
             utils.closestFloorValue(-1, lst)
 
-        ctx = context.get_context('test/src/test-ctypes3.32.dump')
+        memory_handler = dump_loader.load('test/src/test-ctypes3.32.dump')
+        finder = memory_handler.get_heap_finder()
+        heap = finder.get_heap_mappings()[0]
+        heap_addr = heap.get_marked_heap_address()
+        ctx = context.get_context_for_address(memory_handler, heap_addr)
         lst = ctx._structures_addresses
         # print ['0x%0.8x'%i for i in lst]