Implement of privet/v3/auth after pairing.

chrome/browser/local_discovery/privetv3_session_unittest.cc

 // 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 "chrome/browser/local_discovery/privetv3_session.h"
 
 #include "base/base64.h"
 #include "base/strings/stringprintf.h"
 #include "chrome/browser/local_discovery/privet_http.h"
 #include "content/public/test/test_utils.h"
 #include "crypto/hmac.h"
 #include "crypto/p224_spake.h"
 #include "net/url_request/test_url_fetcher_factory.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace local_discovery {
 
 namespace {
 
+using testing::InSequence;
 using testing::Invoke;
 using testing::SaveArg;
 using testing::StrictMock;
 using testing::_;
 
 using PairingType = PrivetV3Session::PairingType;
 using Result = PrivetV3Session::Result;
 
 const char kInfoResponse[] =
     "{\"version\":\"3.0\","
@@ skipping 143 matching lines @@
                 base::StringPrintf(
                     "{\"deviceCommitment\":\"%s\",\"sessionId\":\"testId\"}",
                     device_commitment.c_str()),
                 net::HTTP_OK, net::URLRequestStatus::SUCCESS);
           }));
   session_.StartPairing(PairingType::PAIRING_TYPE_EMBEDDEDCODE,
                         base::Bind(&PrivetV3SessionTest::OnPairingStarted,
                                    base::Unretained(this)));
   base::RunLoop().RunUntilIdle();
 
+  EXPECT_TRUE(session_.fingerprint_.empty());
+  EXPECT_EQ("Privet anonymous", session_.privet_auth_token_);
+
   EXPECT_CALL(*this, OnCodeConfirmed(Result::STATUS_SUCCESS)).Times(1);
+  InSequence in_sequence;
   EXPECT_CALL(*this, OnPostData(_))
       .WillOnce(
           testing::Invoke([this, &spake](const base::DictionaryValue& data) {
             std::string commitment_base64;
             EXPECT_TRUE(data.GetString("clientCommitment", &commitment_base64));
             std::string commitment;
             EXPECT_TRUE(base::Base64Decode(commitment_base64, &commitment));
 
             std::string session_id;
             EXPECT_TRUE(data.GetString("sessionId", &session_id));
@@ skipping 17 matching lines @@
             std::string signature_base64;
             base::Base64Encode(signature, &signature_base64);
 
             fetcher_factory_.SetFakeResponse(
                 GURL("http://host/privet/v3/pairing/confirm"),
                 base::StringPrintf(
                     "{\"certFingerprint\":\"%s\",\"certSignature\":\"%s\"}",
                     fingerprint_base64.c_str(), signature_base64.c_str()),
                 net::HTTP_OK, net::URLRequestStatus::SUCCESS);
           }));
+  EXPECT_CALL(*this, OnPostData(_))
+      .WillOnce(
+          testing::Invoke([this, &spake](const base::DictionaryValue& data) {
+            std::string access_token_base64;
+            EXPECT_TRUE(data.GetString("authCode", &access_token_base64));
+            std::string access_token;
+            EXPECT_TRUE(base::Base64Decode(access_token_base64, &access_token));
+
+            crypto::HMAC hmac(crypto::HMAC::SHA256);
+            const std::string& key = spake.GetUnverifiedKey();
+            EXPECT_TRUE(hmac.Init(key));
+            EXPECT_TRUE(hmac.Verify("testId", access_token));
+
+            fetcher_factory_.SetFakeResponse(
+                GURL("http://host/privet/v3/auth"),
+                "{\"accessToken\":\"567\",\"tokenType\":\"testType\","
+                "\"scope\":\"owner\"}",
+                net::HTTP_OK, net::URLRequestStatus::SUCCESS);
+          }));
   session_.ConfirmCode("testPin",
                        base::Bind(&PrivetV3SessionTest::OnCodeConfirmed,
                                   base::Unretained(this)));
   base::RunLoop().RunUntilIdle();
+
+  EXPECT_FALSE(session_.fingerprint_.empty());
+  EXPECT_EQ("testType 567", session_.privet_auth_token_);
 }
 
 // TODO(vitalybuka): replace PrivetHTTPClient with regular URL fetcher and
 // implement SendMessage test.
 
 }  // namespace local_discovery