vm/service/observatory: Add coverage for functions, some tests, and fix buggy filter

runtime/vm/coverage_test.cc

 // Copyright (c) 2014, 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 "vm/coverage.h"
 #include "vm/dart_api_impl.h"
 #include "vm/unit_test.h"
 
 namespace dart {
 
-static void ExecuteScript(const char* script) {
+static RawObject* ExecuteScript(const char* script) {
   Dart_Handle h_lib = TestCase::LoadTestScript(script, NULL);
   EXPECT_VALID(h_lib);
   Library& lib = Library::Handle();
   lib ^= Api::UnwrapHandle(h_lib);
   EXPECT(!lib.IsNull());
   Dart_Handle result = Dart_Invoke(h_lib, NewString("main"), 0, NULL);
   EXPECT_VALID(result);
+  return Api::UnwrapHandle(h_lib);
 }
 
 
+class FunctionCoverageFilter : public CoverageFilter {
+ public:
+  explicit FunctionCoverageFilter(const Function& func) : func_(func) {}
+  bool ShouldOutputCoverageFor(const Library& lib,
+                               const String& script_url,
+                               const Class& cls,
+                               const Function& func) const {
+    return func.raw() == func_.raw();
+  }
+ private:
+  const Function& func_;
+};
+
+
 TEST_CASE(Coverage_Empty) {
   const char* kScript =
       "main() {\n"
       "}";
 
   Isolate* isolate = Isolate::Current();
   ExecuteScript(kScript);
 
   JSONStream js;
   CodeCoverage::PrintJSON(isolate, &js, NULL);
@@ skipping 38 matching lines @@
       "\"kind\":\"script\"},\"hits\":[3,1,5,4,6,3]}", js.ToCString());
 
   // Data for the fake class containing main().
   EXPECT_SUBSTRING(
       "{\"source\":\"test-lib\",\"script\":{"
       "\"type\":\"@Script\",\"id\":\"scripts\\/test-lib\","
       "\"name\":\"test-lib\",\"user_name\":\"test-lib\","
       "\"kind\":\"script\"},\"hits\":[10,1,11,1]}", js.ToCString());
 }
 
+
+TEST_CASE(Coverage_FilterFunction) {
+  const char* kScript =
+      "class Foo {\n"
+      "  var x;\n"
+      "  var y;\n"
+      "  Foo(this.x);\n"
+      "  Foo.other(this.x, this.y);\n"
+      "  Foo.yetAnother();\n"
+      "}\n"
+      "main() {\n"
+      "  var foo = new Foo(7);\n"
+      "}\n";
+
+  Isolate* isolate = Isolate::Current();
+  Library& lib = Library::Handle();
+  lib ^= ExecuteScript(kScript);
+  ASSERT(!lib.IsNull());
+  const Class& cls = Class::Handle(
+      lib.LookupClass(String::Handle(String::New("Foo"))));
+  ASSERT(!cls.IsNull());
+  const Function& func = Function::Handle(
+      cls.LookupFunction(String::Handle(String::New("Foo.yetAnother"))));
+  ASSERT(!func.IsNull());
+
+  JSONStream js;
+  FunctionCoverageFilter filter(func);
+  CodeCoverage::PrintJSON(isolate, &js, &filter);
+  // Only expect coverage data for Foo.yetAnother() on line 6.
+  EXPECT_SUBSTRING(
+      "{\"source\":\"test-lib\",\"script\":{"
+      "\"type\":\"@Script\",\"id\":\"scripts\\/test-lib\","
+      "\"name\":\"test-lib\",\"user_name\":\"test-lib\","
+      "\"kind\":\"script\"},\"hits\":[6,0]}", js.ToCString());
+}
+
 }  // namespace dart