WKWebView: Added cert verification API to web controller.

ios/web/net/cert_verifier_block_adapter_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_verifier_block_adapter.h"
 
+#include "base/location.h"
+#include "base/message_loop/message_loop.h"

[Ryan Sleevi, 2015/08/14 21:43:54]

Blergh. It seems the iOS test code still uses the deprecated interface. Filed
https://code.google.com/p/chromium/issues/detail?id=521138 so that you shouldn't
have to use base::MessageLoop

[Eugene But (OOO till 7-30), 2015/08/19 17:57:36]

Acknowledged.

 #include "base/test/ios/wait_util.h"
+#include "base/thread_task_runner_handle.h"
+#include "ios/web/public/test/test_web_thread_bundle.h"
 #include "net/base/net_errors.h"
+#include "net/base/test_data_directory.h"
+#include "net/cert/cert_verifier.h"
 #include "net/cert/cert_verify_result.h"
 #include "net/cert/crl_set.h"
 #include "net/cert/x509_certificate.h"
+#include "net/log/net_log.h"
+#include "net/test/cert_test_util.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/platform_test.h"
 
 namespace net {
 
 using testing::_;
 
 namespace {
-
+// Test cert filename.
+const char kCertFileName[] = "ok_cert.pem";
 // Test hostname for CertVerifier.
 const char kHostName[] = "chromium.org";
 // Test OCSP response for CertVerifier.
 const char kOcspResponse[] = "ocsp";
 
 // Mocks CertVerifier for CertVerifierBlockAdapter testing.
 class CertVerifierMock : public CertVerifier {
  public:
   MOCK_METHOD9(Verify,
                int(X509Certificate* cert,
                    const std::string& hostname,
                    const std::string& ocsp_response,
                    int flags,
                    CRLSet* crl_set,
                    CertVerifyResult* verify_result,
                    const CompletionCallback& callback,
                    scoped_ptr<Request>* out_req,
                    const BoundNetLog& net_log));
 };
 
 // Sets CertVerifyResult to emulate CertVerifier behavior.
 ACTION_P(SetVerifyResult, result) {
   *arg5 = result;
 }
 
 // Calls CompletionCallback to emulate CertVerifier behavior.
-ACTION(RunCallback) {
-  arg6.Run(0);
+ACTION_P(RunCallback, status_result) {
+  base::ThreadTaskRunnerHandle::Get()->PostTask(
+      FROM_HERE, base::Bind(arg6, status_result));
 }
 
 }  // namespace
 
 // Test fixture to test CertVerifierBlockAdapter class.
 class CertVerifierBlockAdapterTest : public PlatformTest {
  protected:
-  void SetUp() override {
-    PlatformTest::SetUp();
-
-    cert_ = new X509Certificate("test", "test", base::Time(), base::Time());
-    scoped_ptr<CertVerifierMock> cert_verifier_mock(new CertVerifierMock());
-    cert_verifier_mock_ = cert_verifier_mock.get();
-    test_adapter_.reset(
-        new CertVerifierBlockAdapter(cert_verifier_mock.Pass()));
+  CertVerifierBlockAdapterTest()
+      : cert_(ImportCertFromFile(GetTestCertsDirectory(), kCertFileName)),
+        test_adapter_(
+            new CertVerifierBlockAdapter(&cert_verifier_mock_, &net_log_)) {
+    DCHECK(cert_);
   }
 
   // Performs synchronous verification.
   void Verify(CertVerifierBlockAdapter::Params params,
-              scoped_ptr<net::CertVerifyResult>* result,
-              int* status) {
+              net::CertVerifyResult* cert_verify_result,
+              int* status_result) {
     __block bool verification_completed = false;
     test_adapter_->Verify(params,
-                          ^(scoped_ptr<net::CertVerifyResult> callback_result,
-                            int callback_status) {
-                            *result = callback_result.Pass();
-                            *status = callback_status;
+                          ^(net::CertVerifyResult callback_cert_verify_result,
+                            int callback_status_result) {
+                            *cert_verify_result = callback_cert_verify_result;
+                            *status_result = callback_status_result;
                             verification_completed = true;
                           });
     base::test::ios::WaitUntilCondition(^{
       return verification_completed;
-    });
+    }, base::MessageLoop::current(), base::TimeDelta());
   }
 
-  // Fake certificate created for testing.
+  web::TestWebThreadBundle thread_bundle_;
   scoped_refptr<X509Certificate> cert_;
-  // Testable |CertVerifierBlockAdapter| object.
+  CertVerifierMock cert_verifier_mock_;
+  NetLog net_log_;
   scoped_ptr<CertVerifierBlockAdapter> test_adapter_;
-  // CertVerifier mock owned by |test_adapter_|.
-  CertVerifierMock* cert_verifier_mock_;
 };
 
 // Tests |Verify| with default params and synchronous verification.
 TEST_F(CertVerifierBlockAdapterTest, DefaultParamsAndSync) {
   // Set up expectation.
-  net::CertVerifyResult expectedResult;
-  expectedResult.cert_status = net::CERT_STATUS_AUTHORITY_INVALID;
-  const int kExpectedStatus = 0;
-  EXPECT_CALL(*cert_verifier_mock_,
-              Verify(cert_.get(), kHostName, "", 0, nullptr, _, _, _, _))
+  net::CertVerifyResult expected_cert_verify_result;
+  expected_cert_verify_result.cert_status = net::CERT_STATUS_AUTHORITY_INVALID;
+  const int kExpectedStatus = OK;
+  EXPECT_CALL(cert_verifier_mock_, Verify(cert_.get(), kHostName, std::string(),
+                                          0, nullptr, _, _, _, _))
       .Times(1)
-      .WillOnce(testing::DoAll(SetVerifyResult(expectedResult),
+      .WillOnce(testing::DoAll(SetVerifyResult(expected_cert_verify_result),
                                testing::Return(kExpectedStatus)));
 
   // Call |Verify|.
-  scoped_ptr<CertVerifyResult> actualResult;
-  int actualStatus = -1;
+  CertVerifyResult actual_cert_verify_result;
+  int actual_status = -1;
   CertVerifierBlockAdapter::Params params(cert_.get(), kHostName);
-  Verify(params, &actualResult, &actualStatus);
+  Verify(params, &actual_cert_verify_result, &actual_status);
 
   // Ensure that Verification results are correct.
-  EXPECT_EQ(kExpectedStatus, actualStatus);
-  EXPECT_EQ(expectedResult.cert_status, actualResult->cert_status);
+  EXPECT_EQ(kExpectedStatus, actual_status);
+  EXPECT_EQ(expected_cert_verify_result.cert_status,
+            actual_cert_verify_result.cert_status);
 }
 
 // Tests |Verify| with default params and asynchronous verification.
 TEST_F(CertVerifierBlockAdapterTest, DefaultParamsAndAsync) {
   // Set up expectation.
-  net::CertVerifyResult expectedResult;
-  expectedResult.is_issued_by_known_root = true;
-  const int kExpectedStatus = 0;
-  EXPECT_CALL(*cert_verifier_mock_,
-              Verify(cert_.get(), kHostName, "", 0, nullptr, _, _, _, _))
+  net::CertVerifyResult expected_cert_verify_result;
+  expected_cert_verify_result.is_issued_by_known_root = true;
+  const int kExpectedStatus = OK;
+  EXPECT_CALL(cert_verifier_mock_, Verify(cert_.get(), kHostName, std::string(),
+                                          0, nullptr, _, _, _, _))
       .Times(1)
-      .WillOnce(testing::DoAll(SetVerifyResult(expectedResult), RunCallback(),
+      .WillOnce(testing::DoAll(SetVerifyResult(expected_cert_verify_result),
+                               RunCallback(kExpectedStatus),
                                testing::Return(ERR_IO_PENDING)));
 
   // Call |Verify|.
-  scoped_ptr<CertVerifyResult> actualResult;
-  int actualStatus = -1;
+  CertVerifyResult actual_cert_verify_result;
+  int actual_status = -1;
   CertVerifierBlockAdapter::Params params(cert_.get(), kHostName);
-  Verify(params, &actualResult, &actualStatus);
+  Verify(params, &actual_cert_verify_result, &actual_status);
 
   // Ensure that Verification results are correct.
-  EXPECT_EQ(kExpectedStatus, actualStatus);
-  EXPECT_EQ(expectedResult.is_issued_by_known_root,
-            actualResult->is_issued_by_known_root);
+  EXPECT_EQ(kExpectedStatus, actual_status);
+  EXPECT_EQ(expected_cert_verify_result.is_issued_by_known_root,
+            actual_cert_verify_result.is_issued_by_known_root);
 }
 
-// Tests |Verify| with invalid arguments.
-TEST_F(CertVerifierBlockAdapterTest, InvalidParamsAndError) {
+// Tests |Verify| with invalid cert argument.
+TEST_F(CertVerifierBlockAdapterTest, InvalidCert) {
+  // Call |Verify|.
+  CertVerifyResult actual_cert_verify_result;
+  int actual_status = -1;
+  CertVerifierBlockAdapter::Params params(nullptr, kHostName);
+  Verify(params, &actual_cert_verify_result, &actual_status);
+
+  // Ensure that Verification results are correct.
+  EXPECT_EQ(ERR_INVALID_ARGUMENT, actual_status);
+}
+
+// Tests |Verify| with invalid hostname argument.
+TEST_F(CertVerifierBlockAdapterTest, InvalidHostname) {
+  // Call |Verify|.
+  CertVerifyResult actual_cert_verify_result;
+  int actual_status = -1;
+  CertVerifierBlockAdapter::Params params(cert_.get(), std::string());
+  Verify(params, &actual_cert_verify_result, &actual_status);
+
+  // Ensure that Verification results are correct.
+  EXPECT_EQ(ERR_INVALID_ARGUMENT, actual_status);
+}
+
+// Tests |Verify| with synchronous error.
+TEST_F(CertVerifierBlockAdapterTest, DefaultParamsAndSyncError) {
   // Set up expectation.
-  const int kExpectedStatus = ERR_INVALID_ARGUMENT;
-  EXPECT_CALL(*cert_verifier_mock_,
-              Verify(nullptr, "", "", 0, nullptr, _, _, _, _))
+  const int kExpectedStatus = ERR_INSUFFICIENT_RESOURCES;
+  EXPECT_CALL(cert_verifier_mock_, Verify(cert_.get(), kHostName, std::string(),
+                                          0, nullptr, _, _, _, _))
       .Times(1)
       .WillOnce(testing::Return(kExpectedStatus));
 
   // Call |Verify|.
-  scoped_ptr<CertVerifyResult> actualResult;
-  int actualStatus = -1;
-  CertVerifierBlockAdapter::Params params(nullptr, "");
-  Verify(params, &actualResult, &actualStatus);
+  CertVerifyResult actual_cert_verify_result;
+  int actual_status = -1;
+  CertVerifierBlockAdapter::Params params(cert_.get(), kHostName);
+  Verify(params, &actual_cert_verify_result, &actual_status);
 
   // Ensure that Verification results are correct.
-  EXPECT_EQ(kExpectedStatus, actualStatus);
-  EXPECT_FALSE(actualResult);
+  EXPECT_EQ(kExpectedStatus, actual_status);
 }
 
-// Tests |Verify| with error.
-TEST_F(CertVerifierBlockAdapterTest, DefaultParamsAndError) {
+// Tests |Verify| with asynchronous error.
+TEST_F(CertVerifierBlockAdapterTest, DefaultParamsAndAsyncError) {
   // Set up expectation.
-  const int kExpectedStatus = ERR_INSUFFICIENT_RESOURCES;
-  EXPECT_CALL(*cert_verifier_mock_,
-              Verify(cert_.get(), kHostName, "", 0, nullptr, _, _, _, _))
+  net::CertVerifyResult expected_cert_verify_result;
+  expected_cert_verify_result.is_issued_by_known_root = true;
+  const int kExpectedStatus = ERR_ACCESS_DENIED;
+  EXPECT_CALL(cert_verifier_mock_, Verify(cert_.get(), kHostName, std::string(),
+                                          0, nullptr, _, _, _, _))
       .Times(1)
-      .WillOnce(testing::Return(kExpectedStatus));
+      .WillOnce(testing::DoAll(SetVerifyResult(expected_cert_verify_result),
+                               RunCallback(kExpectedStatus),
+                               testing::Return(ERR_IO_PENDING)));
 
   // Call |Verify|.
-  scoped_ptr<CertVerifyResult> actualResult;
-  int actualStatus = -1;
+  CertVerifyResult actual_cert_verify_result;
+  int actual_status = -1;
   CertVerifierBlockAdapter::Params params(cert_.get(), kHostName);
-  Verify(params, &actualResult, &actualStatus);
+  Verify(params, &actual_cert_verify_result, &actual_status);
 
   // Ensure that Verification results are correct.
-  EXPECT_EQ(kExpectedStatus, actualStatus);
-  EXPECT_FALSE(actualResult);
+  EXPECT_EQ(kExpectedStatus, actual_status);
+  EXPECT_EQ(expected_cert_verify_result.is_issued_by_known_root,
+            actual_cert_verify_result.is_issued_by_known_root);
 }
 
 // Tests |Verify| with all params and synchronous verification.
 TEST_F(CertVerifierBlockAdapterTest, AllParamsAndSync) {
   // Set up expectation.
-  net::CertVerifyResult expectedResult;
-  expectedResult.verified_cert = cert_;
-  const int kExpectedStatus = 0;
+  net::CertVerifyResult expected_cert_verify_result;
+  expected_cert_verify_result.verified_cert = cert_;
+  const int kExpectedStatus = OK;
   scoped_refptr<CRLSet> crl_set(CRLSet::EmptyCRLSetForTesting());
-  EXPECT_CALL(*cert_verifier_mock_,
+  EXPECT_CALL(cert_verifier_mock_,
               Verify(cert_.get(), kHostName, kOcspResponse,
                      CertVerifier::VERIFY_EV_CERT, crl_set.get(), _, _, _, _))
       .Times(1)
-      .WillOnce(testing::DoAll(SetVerifyResult(expectedResult),
+      .WillOnce(testing::DoAll(SetVerifyResult(expected_cert_verify_result),
                                testing::Return(kExpectedStatus)));
 
   // Call |Verify|.
-  scoped_ptr<CertVerifyResult> actualResult;
-  int actualStatus = -1;
+  CertVerifyResult actual_cert_verify_result;
+  int actual_status = -1;
   CertVerifierBlockAdapter::Params params(cert_.get(), kHostName);
   params.ocsp_response = kOcspResponse;
   params.flags = CertVerifier::VERIFY_EV_CERT;
   params.crl_set = crl_set;
-  Verify(params, &actualResult, &actualStatus);
+  Verify(params, &actual_cert_verify_result, &actual_status);
 
   // Ensure that Verification results are correct.
-  EXPECT_EQ(kExpectedStatus, actualStatus);
-  EXPECT_EQ(expectedResult.verified_cert, actualResult->verified_cert);
+  EXPECT_EQ(kExpectedStatus, actual_status);
+  EXPECT_EQ(expected_cert_verify_result.verified_cert,
+            actual_cert_verify_result.verified_cert);
 }
 
 }  // namespace