memory_map_unittest.py revision 5d1f7b1de12d16ceb2c938c56701a3e8bfa558f7
1# Copyright 2014 The Chromium Authors. All rights reserved. 2# Use of this source code is governed by a BSD-style license that can be 3# found in the LICENSE file. 4 5import unittest 6 7from memory_inspector.core import memory_map 8 9 10class MemoryMapTest(unittest.TestCase): 11 def runTest(self): 12 mmap = memory_map.Map() 13 map_entry1 = memory_map.MapEntry(4096, 8191, 'rw--', '/foo', 0) 14 map_entry2 = memory_map.MapEntry(65536, 81919, 'rw--', '/bar', 4096) 15 16 # Test the de-offset logic. 17 self.assertEqual(map_entry1.GetRelativeOffset(4096), 0) 18 self.assertEqual(map_entry1.GetRelativeOffset(4100), 4) 19 self.assertEqual(map_entry2.GetRelativeOffset(65536), 4096) 20 21 # Test the page-resident logic. 22 map_entry2.resident_pages = [5] # 5 -> 101b. 23 self.assertTrue(map_entry2.IsPageResident(0)) 24 self.assertFalse(map_entry2.IsPageResident(1)) 25 self.assertTrue(map_entry2.IsPageResident(2)) 26 27 # Test the lookup logic. 28 mmap.Add(map_entry1) 29 mmap.Add(map_entry2) 30 self.assertIsNone(mmap.Lookup(1024)) 31 self.assertEqual(mmap.Lookup(4096), map_entry1) 32 self.assertEqual(mmap.Lookup(6000), map_entry1) 33 self.assertEqual(mmap.Lookup(8191), map_entry1) 34 self.assertIsNone(mmap.Lookup(8192)) 35 self.assertIsNone(mmap.Lookup(65535)) 36 self.assertEqual(mmap.Lookup(65536), map_entry2) 37 self.assertEqual(mmap.Lookup(67000), map_entry2) 38 self.assertEqual(mmap.Lookup(81919), map_entry2) 39 self.assertIsNone(mmap.Lookup(81920)) 40