| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "google_apis/gcm/engine/mcs_client.h" | 5 #include "google_apis/gcm/engine/mcs_client.h" |
| 6 | 6 |
| 7 #include "base/command_line.h" |
| 7 #include "base/files/scoped_temp_dir.h" | 8 #include "base/files/scoped_temp_dir.h" |
| 8 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
| 9 #include "base/run_loop.h" | 10 #include "base/run_loop.h" |
| 10 #include "base/strings/string_number_conversions.h" | 11 #include "base/strings/string_number_conversions.h" |
| 11 #include "base/test/simple_test_clock.h" | 12 #include "base/test/simple_test_clock.h" |
| 12 #include "components/os_crypt/os_crypt.h" | 13 #include "components/os_crypt/os_crypt_switches.h" |
| 13 #include "google_apis/gcm/base/mcs_util.h" | 14 #include "google_apis/gcm/base/mcs_util.h" |
| 14 #include "google_apis/gcm/engine/fake_connection_factory.h" | 15 #include "google_apis/gcm/engine/fake_connection_factory.h" |
| 15 #include "google_apis/gcm/engine/fake_connection_handler.h" | 16 #include "google_apis/gcm/engine/fake_connection_handler.h" |
| 16 #include "google_apis/gcm/engine/gcm_store_impl.h" | 17 #include "google_apis/gcm/engine/gcm_store_impl.h" |
| 17 #include "testing/gtest/include/gtest/gtest.h" | 18 #include "testing/gtest/include/gtest/gtest.h" |
| 18 | 19 |
| 19 namespace gcm { | 20 namespace gcm { |
| 20 | 21 |
| 21 namespace { | 22 namespace { |
| 22 | 23 |
| (...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 77 | 78 |
| 78 private: | 79 private: |
| 79 uint32 next_id_; | 80 uint32 next_id_; |
| 80 }; | 81 }; |
| 81 | 82 |
| 82 class MCSClientTest : public testing::Test { | 83 class MCSClientTest : public testing::Test { |
| 83 public: | 84 public: |
| 84 MCSClientTest(); | 85 MCSClientTest(); |
| 85 virtual ~MCSClientTest(); | 86 virtual ~MCSClientTest(); |
| 86 | 87 |
| 88 virtual void SetUp() OVERRIDE; |
| 89 |
| 87 void BuildMCSClient(); | 90 void BuildMCSClient(); |
| 88 void InitializeClient(); | 91 void InitializeClient(); |
| 89 void StoreCredentials(); | 92 void StoreCredentials(); |
| 90 void LoginClient(const std::vector<std::string>& acknowledged_ids); | 93 void LoginClient(const std::vector<std::string>& acknowledged_ids); |
| 91 | 94 |
| 92 base::SimpleTestClock* clock() { return &clock_; } | 95 base::SimpleTestClock* clock() { return &clock_; } |
| 93 TestMCSClient* mcs_client() const { return mcs_client_.get(); } | 96 TestMCSClient* mcs_client() const { return mcs_client_.get(); } |
| 94 FakeConnectionFactory* connection_factory() { | 97 FakeConnectionFactory* connection_factory() { |
| 95 return &connection_factory_; | 98 return &connection_factory_; |
| 96 } | 99 } |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 137 | 140 |
| 138 MCSClientTest::MCSClientTest() | 141 MCSClientTest::MCSClientTest() |
| 139 : run_loop_(new base::RunLoop()), | 142 : run_loop_(new base::RunLoop()), |
| 140 init_success_(true), | 143 init_success_(true), |
| 141 restored_android_id_(0), | 144 restored_android_id_(0), |
| 142 restored_security_token_(0), | 145 restored_security_token_(0), |
| 143 message_send_status_(MCSClient::SENT) { | 146 message_send_status_(MCSClient::SENT) { |
| 144 EXPECT_TRUE(temp_directory_.CreateUniqueTempDir()); | 147 EXPECT_TRUE(temp_directory_.CreateUniqueTempDir()); |
| 145 run_loop_.reset(new base::RunLoop()); | 148 run_loop_.reset(new base::RunLoop()); |
| 146 | 149 |
| 147 // On OSX, prevent the Keychain permissions popup during unit tests. | |
| 148 #if defined(OS_MACOSX) | |
| 149 OSCrypt::UseMockKeychain(true); | |
| 150 #endif | |
| 151 | |
| 152 // Advance the clock to a non-zero time. | 150 // Advance the clock to a non-zero time. |
| 153 clock_.Advance(base::TimeDelta::FromSeconds(1)); | 151 clock_.Advance(base::TimeDelta::FromSeconds(1)); |
| 154 } | 152 } |
| 155 | 153 |
| 156 MCSClientTest::~MCSClientTest() {} | 154 MCSClientTest::~MCSClientTest() {} |
| 157 | 155 |
| 156 void MCSClientTest::SetUp() { |
| 157 testing::Test::SetUp(); |
| 158 #if defined(OS_MACOSX) |
| 159 base::CommandLine::ForCurrentProcess()->AppendSwitch( |
| 160 os_crypt::switches::kUseMockKeychain); |
| 161 #endif // OS_MACOSX |
| 162 } |
| 163 |
| 158 void MCSClientTest::BuildMCSClient() { | 164 void MCSClientTest::BuildMCSClient() { |
| 159 gcm_store_.reset(new GCMStoreImpl(true, | 165 gcm_store_.reset(new GCMStoreImpl(temp_directory_.path(), |
| 160 temp_directory_.path(), | |
| 161 message_loop_.message_loop_proxy())); | 166 message_loop_.message_loop_proxy())); |
| 162 mcs_client_.reset(new TestMCSClient(&clock_, | 167 mcs_client_.reset(new TestMCSClient(&clock_, |
| 163 &connection_factory_, | 168 &connection_factory_, |
| 164 gcm_store_.get())); | 169 gcm_store_.get())); |
| 165 } | 170 } |
| 166 | 171 |
| 167 void MCSClientTest::InitializeClient() { | 172 void MCSClientTest::InitializeClient() { |
| 168 gcm_store_->Load(base::Bind( | 173 gcm_store_->Load(base::Bind( |
| 169 &MCSClient::Initialize, | 174 &MCSClient::Initialize, |
| 170 base::Unretained(mcs_client_.get()), | 175 base::Unretained(mcs_client_.get()), |
| (...skipping 611 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 782 | 787 |
| 783 LoginClient(std::vector<std::string>()); | 788 LoginClient(std::vector<std::string>()); |
| 784 GetFakeHandler()->ExpectOutgoingMessage(message); | 789 GetFakeHandler()->ExpectOutgoingMessage(message); |
| 785 GetFakeHandler()->ExpectOutgoingMessage(message2); | 790 GetFakeHandler()->ExpectOutgoingMessage(message2); |
| 786 PumpLoop(); | 791 PumpLoop(); |
| 787 } | 792 } |
| 788 | 793 |
| 789 } // namespace | 794 } // namespace |
| 790 | 795 |
| 791 } // namespace gcm | 796 } // namespace gcm |
| OLD | NEW |