Enable security state computation for HeadlessWebContents.

headless/lib/headless_web_contents_browsertest.cc

Index: headless/lib/headless_web_contents_browsertest.cc
diff --git a/headless/lib/headless_web_contents_browsertest.cc b/headless/lib/headless_web_contents_browsertest.cc
index 7865c77b56b9c7e22e5de22556940d5dc2cc7f12..d2ec376047ce75e4e6c2d00f3cd29968ed02ef60 100644
--- a/headless/lib/headless_web_contents_browsertest.cc
+++ b/headless/lib/headless_web_contents_browsertest.cc
@@ -1,4 +1,4 @@
-// Copyright 2016 The Chromium Authors. All rights reserved.
+// Copyright 2015 The Chromium Authors. All rights reserved.

[estark, 2016/11/17 08:44:27]

Did we go back to 2015? That would be great, 2016 sucks

[Eric Seckler, 2016/11/17 08:50:57]

Ha! Good spot, no idea how that happened :)

 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
@@ -8,6 +8,7 @@
 
 #include "content/public/test/browser_test.h"
 #include "headless/public/devtools/domains/page.h"
+#include "headless/public/devtools/domains/security.h"
 #include "headless/public/headless_browser.h"
 #include "headless/public/headless_devtools_client.h"
 #include "headless/public/headless_web_contents.h"
@@ -74,4 +75,25 @@ class HeadlessWebContentsScreenshotTest
 
 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessWebContentsScreenshotTest);
 
+class HeadlessWebContentsSecurityTest
+    : public HeadlessAsyncDevTooledBrowserTest,
+      public security::ExperimentalObserver {
+ public:
+  void RunDevTooledTest() override {
+    devtools_client_->GetSecurity()->GetExperimental()->AddObserver(this);
+    devtools_client_->GetSecurity()->GetExperimental()->Enable(
+        security::EnableParams::Builder().Build());
+  }
+
+  void OnSecurityStateChanged(
+      const security::SecurityStateChangedParams& params) override {
+    EXPECT_EQ(security::SecurityState::NEUTRAL, params.GetSecurityState());
+    EXPECT_TRUE(params.HasExplanations());
+
+    FinishAsynchronousTest();
+  }
+};
+
+HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessWebContentsSecurityTest);
+
 }  // namespace headless