1 files changed, 96 insertions, 0 deletions
diff --git a/runtime/oat/oat_file_test.cc b/runtime/oat/oat_file_test.cc
new file mode 100644
index 0000000000..80265f77f5
--- /dev/null
+++ b/runtime/oat/oat_file_test.cc
@@ -0,0 +1,96 @@
+/*
+ * Copyright (C) 2015 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "oat_file.h"
+
+#include <string>
+
+#include "common_runtime_test.h"
+#include "dexopt_test.h"
+#include "gtest/gtest.h"
+#include "scoped_thread_state_change-inl.h"
+#include "vdex_file.h"
+
+namespace art {
+
+class OatFileTest : public DexoptTest {};
+
+TEST_F(OatFileTest, LoadOat) {
+  std::string dex_location = GetScratchDir() + "/LoadOat.jar";
+
+  Copy(GetDexSrc1(), dex_location);
+  GenerateOatForTest(dex_location.c_str(), CompilerFilter::kSpeed);
+
+  std::string oat_location;
+  std::string error_msg;
+  ASSERT_TRUE(OatFileAssistant::DexLocationToOatFilename(
+      dex_location, kRuntimeISA, &oat_location, &error_msg))
+      << error_msg;
+  std::unique_ptr<OatFile> odex_file(OatFile::Open(/*zip_fd=*/-1,
+                                                   oat_location,
+                                                   oat_location,
+                                                   /*executable=*/false,
+                                                   /*low_4gb=*/false,
+                                                   dex_location,
+                                                   &error_msg));
+  ASSERT_TRUE(odex_file.get() != nullptr);
+
+  // Check that the vdex file was loaded in the reserved space of odex file.
+  EXPECT_EQ(odex_file->GetVdexFile()->Begin(), odex_file->VdexBegin());
+}
+
+TEST_F(OatFileTest, ChangingMultiDexUncompressed) {
+  std::string dex_location = GetScratchDir() + "/MultiDexUncompressedAligned.jar";
+
+  Copy(GetTestDexFileName("MultiDexUncompressedAligned"), dex_location);
+  GenerateOatForTest(dex_location.c_str(), CompilerFilter::kVerify);
+
+  std::string oat_location;
+  std::string error_msg;
+  ASSERT_TRUE(OatFileAssistant::DexLocationToOatFilename(
+      dex_location, kRuntimeISA, &oat_location, &error_msg))
+      << error_msg;
+
+  // Ensure we can load that file. Just a precondition.
+  {
+    std::unique_ptr<OatFile> odex_file(OatFile::Open(/*zip_fd=*/-1,
+                                                     oat_location,
+                                                     oat_location,
+                                                     /*executable=*/false,
+                                                     /*low_4gb=*/false,
+                                                     dex_location,
+                                                     &error_msg));
+    ASSERT_TRUE(odex_file != nullptr);
+    ASSERT_EQ(2u, odex_file->GetOatDexFiles().size());
+  }
+
+  // Now replace the source.
+  Copy(GetTestDexFileName("MainUncompressedAligned"), dex_location);
+
+  // And try to load again.
+  std::unique_ptr<OatFile> odex_file(OatFile::Open(/*zip_fd=*/-1,
+                                                   oat_location,
+                                                   oat_location,
+                                                   /*executable=*/false,
+                                                   /*low_4gb=*/false,
+                                                   dex_location,
+                                                   &error_msg));
+  EXPECT_TRUE(odex_file == nullptr);
+  EXPECT_NE(std::string::npos, error_msg.find("expected 2 uncompressed dex files, but found 1"))
+      << error_msg;
+}
+
+}  // namespace art