Framework to ascribe all network data use to a source.

chrome/browser/data_use_measurement/chrome_data_use_ascriber_service_browsertest.cc

+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/data_use_measurement/chrome_data_use_ascriber_service.h"
+
+#include <memory>
+
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/data_use_measurement/chrome_data_use_ascriber.h"
+#include "chrome/browser/io_thread.h"
+#include "chrome/test/base/in_process_browser_test.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/test/test_utils.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace data_use_measurement {
+
+class ChromeDataUseAscriberServiceTest : public InProcessBrowserTest {

[Lei Zhang, 2016/09/08 22:04:43]

Do you plan to add more test cases?

As is, it may be possible to write this as a cheaper unit test using
TestBrowserThreadBundle and TestingBrowserProcess.

[Not at Google. Contact bengr, 2016/09/09 18:37:16]

Done.

It was a little awkward to instantiate IOThread in a unittest, so I had gone
with browsertest.

+ protected:
+  void SetUpOnMainThread() override {
+    content::BrowserThread::PostTask(
+        content::BrowserThread::IO, FROM_HERE,
+        base::Bind(&ChromeDataUseAscriberServiceTest::SetUpOnIOThread,
+                   base::Unretained(this), new ChromeDataUseAscriberService(),
+                   g_browser_process->io_thread()));
+    RunAllPendingInMessageLoop(content::BrowserThread::IO);
+  }
+
+  void SetUpOnIOThread(ChromeDataUseAscriberService* service,
+                       IOThread* io_thread) {
+    io_thread->globals()->data_use_ascriber.reset(
+        new data_use_measurement::ChromeDataUseAscriber());
+    service_.reset(service);
+  }
+
+  void TearDownOnMainThread() override {
+    content::BrowserThread::PostTask(
+        content::BrowserThread::IO, FROM_HERE,
+        base::Bind(&ChromeDataUseAscriberServiceTest::TearDownOnIOThread,
+                   base::Unretained(this), g_browser_process->io_thread()));
+    RunAllPendingInMessageLoop(content::BrowserThread::IO);
+  }
+
+  void TearDownOnIOThread(IOThread* io_thread) {
+    io_thread->globals()->data_use_ascriber.reset();
+  }
+
+  ChromeDataUseAscriber* ascriber() { return service_->ascriber_; }
+
+ private:
+  std::unique_ptr<ChromeDataUseAscriberService> service_;
+};
+
+IN_PROC_BROWSER_TEST_F(ChromeDataUseAscriberServiceTest, Initialization) {
+  EXPECT_NE(nullptr, ascriber());
+}
+
+}  // namespace data_use_measurement