// Copyright 2017 The Chromium OS 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 <memory>
#include <string>

#include <gtest/gtest.h>

#include "base/logging.h"
#include "libpasswordprovider/password.h"
#include "libpasswordprovider/password_provider.h"

namespace password_provider {

// Saving and retrieving password should succeed.
TEST(PasswordProvider, SaveAndGetPassword) {
  const std::string kPasswordStr("thepassword");
  Password password;
  ASSERT_TRUE(password.Init());
  memcpy(password.GetMutableRaw(), kPasswordStr.c_str(), kPasswordStr.size());
  password.SetSize(kPasswordStr.size());

  EXPECT_TRUE(SavePassword(password));
  std::unique_ptr<Password> retrieved_password = GetPassword();
  ASSERT_TRUE(retrieved_password);
  EXPECT_EQ(std::string(retrieved_password->GetRaw()), kPasswordStr);
  EXPECT_EQ(retrieved_password->size(), kPasswordStr.size());
}

// Reading password should fail if password was already discarded.
TEST(PasswordProvider, DiscardAndGetPassword) {
  const std::string kPasswordStr("thepassword");
  Password password;
  ASSERT_TRUE(password.Init());
  memcpy(password.GetMutableRaw(), kPasswordStr.c_str(), kPasswordStr.size());
  password.SetSize(kPasswordStr.size());

  EXPECT_TRUE(SavePassword(password));
  EXPECT_TRUE(DiscardPassword());
  std::unique_ptr<Password> retrieved_password = GetPassword();
  EXPECT_FALSE(retrieved_password);
}

// Retrieving a very long password should succeed.
TEST(PasswordProvider, GetLongPassword) {
  Password password;
  ASSERT_TRUE(password.Init());

  // Create a very long password
  auto long_password = std::make_unique<char[]>(password.max_size());
  memset(long_password.get(), 'a', password.max_size());
  std::string password_str(long_password.get());
  memcpy(password.GetMutableRaw(), long_password.get(), password_str.size());
  password.SetSize(password_str.size());

  EXPECT_TRUE(SavePassword(password));
  std::unique_ptr<Password> retrieved_password = GetPassword();
  ASSERT_TRUE(retrieved_password);
  EXPECT_EQ(std::string(retrieved_password->GetRaw()), password_str);
}

}  // namespace password_provider