Google OAuth Device Flow support for FNL

services/authentication/accounts_db_manager_unittest.cc

+// Copyright 2015 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 "accounts_db_manager.h"
+
+#include "base/logging.h"
+#include "base/strings/string_tokenizer.h"
+#include "mojo/public/cpp/application/application_impl.h"
+#include "mojo/public/cpp/application/application_test_base.h"
+#include "mojo/services/files/interfaces/types.mojom.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace authentication {
+namespace {
+
+class AccountsDBTest : public mojo::test::ApplicationTestBase {
+ public:
+  AccountsDBTest(){};
+  ~AccountsDBTest() override{};
+
+ protected:
+  void SetUp() override {
+    mojo::test::ApplicationTestBase::SetUp();
+    mojo::files::FilesPtr files;
+    application_impl()->ConnectToService("mojo:files", &files);
+
+    mojo::files::Error error = mojo::files::Error::INTERNAL;
+    mojo::files::DirectoryPtr directory;
+    files->OpenFileSystem(nullptr, GetProxy(&directory), Capture(&error));
+    files.WaitForIncomingResponse();
+    if (mojo::files::Error::OK != error) {
+      LOG(FATAL) << "Unable to initialize accounts DB";
+    }
+    accounts_db_manager_ = new AccountsDbManager(directory);
+  }
+
+  AccountsDbManager* accountsDBPtr() { return accounts_db_manager_; }
+
+ private:
+  AccountsDbManager* accounts_db_manager_;
+
+  DISALLOW_COPY_AND_ASSIGN(AccountsDBTest);
+};
+
+TEST_F(AccountsDBTest, CanAddNewAccount) {
+  std::string test_str("test:abcdefgh");
+  bool result = accountsDBPtr()->UpdateAccount("test", test_str);
+  ASSERT_TRUE(result);
+
+  const mojo::Array<uint8_t>& accounts = accountsDBPtr()->FetchAllAccounts();
+  const std::vector<uint8_t> vec = accounts.storage();
+  std::string db_contents(vec.begin(), vec.end());
+
+  // db_contents adds extra '\0' to terminate the file contents'
+  EXPECT_EQ(test_str.size(), db_contents.size() - 1);
+}
+
+TEST_F(AccountsDBTest, CanUpdateAnExistingAccount) {
+  bool result = accountsDBPtr()->UpdateAccount("user1", "user1:hello");
+  ASSERT_TRUE(result);
+
+  result = accountsDBPtr()->UpdateAccount("user1", "user1:world");
+  ASSERT_TRUE(result);
+
+  result = accountsDBPtr()->UpdateAccount("user3", "user3:google");
+  ASSERT_TRUE(result);
+
+  const mojo::Array<uint8_t>& accounts = accountsDBPtr()->FetchAllAccounts();
+  const std::vector<uint8_t> vec = accounts.storage();
+  std::string db_contents(vec.begin(), vec.end());
+
+  LOG(INFO) << "Db contents are: " << db_contents;
+  base::StringTokenizer lines(db_contents, "\n");
+  int n = 0;
+  while (lines.GetNext()) {
+    n++;
+  }
+  ASSERT_FALSE(db_contents.find("hello") != std::string::npos);
+  ASSERT_TRUE(db_contents.find("world") != std::string::npos);
+  EXPECT_EQ(2, n);
+}
+
+TEST_F(AccountsDBTest, CanGetUserAccountDataByName) {
+  bool result =
+      accountsDBPtr()->UpdateAccount("test_user", "test_user:turquoise_creds");
+  ASSERT_TRUE(result);
+
+  mojo::String user_data;
+  accountsDBPtr()->GetAccountDataForUser("test_user", user_data);
+
+  ASSERT_TRUE(!user_data.is_null());
+  EXPECT_EQ("test_user:turquoise_creds", user_data);
+}
+
+TEST_F(AccountsDBTest, CanGetAllAccountsData) {
+  bool result = accountsDBPtr()->UpdateAccount("user1", "user1:hello");
+  ASSERT_TRUE(result);
+
+  result = accountsDBPtr()->UpdateAccount("user2", "user2:world");
+  ASSERT_TRUE(result);
+
+  result = accountsDBPtr()->UpdateAccount("user3", "user3:google");
+  ASSERT_TRUE(result);
+
+  const mojo::Array<uint8_t>& accounts = accountsDBPtr()->FetchAllAccounts();
+  const std::vector<uint8_t> vec = accounts.storage();
+  std::string db_contents(vec.begin(), vec.end());
+
+  LOG(INFO) << "Db contents are: " << db_contents;
+  base::StringTokenizer lines(db_contents, "\n");
+  int n = 0;
+  while (lines.GetNext()) {
+    n++;
+  }
+  EXPECT_EQ(3, n);
+}
+
+}  // namespace
+}  // namespace authentication