CrOS policies: Store device ID with device token

chrome/browser/policy/device_token_fetcher_unittest.cc

 // Copyright (c) 2010 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 <string>
 
 #include "base/file_util.h"
 #include "base/message_loop.h"
 #include "base/scoped_temp_dir.h"
 #include "chrome/browser/browser_thread.h"
@@ skipping 67 matching lines @@
 };
 
 TEST_F(DeviceTokenFetcherTest, IsPending) {
   ASSERT_TRUE(fetcher_->IsTokenPending());
   backend_->AllShouldSucceed();
   EXPECT_CALL(*backend_, ProcessRegisterRequest(_, _, _, _)).Times(1);
   SimulateSuccessfulLoginAndRunPending();
   ASSERT_FALSE(fetcher_->IsTokenPending());
 }
 
+TEST_F(DeviceTokenFetcherTest, StoreAndLoad) {
+  backend_->AllShouldSucceed();
+  EXPECT_CALL(*backend_, ProcessRegisterRequest(_, _, _, _)).Times(1);
+  SimulateSuccessfulLoginAndRunPending();
+  ASSERT_FALSE(fetcher_->IsTokenPending());
+  std::string device_token = fetcher_->GetDeviceToken();
+  std::string device_id = fetcher_->GetDeviceID();
+  ASSERT_NE("", device_id);
+
+  FilePath token_path;
+  GetDeviceTokenPath(fetcher_, &token_path);
+  scoped_refptr<DeviceTokenFetcher> fetcher2(
+      new DeviceTokenFetcher(backend_.get(), token_path));
+  fetcher2->StartFetching();
+  loop_.RunAllPending();
+  ASSERT_EQ(device_id, fetcher2->GetDeviceID());
+  ASSERT_EQ(device_token, fetcher2->GetDeviceToken());
+}
+
 TEST_F(DeviceTokenFetcherTest, SimpleFetchSingleLogin) {
   backend_->AllShouldSucceed();
   EXPECT_CALL(*backend_, ProcessRegisterRequest(_, _, _, _)).Times(1);
   SimulateSuccessfulLoginAndRunPending();
   ASSERT_FALSE(fetcher_->IsTokenPending());
   ASSERT_TRUE(fetcher_->IsTokenValid());
   const std::string token(fetcher_->GetDeviceToken());
   EXPECT_NE("", token);
 }
 
@@ skipping 43 matching lines @@
 TEST_F(DeviceTokenFetcherTest, FailedServerRequest) {
   backend_->AllShouldFail();
   EXPECT_CALL(*backend_, ProcessRegisterRequest(_, _, _, _)).Times(1);
   SimulateSuccessfulLoginAndRunPending();
   ASSERT_FALSE(fetcher_->IsTokenPending());
   const std::string token(fetcher_->GetDeviceToken());
   EXPECT_EQ("", token);
 }
 
 }  // namespace policy