Fix for issue 1456 - register libraries read from a partial snapshot

vm/snapshot_test.cc

 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
+#include "include/dart_debugger_api.h"
 #include "platform/assert.h"
 #include "vm/bigint_operations.h"
 #include "vm/class_finalizer.h"
 #include "vm/dart_api_impl.h"
 #include "vm/dart_api_state.h"
 #include "vm/snapshot.h"
 #include "vm/unit_test.h"
 
 namespace dart {
 
@@ skipping 522 matching lines @@
     }
     EXPECT_VALID(result);
     Dart_ExitScope();
   }
   Dart_ShutdownIsolate();
   free(buffer);
 }
 
 
 UNIT_TEST_CASE(ScriptSnapshot) {
+  const char* kLibScriptChars =
+      "#library('dart:import-lib');"
+      "class LibFields  {"
+      "  LibFields(int i, int j) : fld1 = i, fld2 = j {}"
+      "  int fld1;"
+      "  final int fld2;"
+      "}";
   const char* kScriptChars =
       "class Fields  {"
       "  Fields(int i, int j) : fld1 = i, fld2 = j {}"
       "  int fld1;"
       "  final int fld2;"
       "  static int fld3;"
       "  static final int fld4 = 10;"
       "}"
       "class FieldsTest {"
       "  static Fields testMain() {"
@@ skipping 16 matching lines @@
       "    }"
       "    return obj;"
       "  }"
       "}";
   Dart_Handle result;
 
   uint8_t* buffer;
   intptr_t size;
   uint8_t* full_snapshot = NULL;
   uint8_t* script_snapshot = NULL;
+  intptr_t expected_num_libs;
+  intptr_t actual_num_libs;
 
   {
     // Start an Isolate, and create a full snapshot of it.
     TestIsolateScope __test_isolate__;
     Dart_EnterScope();  // Start a Dart API scope for invoking API functions.
 
     // Write out the script snapshot.
     result = Dart_CreateSnapshot(&buffer, &size);
     EXPECT_VALID(result);
     full_snapshot = reinterpret_cast<uint8_t*>(malloc(size));
     memmove(full_snapshot, buffer, size);
     Dart_ExitScope();
   }
 
   {
     // Create an Isolate using the full snapshot, load a script and create
     // a script snapshot of the script.
     TestCase::CreateTestIsolateFromSnapshot(full_snapshot);
     Dart_EnterScope();  // Start a Dart API scope for invoking API functions.
 
+    // Load the library.
+    Dart_Handle import_lib = Dart_LoadLibrary(Dart_NewString("dart:import-lib"),
+                                              Dart_NewString(kLibScriptChars));
+    EXPECT_VALID(import_lib);
+
     // Create a test library and Load up a test script in it.
     TestCase::LoadTestScript(kScriptChars, NULL);
 
+    EXPECT_VALID(Dart_LibraryImportLibrary(TestCase::lib(), import_lib));
+
+    // Get list of library URLs loaded and save the count.
+    Dart_Handle libs = Dart_GetLibraryURLs();
+    EXPECT(Dart_IsList(libs));
+    Dart_ListLength(libs, &expected_num_libs);
+
     // Write out the script snapshot.
     result = Dart_CreateScriptSnapshot(&buffer, &size);
     EXPECT_VALID(result);
     script_snapshot = reinterpret_cast<uint8_t*>(malloc(size));
     memmove(script_snapshot, buffer, size);
     Dart_ExitScope();
     Dart_ShutdownIsolate();
   }
 
   {
     // Now Create an Isolate using the full snapshot and load the
     // script snapshot created above and execute it.
     TestCase::CreateTestIsolateFromSnapshot(full_snapshot);
     Dart_EnterScope();  // Start a Dart API scope for invoking API functions.
 
     // Load the test library from the snapshot.
     EXPECT(script_snapshot != NULL);
     result = Dart_LoadScriptFromSnapshot(script_snapshot);
     EXPECT_VALID(result);
 
+    // Get list of library URLs loaded and compare with expected count.
+    Dart_Handle libs = Dart_GetLibraryURLs();
+    EXPECT(Dart_IsList(libs));
+    Dart_ListLength(libs, &actual_num_libs);
+
+    EXPECT_EQ(expected_num_libs, actual_num_libs);
+
     // Invoke a function which returns an object.
     result = Dart_InvokeStatic(result,
                                Dart_NewString("FieldsTest"),
                                Dart_NewString("testMain"),
                                0,
                                NULL);
     EXPECT_VALID(result);
     Dart_ExitScope();
   }
   Dart_ShutdownIsolate();
@@ skipping 124 matching lines @@
       free(buffer);
     }
   }
   Dart_ExitScope();
   Dart_ShutdownIsolate();
 }
 
 #endif  // defined(TARGET_ARCH_IA32) || defined(TARGET_ARCH_X64).
 
 }  // namespace dart