Downgrade lock icon for broken-HTTPS subresources

third_party/WebKit/Source/core/loader/MixedContentCheckerTest.cpp

 // Copyright 2014 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 "config.h"
 #include "core/loader/MixedContentChecker.h"
 
+#include "core/loader/EmptyClients.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/weborigin/KURL.h"
 #include "platform/weborigin/SecurityOrigin.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "wtf/RefPtr.h"
 #include <base/macros.h>
 
 namespace blink {
 
 TEST(MixedContentCheckerTest, IsMixedContent)
@@ skipping 42 matching lines @@
     blockableMixedContent.setFrameType(WebURLRequest::FrameTypeAuxiliary);
     blockableMixedContent.setRequestContext(WebURLRequest::RequestContextScript);
     EXPECT_EQ(MixedContentChecker::ContextTypeBlockable, MixedContentChecker::contextTypeForInspector(&dummyPageHolder->frame(), blockableMixedContent));
 
     ResourceRequest optionallyBlockableMixedContent("http://example.test/foo.jpg");
     blockableMixedContent.setFrameType(WebURLRequest::FrameTypeAuxiliary);
     blockableMixedContent.setRequestContext(WebURLRequest::RequestContextImage);
     EXPECT_EQ(MixedContentChecker::ContextTypeOptionallyBlockable, MixedContentChecker::contextTypeForInspector(&dummyPageHolder->frame(), blockableMixedContent));
 }
 
+namespace {
+
+    enum CertErrorsContentType {

[Mike West, 2015/11/27 05:30:38]

Nit: No indentation.

[estark, 2015/11/28 02:46:56]

Hmm, `git cl format` seems to like it this way.

+        DisplayedContentWithCertErrors,
+        RanContentWithCertErrors
+    };
+
+    struct ContentWithCertErrorsInfo {
+        KURL url;
+        KURL mainResourceUrl;
+        CertErrorsContentType contentType;
+    };
+
+    class CertErrorsTracker {
+    public:
+        CertErrorsTracker() {}
+        ~CertErrorsTracker() {}
+
+        const ContentWithCertErrorsInfo& contentWithCertErrorsInfo() { return m_contentWithCertErrors; }
+
+        void didDisplayContentWithCertificateErrors(const KURL& url, const WebURL& mainResourceUrl)
+        {
+            m_contentWithCertErrors.url = url;
+            m_contentWithCertErrors.mainResourceUrl = mainResourceUrl;
+            m_contentWithCertErrors.contentType = DisplayedContentWithCertErrors;
+        }
+
+        void didRunContentWithCertificateErrors(const KURL& url, const WebURL& mainResourceUrl)
+        {
+            m_contentWithCertErrors.url = url;
+            m_contentWithCertErrors.mainResourceUrl = mainResourceUrl;
+            m_contentWithCertErrors.contentType = RanContentWithCertErrors;
+        }
+
+    private:
+        ContentWithCertErrorsInfo m_contentWithCertErrors;
+    };
+
+    class TestCertErrorsFrameLoaderClient : public EmptyFrameLoaderClient {
+    public:
+        ~TestCertErrorsFrameLoaderClient() {}
+
+        static PassOwnPtrWillBeRawPtr<TestCertErrorsFrameLoaderClient> create(CertErrorsTracker* tracker) { return adoptPtrWillBeNoop(new TestCertErrorsFrameLoaderClient(tracker)); }
+
+        void didDisplayContentWithCertificateErrors(const KURL& url, const CString& securityInfo, const WebURL& mainResourceUrl, const CString& mainResourceSecurityInfo)
+        {
+            m_tracker->didDisplayContentWithCertificateErrors(url, mainResourceUrl);
+        }
+
+        void didRunContentWithCertificateErrors(const KURL& url, const CString& securityInfo, const WebURL& mainResourceUrl, const CString& mainResourceSecurityInfo)
+        {
+            m_tracker->didRunContentWithCertificateErrors(url, mainResourceUrl);
+        }
+
+    private:
+        TestCertErrorsFrameLoaderClient(CertErrorsTracker* tracker)
+            : m_tracker(tracker)
+        {
+        }
+        CertErrorsTracker* m_tracker;
+    };

[Mike West, 2015/11/27 05:30:38]

This is fine, but I suspect you could do it more simply with a mock and some
EXPECT_CALL macros.

[estark, 2015/11/28 02:46:56]

Ah, that is much better, thanks!

+
+} // namespace
+
+TEST(MixedContentCheckerTest, HandleCertificateError)
+{
+    CertErrorsTracker certErrorsTracker;
+    OwnPtr<DummyPageHolder> dummyPageHolder = DummyPageHolder::create(IntSize(1, 1), nullptr, TestCertErrorsFrameLoaderClient::create(&certErrorsTracker));
+
+    KURL mainResourceUrl(KURL(), "https://example.test");
+    KURL displayedUrl(KURL(), "https://example-displayed.test");
+    KURL ranUrl(KURL(), "https://example-ran.test");
+
+    dummyPageHolder->frame().document()->setURL(mainResourceUrl);
+    EXPECT_EQ(KURL(), certErrorsTracker.contentWithCertErrorsInfo().url);
+    MixedContentChecker::handleCertificateError(&dummyPageHolder->frame(), ranUrl, WebURLRequest::RequestContextScript, WebURLRequest::FrameTypeNone, "");
+    EXPECT_EQ(ranUrl, certErrorsTracker.contentWithCertErrorsInfo().url);
+    EXPECT_EQ(mainResourceUrl, certErrorsTracker.contentWithCertErrorsInfo().mainResourceUrl);
+
+    MixedContentChecker::handleCertificateError(&dummyPageHolder->frame(), displayedUrl, WebURLRequest::RequestContextImage, WebURLRequest::FrameTypeNone, "");

[Mike West, 2015/11/27 05:30:38]

In a magical future world where we've blocked all mixed content, this will fail,
because RequestContextImage will be blockable. Would you mind humoring me by
adding an `ASSERT_EQ` that `MixedContentChecker::contextTypeFromContext` returns
optionally blockable? :)

[estark, 2015/11/28 02:46:56]

Done.

+    EXPECT_EQ(displayedUrl, certErrorsTracker.contentWithCertErrorsInfo().url);
+    EXPECT_EQ(mainResourceUrl, certErrorsTracker.contentWithCertErrorsInfo().mainResourceUrl);
+}
+
 } // namespace blink