| Index: google_apis/gcm/engine/rmq_store_unittest.cc
|
| diff --git a/google_apis/gcm/engine/rmq_store_unittest.cc b/google_apis/gcm/engine/rmq_store_unittest.cc
|
| deleted file mode 100644
|
| index a2b098ffd1bb0f11957b9cd0a3223e06a35fd4f1..0000000000000000000000000000000000000000
|
| --- a/google_apis/gcm/engine/rmq_store_unittest.cc
|
| +++ /dev/null
|
| @@ -1,364 +0,0 @@
|
| -// Copyright 2013 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 "google_apis/gcm/engine/rmq_store.h"
|
| -
|
| -#include <string>
|
| -#include <vector>
|
| -
|
| -#include "base/bind.h"
|
| -#include "base/files/file_path.h"
|
| -#include "base/files/scoped_temp_dir.h"
|
| -#include "base/memory/scoped_ptr.h"
|
| -#include "base/message_loop/message_loop.h"
|
| -#include "base/run_loop.h"
|
| -#include "base/strings/string_number_conversions.h"
|
| -#include "components/webdata/encryptor/encryptor.h"
|
| -#include "google_apis/gcm/base/mcs_message.h"
|
| -#include "google_apis/gcm/base/mcs_util.h"
|
| -#include "google_apis/gcm/protocol/mcs.pb.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -
|
| -namespace gcm {
|
| -
|
| -namespace {
|
| -
|
| -// Number of persistent ids to use in tests.
|
| -const int kNumPersistentIds = 10;
|
| -
|
| -const uint64 kDeviceId = 22;
|
| -const uint64 kDeviceToken = 55;
|
| -
|
| -class RMQStoreTest : public testing::Test {
|
| - public:
|
| - RMQStoreTest();
|
| - virtual ~RMQStoreTest();
|
| -
|
| - scoped_ptr<RMQStore> BuildRMQStore();
|
| -
|
| - std::string GetNextPersistentId();
|
| -
|
| - void PumpLoop();
|
| -
|
| - void LoadCallback(RMQStore::LoadResult* result_dst,
|
| - const RMQStore::LoadResult& result);
|
| - void UpdateCallback(bool success);
|
| -
|
| - private:
|
| - base::MessageLoop message_loop_;
|
| - base::ScopedTempDir temp_directory_;
|
| - scoped_ptr<base::RunLoop> run_loop_;
|
| -};
|
| -
|
| -RMQStoreTest::RMQStoreTest() {
|
| - EXPECT_TRUE(temp_directory_.CreateUniqueTempDir());
|
| - run_loop_.reset(new base::RunLoop());
|
| -
|
| - // On OSX, prevent the Keychain permissions popup during unit tests.
|
| - #if defined(OS_MACOSX)
|
| - Encryptor::UseMockKeychain(true);
|
| - #endif
|
| -}
|
| -
|
| -RMQStoreTest::~RMQStoreTest() {
|
| -}
|
| -
|
| -scoped_ptr<RMQStore> RMQStoreTest::BuildRMQStore() {
|
| - return scoped_ptr<RMQStore>(new RMQStore(temp_directory_.path(),
|
| - message_loop_.message_loop_proxy()));
|
| -}
|
| -
|
| -std::string RMQStoreTest::GetNextPersistentId() {
|
| - return base::Uint64ToString(base::Time::Now().ToInternalValue());
|
| -}
|
| -
|
| -void RMQStoreTest::PumpLoop() {
|
| - message_loop_.RunUntilIdle();
|
| -}
|
| -
|
| -void RMQStoreTest::LoadCallback(RMQStore::LoadResult* result_dst,
|
| - const RMQStore::LoadResult& result) {
|
| - ASSERT_TRUE(result.success);
|
| - *result_dst = result;
|
| - run_loop_->Quit();
|
| - run_loop_.reset(new base::RunLoop());
|
| -}
|
| -
|
| -void RMQStoreTest::UpdateCallback(bool success) {
|
| - ASSERT_TRUE(success);
|
| -}
|
| -
|
| -// Verify creating a new database and loading it.
|
| -TEST_F(RMQStoreTest, LoadNew) {
|
| - scoped_ptr<RMQStore> rmq_store(BuildRMQStore());
|
| - RMQStore::LoadResult load_result;
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - EXPECT_EQ(0U, load_result.device_android_id);
|
| - EXPECT_EQ(0U, load_result.device_security_token);
|
| - EXPECT_TRUE(load_result.incoming_messages.empty());
|
| - EXPECT_TRUE(load_result.outgoing_messages.empty());
|
| - EXPECT_EQ(1LL, load_result.next_serial_number);
|
| - EXPECT_TRUE(load_result.user_serial_numbers.empty());
|
| -}
|
| -
|
| -TEST_F(RMQStoreTest, DeviceCredentials) {
|
| - scoped_ptr<RMQStore> rmq_store(BuildRMQStore());
|
| - RMQStore::LoadResult load_result;
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - rmq_store->SetDeviceCredentials(kDeviceId,
|
| - kDeviceToken,
|
| - base::Bind(&RMQStoreTest::UpdateCallback,
|
| - base::Unretained(this)));
|
| - PumpLoop();
|
| -
|
| - rmq_store = BuildRMQStore().Pass();
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - ASSERT_EQ(kDeviceId, load_result.device_android_id);
|
| - ASSERT_EQ(kDeviceToken, load_result.device_security_token);
|
| -}
|
| -
|
| -// Verify saving some incoming messages, reopening the directory, and then
|
| -// removing those incoming messages.
|
| -TEST_F(RMQStoreTest, IncomingMessages) {
|
| - scoped_ptr<RMQStore> rmq_store(BuildRMQStore());
|
| - RMQStore::LoadResult load_result;
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - std::vector<std::string> persistent_ids;
|
| - for (int i = 0; i < kNumPersistentIds; ++i) {
|
| - persistent_ids.push_back(GetNextPersistentId());
|
| - rmq_store->AddIncomingMessage(persistent_ids.back(),
|
| - base::Bind(&RMQStoreTest::UpdateCallback,
|
| - base::Unretained(this)));
|
| - PumpLoop();
|
| - }
|
| -
|
| - rmq_store = BuildRMQStore().Pass();
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - ASSERT_EQ(persistent_ids, load_result.incoming_messages);
|
| - ASSERT_TRUE(load_result.outgoing_messages.empty());
|
| -
|
| - rmq_store->RemoveIncomingMessages(persistent_ids,
|
| - base::Bind(&RMQStoreTest::UpdateCallback,
|
| - base::Unretained(this)));
|
| - PumpLoop();
|
| -
|
| - rmq_store = BuildRMQStore().Pass();
|
| - load_result.incoming_messages.clear();
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - ASSERT_TRUE(load_result.incoming_messages.empty());
|
| - ASSERT_TRUE(load_result.outgoing_messages.empty());
|
| -}
|
| -
|
| -// Verify saving some outgoing messages, reopening the directory, and then
|
| -// removing those outgoing messages.
|
| -TEST_F(RMQStoreTest, OutgoingMessages) {
|
| - scoped_ptr<RMQStore> rmq_store(BuildRMQStore());
|
| - RMQStore::LoadResult load_result;
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - std::vector<std::string> persistent_ids;
|
| - const int kNumPersistentIds = 10;
|
| - for (int i = 0; i < kNumPersistentIds; ++i) {
|
| - persistent_ids.push_back(GetNextPersistentId());
|
| - mcs_proto::DataMessageStanza message;
|
| - message.set_from(persistent_ids.back());
|
| - message.set_category(persistent_ids.back());
|
| - rmq_store->AddOutgoingMessage(persistent_ids.back(),
|
| - MCSMessage(message),
|
| - base::Bind(&RMQStoreTest::UpdateCallback,
|
| - base::Unretained(this)));
|
| - PumpLoop();
|
| - }
|
| -
|
| - rmq_store = BuildRMQStore().Pass();
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - ASSERT_TRUE(load_result.incoming_messages.empty());
|
| - ASSERT_EQ(load_result.outgoing_messages.size(), persistent_ids.size());
|
| - for (int i =0 ; i < kNumPersistentIds; ++i) {
|
| - std::string id = persistent_ids[i];
|
| - ASSERT_TRUE(load_result.outgoing_messages[id]);
|
| - const mcs_proto::DataMessageStanza* message =
|
| - reinterpret_cast<mcs_proto::DataMessageStanza *>(
|
| - load_result.outgoing_messages[id]);
|
| - ASSERT_EQ(message->from(), id);
|
| - ASSERT_EQ(message->category(), id);
|
| - }
|
| -
|
| - rmq_store->RemoveOutgoingMessages(persistent_ids,
|
| - base::Bind(&RMQStoreTest::UpdateCallback,
|
| - base::Unretained(this)));
|
| - PumpLoop();
|
| -
|
| - rmq_store = BuildRMQStore().Pass();
|
| - load_result.outgoing_messages.clear();
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - ASSERT_TRUE(load_result.incoming_messages.empty());
|
| - ASSERT_TRUE(load_result.outgoing_messages.empty());
|
| -}
|
| -
|
| -// Verify incoming and outgoing messages don't conflict.
|
| -TEST_F(RMQStoreTest, IncomingAndOutgoingMessages) {
|
| - scoped_ptr<RMQStore> rmq_store(BuildRMQStore());
|
| - RMQStore::LoadResult load_result;
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - std::vector<std::string> persistent_ids;
|
| - const int kNumPersistentIds = 10;
|
| - for (int i = 0; i < kNumPersistentIds; ++i) {
|
| - persistent_ids.push_back(GetNextPersistentId());
|
| - rmq_store->AddIncomingMessage(persistent_ids.back(),
|
| - base::Bind(&RMQStoreTest::UpdateCallback,
|
| - base::Unretained(this)));
|
| - PumpLoop();
|
| -
|
| - mcs_proto::DataMessageStanza message;
|
| - message.set_from(persistent_ids.back());
|
| - message.set_category(persistent_ids.back());
|
| - rmq_store->AddOutgoingMessage(persistent_ids.back(),
|
| - MCSMessage(message),
|
| - base::Bind(&RMQStoreTest::UpdateCallback,
|
| - base::Unretained(this)));
|
| - PumpLoop();
|
| - }
|
| -
|
| -
|
| - rmq_store = BuildRMQStore().Pass();
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - ASSERT_EQ(persistent_ids, load_result.incoming_messages);
|
| - ASSERT_EQ(load_result.outgoing_messages.size(), persistent_ids.size());
|
| - for (int i =0 ; i < kNumPersistentIds; ++i) {
|
| - std::string id = persistent_ids[i];
|
| - ASSERT_TRUE(load_result.outgoing_messages[id]);
|
| - const mcs_proto::DataMessageStanza* message =
|
| - reinterpret_cast<mcs_proto::DataMessageStanza *>(
|
| - load_result.outgoing_messages[id]);
|
| - ASSERT_EQ(message->from(), id);
|
| - ASSERT_EQ(message->category(), id);
|
| - }
|
| -
|
| - rmq_store->RemoveIncomingMessages(persistent_ids,
|
| - base::Bind(&RMQStoreTest::UpdateCallback,
|
| - base::Unretained(this)));
|
| - PumpLoop();
|
| - rmq_store->RemoveOutgoingMessages(persistent_ids,
|
| - base::Bind(&RMQStoreTest::UpdateCallback,
|
| - base::Unretained(this)));
|
| - PumpLoop();
|
| -
|
| - rmq_store = BuildRMQStore().Pass();
|
| - load_result.incoming_messages.clear();
|
| - load_result.outgoing_messages.clear();
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - ASSERT_TRUE(load_result.incoming_messages.empty());
|
| - ASSERT_TRUE(load_result.outgoing_messages.empty());
|
| -}
|
| -
|
| -TEST_F(RMQStoreTest, NextSerialNumber) {
|
| - const int64 kNextSerialNumber = 77LL;
|
| - scoped_ptr<RMQStore> rmq_store(BuildRMQStore());
|
| - RMQStore::LoadResult load_result;
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - rmq_store->SetNextSerialNumber(kNextSerialNumber,
|
| - base::Bind(&RMQStoreTest::UpdateCallback,
|
| - base::Unretained(this)));
|
| - PumpLoop();
|
| -
|
| - rmq_store = BuildRMQStore().Pass();
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - EXPECT_EQ(kNextSerialNumber, load_result.next_serial_number);
|
| -}
|
| -
|
| -TEST_F(RMQStoreTest, UserSerialNumberMappings) {
|
| - scoped_ptr<RMQStore> rmq_store(BuildRMQStore());
|
| - RMQStore::LoadResult load_result;
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - std::string username1 = "username1";
|
| - int64 serial_number1 = 34LL;
|
| - rmq_store->AddUserSerialNumber(username1, serial_number1,
|
| - base::Bind(&RMQStoreTest::UpdateCallback,
|
| - base::Unretained(this)));
|
| -
|
| - std::string username2 = "username2";
|
| - int64 serial_number2 = 56LL;
|
| - rmq_store->AddUserSerialNumber(username2, serial_number2,
|
| - base::Bind(&RMQStoreTest::UpdateCallback,
|
| - base::Unretained(this)));
|
| - PumpLoop();
|
| -
|
| - rmq_store = BuildRMQStore().Pass();
|
| - rmq_store->Load(base::Bind(&RMQStoreTest::LoadCallback,
|
| - base::Unretained(this),
|
| - &load_result));
|
| - PumpLoop();
|
| -
|
| - ASSERT_EQ(2u, load_result.user_serial_numbers.size());
|
| - ASSERT_NE(load_result.user_serial_numbers.end(),
|
| - load_result.user_serial_numbers.find(username1));
|
| - EXPECT_EQ(serial_number1, load_result.user_serial_numbers[username1]);
|
| - ASSERT_NE(load_result.user_serial_numbers.end(),
|
| - load_result.user_serial_numbers.find(username2));
|
| - EXPECT_EQ(serial_number2, load_result.user_serial_numbers[username2]);
|
| -}
|
| -
|
| -} // namespace
|
| -
|
| -} // namespace gcm
|
|
|