WKWebView: Implemented recoverable SSL interstitials.

ios/web/net/cert_verification_cache_unittest.cc

+// Copyright 2015 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 "ios/web/net/cert_verification_cache.h"
+
+#include "net/base/test_data_directory.h"
+#include "net/test/cert_test_util.h"
+#include "testing/platform_test.h"
+
+namespace {
+// Test cert filenames.
+const char kCertFileName1[] = "ok_cert.pem";
+const char kCertFileName2[] = "expired_cert.pem";
+// Test hostnames.
+const char kHostName1[] = "www.example.com";
+const char kHostName2[] = "www.chromium.org";
+// Test values.
+const int kValue1 = 5;
+const int kValue2 = 6;
+}  // namespace
+
+// Test fixture to test CertVerificationCache class.
+class CertVerificationCacheTest : public PlatformTest {
+ protected:
+  // Loads cert with the given |file_name|.
+  scoped_refptr<net::X509Certificate> GetCert(const std::string& file_name) {
+    return net::ImportCertFromFile(net::GetTestCertsDirectory(), file_name);
+  }
+
+  web::CertVerificationCache<int> cache_;
+};
+
+// Tests retrieving from empty cache.
+TEST_F(CertVerificationCacheTest, EmptyCache) {
+  // Cache miss because it is empty.
+  int value;
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName1), kHostName1, &value));
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName1), kHostName2, &value));
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName2), kHostName1, &value));
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName2), kHostName2, &value));
+}
+
+// Tests Get-Set cache operations.
+TEST_F(CertVerificationCacheTest, TestReadingWriting) {
+  // Insert first cert.
+  cache_.Set(GetCert(kCertFileName1), kHostName1, kValue1);
+
+  // Cache miss because hostname/cert mismatch.
+  int value;
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName1), kHostName2, &value));
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName2), kHostName1, &value));
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName2), kHostName2, &value));
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName1), "", &value));
+
+  // Cahe hit.
+  EXPECT_TRUE(cache_.Get(GetCert(kCertFileName1), kHostName1, &value));
+  EXPECT_EQ(kValue1, value);
+
+  // Insert second cert.
+  cache_.Set(GetCert(kCertFileName2), kHostName2, kValue2);
+
+  // Cache miss because hostname/cert mismatch.
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName1), kHostName2, &value));
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName2), kHostName1, &value));
+
+  // Cahe hits.
+  EXPECT_TRUE(cache_.Get(GetCert(kCertFileName1), kHostName1, &value));
+  EXPECT_EQ(kValue1, value);
+  EXPECT_TRUE(cache_.Get(GetCert(kCertFileName2), kHostName2, &value));
+  EXPECT_EQ(kValue2, value);
+}
+
+// Tests resetting the cache.
+TEST_F(CertVerificationCacheTest, Reset) {
+  cache_.Set(GetCert(kCertFileName1), kHostName1, kValue1);
+  cache_.Set(GetCert(kCertFileName2), kHostName2, kValue2);
+
+  cache_.reset();
+
+  // Cache miss because it is empty.
+  int value;
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName1), kHostName1, &value));
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName1), kHostName2, &value));
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName2), kHostName1, &value));
+  EXPECT_FALSE(cache_.Get(GetCert(kCertFileName2), kHostName2, &value));
+}