chrome/common/ini_parser_unittest.cc

   1 // Copyright 2014 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 <string>
   6 #include <vector>
   7
   8 #include "chrome/common/ini_parser.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 namespace {
  12
  13 struct TestTriplet {
  14   TestTriplet(const std::string& section,
  15           const std::string& key,
  16           const std::string& value)
  17       : section(section),
  18         key(key),
  19         value(value) {
  20   }
  21
  22   std::string section;
  23   std::string key;
  24   std::string value;
  25 };
  26
  27 class TestINIParser : public INIParser {
  28  public:
  29   explicit TestINIParser(
  30       const std::vector<TestTriplet>& expected_triplets)
  31       : expected_triplets_(expected_triplets),
  32         pair_i_(0) {
  33   }
  34   virtual ~TestINIParser() {}
  35
  36   size_t pair_i() {
  37     return pair_i_;
  38   }
  39
  40  private:
  41   virtual void HandleTriplet(const std::string& section, const std::string& key,
  42                              const std::string& value) OVERRIDE {
  43     EXPECT_EQ(expected_triplets_[pair_i_].section, section);
  44     EXPECT_EQ(expected_triplets_[pair_i_].key, key);
  45     EXPECT_EQ(expected_triplets_[pair_i_].value, value);
  46     ++pair_i_;
  47   }
  48
  49   std::vector<TestTriplet> expected_triplets_;
  50   size_t pair_i_;
  51 };
  52
  53 TEST(INIParserTest, BasicValid) {
  54   std::vector<TestTriplet> expected_triplets;
  55   expected_triplets.push_back(TestTriplet("section1", "key1", "value1"));
  56   expected_triplets.push_back(TestTriplet("section1", "key2", "value2"));
  57   expected_triplets.push_back(TestTriplet("section1", "key3", "value3"));
  58   expected_triplets.push_back(TestTriplet("section2", "key4", "value4"));
  59   expected_triplets.push_back(TestTriplet("section2", "key5",
  60                                           "value=with=equals"));
  61   expected_triplets.push_back(TestTriplet("section2", "key6", "value6"));
  62   TestINIParser test_parser(expected_triplets);
  63
  64   test_parser.Parse(
  65       "[section1]\n"
  66       "key1=value1\n"
  67       "key2=value2\r\n"  // Testing DOS "\r\n" line endings.
  68       "key3=value3\n"
  69       "[section2\n"      // Testing omitted closing bracket.
  70       "key4=value4\r"    // Testing "\r" line endings.
  71       "key5=value=with=equals\n"
  72       "key6=value6");    // Testing omitted final line ending.
  73 }
  74
  75 TEST(INIParserTest, IgnoreBlankLinesAndComments) {
  76   std::vector<TestTriplet> expected_triplets;
  77   expected_triplets.push_back(TestTriplet("section1", "key1", "value1"));
  78   expected_triplets.push_back(TestTriplet("section1", "key2", "value2"));
  79   expected_triplets.push_back(TestTriplet("section1", "key3", "value3"));
  80   expected_triplets.push_back(TestTriplet("section2", "key4", "value4"));
  81   expected_triplets.push_back(TestTriplet("section2", "key5", "value5"));
  82   expected_triplets.push_back(TestTriplet("section2", "key6", "value6"));
  83   TestINIParser test_parser(expected_triplets);
  84
  85   test_parser.Parse(
  86       "\n"
  87       "[section1]\n"
  88       "key1=value1\n"
  89       "\n"
  90       "\n"
  91       "key2=value2\n"
  92       "key3=value3\n"
  93       "\n"
  94       ";Comment1"
  95       "\n"
  96       "[section2]\n"
  97       "key4=value4\n"
  98       "#Comment2\n"
  99       "key5=value5\n"
 100       "\n"
 101       "key6=value6\n");
 102 }
 103
 104 TEST(INIParserTest, DictionaryValueINIParser) {
 105   DictionaryValueINIParser test_parser;
 106
 107   test_parser.Parse(
 108       "[section1]\n"
 109       "key1=value1\n"
 110       "key.2=value2\n"
 111       "key3=va.lue3\n"
 112       "[se.ction2]\n"
 113       "key.4=value4\n"
 114       "key5=value5\n");
 115
 116   const base::DictionaryValue& root = test_parser.root();
 117   std::string value;
 118   EXPECT_TRUE(root.GetString("section1.key1", &value));
 119   EXPECT_EQ("value1", value);
 120   EXPECT_FALSE(root.GetString("section1.key.2", &value));
 121   EXPECT_TRUE(root.GetString("section1.key3", &value));
 122   EXPECT_EQ("va.lue3", value);
 123   EXPECT_FALSE(root.GetString("se.ction2.key.4", &value));
 124   EXPECT_FALSE(root.GetString("se.ction2.key5", &value));
 125 }
 126
 127 }  // namespace