commit fddf6f6b4af42c32dd2e92234bab0741118d092e
author Brian Carlstrom <bdc@google.com> Thu Mar 15 16:56:45 2012 -0700
committer Brian Carlstrom <bdc@google.com> Thu Mar 15 16:57:03 2012 -0700
tree 3c6db33df447f729abfa0671e40936a89384dbd6
parent cbf0b61f5415e96a76284deff87e7cb9aba022b9

Add Heap::GetImageSpace()

Change-Id: Ib52898043948a3db2539f7b689d5d97af370e9ca

src/image_test.cc

diff --git a/src/image_test.cc b/src/image_test.cc
index e886e49..fcea695 100644
--- a/src/image_test.cc
+++ b/src/image_test.cc
@@ -64,6 +64,7 @@
     Space* space = heap->GetSpaces()[0];
     ASSERT_FALSE(space->IsImageSpace());
     ASSERT_TRUE(space != NULL);
+    ASSERT_EQ(space, heap->GetAllocSpace());
     ASSERT_GE(sizeof(image_header) + space->Size(), static_cast<size_t>(file->Length()));
   }
 
@@ -88,9 +89,13 @@
   Heap* heap = Runtime::Current()->GetHeap();
   ASSERT_EQ(2U, heap->GetSpaces().size());
   ASSERT_TRUE(heap->GetSpaces()[0]->IsImageSpace());
+  ASSERT_FALSE(heap->GetSpaces()[0]->IsAllocSpace());
   ASSERT_FALSE(heap->GetSpaces()[1]->IsImageSpace());
+  ASSERT_TRUE(heap->GetSpaces()[1]->IsAllocSpace());
+  ASSERT_TRUE(heap->GetImageSpace() != NULL);
+  ASSERT_TRUE(heap->GetAllocSpace() != NULL);
 
-  ImageSpace* image_space = heap->GetSpaces()[0]->AsImageSpace();
+  ImageSpace* image_space = heap->GetImageSpace();
   byte* image_begin = image_space->Begin();
   byte* image_end = image_space->End();
   CHECK_EQ(requested_image_base, reinterpret_cast<uintptr_t>(image_begin));