extensions/browser/process_map_unittest.cc

   1 // Copyright 2013 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 "extensions/browser/process_map.h"
   6
   7 #include "testing/gtest/include/gtest/gtest.h"
   8
   9 using extensions::ProcessMap;
  10
  11 TEST(ExtensionProcessMapTest, Test) {
  12   ProcessMap map;
  13
  14   // Test behavior when empty.
  15   EXPECT_FALSE(map.Contains("a", 1));
  16   EXPECT_FALSE(map.Remove("a", 1, 1));
  17   EXPECT_EQ(0u, map.size());
  18
  19   // Test insertion and behavior with one item.
  20   EXPECT_TRUE(map.Insert("a", 1, 1));
  21   EXPECT_TRUE(map.Contains("a", 1));
  22   EXPECT_FALSE(map.Contains("a", 2));
  23   EXPECT_FALSE(map.Contains("b", 1));
  24   EXPECT_EQ(1u, map.size());
  25
  26   // Test inserting a duplicate item.
  27   EXPECT_FALSE(map.Insert("a", 1, 1));
  28   EXPECT_TRUE(map.Contains("a", 1));
  29   EXPECT_EQ(1u, map.size());
  30
  31   // Insert some more items.
  32   EXPECT_TRUE(map.Insert("a", 2, 2));
  33   EXPECT_TRUE(map.Insert("b", 1, 3));
  34   EXPECT_TRUE(map.Insert("b", 2, 4));
  35   EXPECT_EQ(4u, map.size());
  36
  37   EXPECT_TRUE(map.Contains("a", 1));
  38   EXPECT_TRUE(map.Contains("a", 2));
  39   EXPECT_TRUE(map.Contains("b", 1));
  40   EXPECT_TRUE(map.Contains("b", 2));
  41   EXPECT_FALSE(map.Contains("a", 3));
  42
  43   // Note that this only differs from an existing item because of the site
  44   // instance id.
  45   EXPECT_TRUE(map.Insert("a", 1, 5));
  46   EXPECT_TRUE(map.Contains("a", 1));
  47
  48   // Test removal.
  49   EXPECT_TRUE(map.Remove("a", 1, 1));
  50   EXPECT_FALSE(map.Remove("a", 1, 1));
  51   EXPECT_EQ(4u, map.size());
  52
  53   // Should still return true because there were two site instances for this
  54   // extension/process pair.
  55   EXPECT_TRUE(map.Contains("a", 1));
  56
  57   EXPECT_TRUE(map.Remove("a", 1, 5));
  58   EXPECT_EQ(3u, map.size());
  59   EXPECT_FALSE(map.Contains("a", 1));
  60
  61   EXPECT_EQ(2, map.RemoveAllFromProcess(2));
  62   EXPECT_EQ(1u, map.size());
  63   EXPECT_EQ(0, map.RemoveAllFromProcess(2));
  64   EXPECT_EQ(1u, map.size());
  65 }