src/chrome/browser/content_settings/content_settings_rule_unittest.cc

   1 // Copyright (c) 2011 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 <list>
   6
   7 #include "chrome/common/content_settings_pattern.h"
   8 #include "chrome/browser/content_settings/content_settings_rule.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 namespace content_settings {
  12
  13 namespace {
  14
  15 class ListIterator : public RuleIterator {
  16  public:
  17   explicit ListIterator(const std::list<Rule>& rules)
  18       : rules_(rules) {}
  19
  20   virtual ~ListIterator() {}
  21
  22   virtual bool HasNext() const OVERRIDE {
  23     return !rules_.empty();
  24   }
  25
  26   virtual Rule Next() OVERRIDE {
  27     EXPECT_FALSE(rules_.empty());
  28     // |front()| returns a reference but we're going to discard the object
  29     // referred to; force copying here.
  30     Rule rule = rules_.front();
  31     rules_.pop_front();
  32     return rule;
  33   }
  34
  35  private:
  36   std::list<Rule> rules_;
  37 };
  38
  39 }
  40
  41 TEST(RuleTest, ConcatenationIterator) {
  42   std::list<Rule> rules1;
  43   rules1.push_back(Rule(ContentSettingsPattern::FromString("a"),
  44                         ContentSettingsPattern::Wildcard(),
  45                         base::Value::CreateIntegerValue(0)));
  46   rules1.push_back(Rule(ContentSettingsPattern::FromString("b"),
  47                         ContentSettingsPattern::Wildcard(),
  48                         base::Value::CreateIntegerValue(0)));
  49   std::list<Rule> rules2;
  50   rules2.push_back(Rule(ContentSettingsPattern::FromString("c"),
  51                         ContentSettingsPattern::Wildcard(),
  52                         base::Value::CreateIntegerValue(0)));
  53   rules2.push_back(Rule(ContentSettingsPattern::FromString("d"),
  54                         ContentSettingsPattern::Wildcard(),
  55                         base::Value::CreateIntegerValue(0)));
  56
  57   ScopedVector<RuleIterator> iterators;
  58   iterators.push_back(new ListIterator(rules1));
  59   iterators.push_back(new ListIterator(rules2));
  60   ConcatenationIterator concatenation_iterator(&iterators, NULL);
  61
  62   Rule rule;
  63   ASSERT_TRUE(concatenation_iterator.HasNext());
  64   rule = concatenation_iterator.Next();
  65   EXPECT_EQ(rule.primary_pattern, ContentSettingsPattern::FromString("a"));
  66
  67   ASSERT_TRUE(concatenation_iterator.HasNext());
  68   rule = concatenation_iterator.Next();
  69   EXPECT_EQ(rule.primary_pattern, ContentSettingsPattern::FromString("b"));
  70
  71   ASSERT_TRUE(concatenation_iterator.HasNext());
  72   rule = concatenation_iterator.Next();
  73   EXPECT_EQ(rule.primary_pattern, ContentSettingsPattern::FromString("c"));
  74
  75   ASSERT_TRUE(concatenation_iterator.HasNext());
  76   rule = concatenation_iterator.Next();
  77   EXPECT_EQ(rule.primary_pattern, ContentSettingsPattern::FromString("d"));
  78
  79   EXPECT_FALSE(concatenation_iterator.HasNext());
  80 }
  81
  82 }  // namespace content_settings