Add support for Omaha cohorts to the component updater.

components/update_client/update_response_unittest.cc

 // Copyright 2013 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 "components/update_client/update_response.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace update_client {
 
 const char* kValidXml =
@@ skipping 202 matching lines @@
     "     </urls>"
     "     <manifest version='1.2.3.4' prodversionmin='2.0.143.0'>"
     "       <packages>"
     "         <package name='extension_1_2_3_4.crx'/>"
     "       </packages>"
     "     </manifest>"
     "   </updatecheck>"
     " </app>"
     "</response>";
 
+// Includes two <app> tags, both of which set the cohort.
+const char* kTwoAppsSetCohort =
+    "<?xml version='1.0' encoding='UTF-8'?>"
+    "<response protocol='3.0'>"
+    " <app appid='aaaaaaaa' cohort='1:2q3/'>"
+    "  <updatecheck status='noupdate'/>"
+    " </app>"
+    " <app appid='bbbbbbbb' cohort='1:33z@0.33' cohortname='cname'>"
+    "   <updatecheck status='ok'>"
+    "     <urls>"
+    "       <url codebase='http://example.com/'/>"
+    "     </urls>"
+    "     <manifest version='1.2.3.4' prodversionmin='2.0.143.0'>"
+    "       <packages>"
+    "         <package name='extension_1_2_3_4.crx'/>"
+    "       </packages>"
+    "     </manifest>"
+    "   </updatecheck>"
+    " </app>"
+    "</response>";
+
 TEST(ComponentUpdaterUpdateResponseTest, TestParser) {
   UpdateResponse parser;
 
   // Test parsing of a number of invalid xml cases
   EXPECT_FALSE(parser.Parse(std::string()));
   EXPECT_FALSE(parser.errors().empty());
 
   EXPECT_TRUE(parser.Parse(kMissingAppId));
   EXPECT_TRUE(parser.results().list.empty());
   EXPECT_FALSE(parser.errors().empty());
@@ skipping 72 matching lines @@
   firstResult = &parser.results().list[0];
   EXPECT_EQ(firstResult->extension_id, "12345");
   EXPECT_EQ(firstResult->manifest.version, "");
 
   // Parse xml with one error and one success <app> tag.
   EXPECT_TRUE(parser.Parse(kTwoAppsOneError));
   EXPECT_FALSE(parser.errors().empty());
   EXPECT_EQ(1u, parser.results().list.size());
   firstResult = &parser.results().list[0];
   EXPECT_EQ(firstResult->extension_id, "bbbbbbbb");
+
+  // Parse xml with two apps setting the cohort info.
+  EXPECT_TRUE(parser.Parse(kTwoAppsSetCohort));
+  EXPECT_TRUE(parser.errors().empty());
+  EXPECT_EQ(2u, parser.results().list.size());
+  firstResult = &parser.results().list[0];
+  EXPECT_EQ(firstResult->extension_id, "aaaaaaaa");
+  EXPECT_NE(firstResult->cohort_attrs.find("cohort"),
+            firstResult->cohort_attrs.end());
+  EXPECT_EQ(firstResult->cohort_attrs.find("cohort")->second, "1:2q3/");
+  EXPECT_EQ(firstResult->cohort_attrs.find("cohortname"),
+            firstResult->cohort_attrs.end());
+  EXPECT_EQ(firstResult->cohort_attrs.find("cohorthint"),
+            firstResult->cohort_attrs.end());
+  const UpdateResponse::Result* secondResult = &parser.results().list[1];
+  EXPECT_EQ(secondResult->extension_id, "bbbbbbbb");
+  EXPECT_NE(secondResult->cohort_attrs.find("cohort"),
+            secondResult->cohort_attrs.end());
+  EXPECT_EQ(secondResult->cohort_attrs.find("cohort")->second, "1:33z@0.33");
+  EXPECT_NE(secondResult->cohort_attrs.find("cohortname"),
+            secondResult->cohort_attrs.end());
+  EXPECT_EQ(secondResult->cohort_attrs.find("cohortname")->second, "cname");
+  EXPECT_EQ(secondResult->cohort_attrs.find("cohorthint"),
+            secondResult->cohort_attrs.end());
 }
 
 }  // namespace update_client