chrome/browser/extensions/signin/gaia_auth_extension_loader_browsertest.cc

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/extensions/signin/gaia_auth_extension_loader.h"
   6 #include "chrome/browser/profiles/profile.h"
   7 #include "chrome/browser/ui/browser.h"
   8 #include "chrome/test/base/in_process_browser_test.h"
   9
  10 namespace extensions {
  11
  12 namespace {
  13
  14 const char kTestData1[] = "A test string";
  15 const char kTestData2[] = "Another test string";
  16
  17 }  // namespace
  18
  19 typedef InProcessBrowserTest GaiaAuthExtensionLoaderTest;
  20
  21 IN_PROC_BROWSER_TEST_F(GaiaAuthExtensionLoaderTest, AddAndGet) {
  22
  23   GaiaAuthExtensionLoader* loader = GaiaAuthExtensionLoader::Get(
  24       browser()->profile());
  25   loader->LoadIfNeeded();
  26
  27   int id1 = loader->AddData(kTestData1);
  28   int id2 = loader->AddData(kTestData2);
  29   EXPECT_NE(id1, id2);
  30
  31   std::string fetched;
  32   EXPECT_TRUE(loader->GetData(id1, &fetched));
  33   EXPECT_EQ(kTestData1, fetched);
  34
  35   EXPECT_TRUE(loader->GetData(id2, &fetched));
  36   EXPECT_EQ(kTestData2, fetched);
  37
  38   const int kUnknownId = 1234;
  39   EXPECT_FALSE(loader->GetData(kUnknownId, &fetched));
  40
  41   loader->UnloadIfNeeded();
  42 }
  43
  44 IN_PROC_BROWSER_TEST_F(GaiaAuthExtensionLoaderTest, ClearDataOnUnload) {
  45   GaiaAuthExtensionLoader* loader = GaiaAuthExtensionLoader::Get(
  46       browser()->profile());
  47   loader->LoadIfNeeded();
  48
  49   int id = loader->AddData(kTestData1);
  50   std::string fetched;
  51   EXPECT_TRUE(loader->GetData(id, &fetched));
  52
  53   loader->UnloadIfNeeded();
  54   EXPECT_FALSE(loader->GetData(id, &fetched));
  55 }
  56
  57 }  // namespace extensions