Support different Google attestation (Privacy CA) servers.

chrome/browser/chromeos/attestation/attestation_ca_client_unittest.cc

Index: chrome/browser/chromeos/attestation/attestation_ca_client_unittest.cc
diff --git a/chrome/browser/chromeos/attestation/attestation_ca_client_unittest.cc b/chrome/browser/chromeos/attestation/attestation_ca_client_unittest.cc
index b5de15266d0b1d3417cfa686cc9d08de1444f2ee..e5533c1288c211c58c58616b447a318fc3666c66 100644
--- a/chrome/browser/chromeos/attestation/attestation_ca_client_unittest.cc
+++ b/chrome/browser/chromeos/attestation/attestation_ca_client_unittest.cc
@@ -3,8 +3,10 @@
 // found in the LICENSE file.
 
 #include "base/bind.h"
+#include "base/command_line.h"
 #include "base/message_loop/message_loop.h"
 #include "chrome/browser/chromeos/attestation/attestation_ca_client.h"
+#include "chromeos/chromeos_switches.h"
 #include "content/public/test/test_browser_thread.h"
 #include "net/base/net_errors.h"
 #include "net/http/http_status_code.h"
@@ -26,15 +28,15 @@ class AttestationCAClientTest : public ::testing::Test {
 
   ~AttestationCAClientTest() override {}
 
-  void DataCallback (bool result, const std::string& data) {
+  void DataCallback(bool result, const std::string& data) {
     ++num_invocations_;
     result_ = result;
     data_ = data;
   }
 
-  void DeleteClientDataCallback (AttestationCAClient* client,
-                                 bool result,
-                                 const std::string& data) {
+  void DeleteClientDataCallback(AttestationCAClient* client,
+                                bool result,
+                                const std::string& data) {
     delete client;
     DataCallback(result, data);
   }
@@ -125,5 +127,60 @@ TEST_F(AttestationCAClientTest, DeleteOnCallback) {
   EXPECT_EQ("certificate_response", data_);
 }
 
+class AttestationCAClientAttestationServerTest
+    : public AttestationCAClientTest {};
+
+TEST_F(AttestationCAClientAttestationServerTest, DefaultEnrollRequest) {
+  base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(
+      chromeos::switches::kAttestationServer, "default");
+  AttestationCAClient client;
+  client.SendEnrollRequest("enroll",
+                           base::Bind(&AttestationCAClientTest::DataCallback,
+                                      base::Unretained(this)));
+  net::TestURLFetcher* fetcher = url_fetcher_factory_.GetFetcherByID(0);
+  CHECK(fetcher);
+  EXPECT_EQ(GURL("https://chromeos-ca.gstatic.com/enroll"),
+            fetcher->GetOriginalURL());
+}
+
+TEST_F(AttestationCAClientAttestationServerTest, DefaultCertificateRequest) {
+  base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(
+      chromeos::switches::kAttestationServer, "default");
+  AttestationCAClient client;
+  client.SendCertificateRequest(
+      "certificate", base::Bind(&AttestationCAClientTest::DataCallback,
+                                base::Unretained(this)));
+  net::TestURLFetcher* fetcher = url_fetcher_factory_.GetFetcherByID(0);
+  CHECK(fetcher);
+  EXPECT_EQ(GURL("https://chromeos-ca.gstatic.com/sign"),
+            fetcher->GetOriginalURL());
+}
+
+TEST_F(AttestationCAClientAttestationServerTest, TestEnrollRequest) {
+  base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(
+      chromeos::switches::kAttestationServer, "test");
+  AttestationCAClient client;
+  client.SendEnrollRequest("enroll",
+                           base::Bind(&AttestationCAClientTest::DataCallback,
+                                      base::Unretained(this)));
+  net::TestURLFetcher* fetcher = url_fetcher_factory_.GetFetcherByID(0);
+  CHECK(fetcher);
+  EXPECT_EQ(GURL("https://asbestos-qa.corp.google.com/enroll"),
+            fetcher->GetOriginalURL());
+}
+
+TEST_F(AttestationCAClientAttestationServerTest, TestCertificateRequest) {
+  base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(
+      chromeos::switches::kAttestationServer, "test");
+  AttestationCAClient client;
+  client.SendCertificateRequest(
+      "certificate", base::Bind(&AttestationCAClientTest::DataCallback,
+                                base::Unretained(this)));
+  net::TestURLFetcher* fetcher = url_fetcher_factory_.GetFetcherByID(0);
+  CHECK(fetcher);
+  EXPECT_EQ(GURL("https://asbestos-qa.corp.google.com/sign"),
+            fetcher->GetOriginalURL());
+}
+
 }  // namespace attestation
 }  // namespace chromeos