// 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();

  ASSERT_EQ(0U, load_result.device_android_id);
  ASSERT_EQ(0U, load_result.device_security_token);
  ASSERT_TRUE(load_result.incoming_messages.empty());
  ASSERT_TRUE(load_result.outgoing_messages.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());
}

}  // namespace

}  // namespace gcm