| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/omaha_query_params/omaha_query_params.h" | |
| 6 | |
| 7 #include "base/strings/stringprintf.h" | 5 #include "base/strings/stringprintf.h" |
| 8 #include "chrome/common/chrome_version_info.h" | 6 #include "components/omaha_query_params/omaha_query_params.h" |
| 9 #include "testing/gtest/include/gtest/gtest.h" | 7 #include "testing/gtest/include/gtest/gtest.h" |
| 10 | 8 |
| 11 using base::StringPrintf; | 9 using base::StringPrintf; |
| 12 | 10 |
| 13 namespace chrome { | 11 namespace { |
| 14 | 12 |
| 15 bool Contains(const std::string& source, const std::string& target) { | 13 bool Contains(const std::string& source, const std::string& target) { |
| 16 return source.find(target) != std::string::npos; | 14 return source.find(target) != std::string::npos; |
| 17 } | 15 } |
| 18 | 16 |
| 19 void TestParams(OmahaQueryParams::ProdId prod_id) { | 17 class TestOmahaQueryParamsDelegate : public OmahaQueryParamsDelegate { |
| 18 virtual std::string GetExtraParams() OVERRIDE { return "&cat=dog"; } |
| 19 }; |
| 20 |
| 21 } // namespace |
| 22 |
| 23 void TestParams(OmahaQueryParams::ProdId prod_id, bool extra_params) { |
| 20 std::string params = OmahaQueryParams::Get(prod_id); | 24 std::string params = OmahaQueryParams::Get(prod_id); |
| 21 | 25 |
| 22 // This doesn't so much test what the values are (since that would be an | 26 // This doesn't so much test what the values are (since that would be an |
| 23 // almost exact duplication of code with omaha_query_params.cc, and wouldn't | 27 // almost exact duplication of code with omaha_query_params.cc, and wouldn't |
| 24 // really test anything) as it is a verification that all the params are | 28 // really test anything) as it is a verification that all the params are |
| 25 // present in the generated string. | 29 // present in the generated string. |
| 26 EXPECT_TRUE( | 30 EXPECT_TRUE( |
| 27 Contains(params, StringPrintf("os=%s", OmahaQueryParams::GetOS()))); | 31 Contains(params, StringPrintf("os=%s", OmahaQueryParams::GetOS()))); |
| 28 EXPECT_TRUE( | 32 EXPECT_TRUE( |
| 29 Contains(params, StringPrintf("arch=%s", OmahaQueryParams::GetArch()))); | 33 Contains(params, StringPrintf("arch=%s", OmahaQueryParams::GetArch()))); |
| 30 EXPECT_TRUE(Contains( | 34 EXPECT_TRUE(Contains( |
| 31 params, | 35 params, |
| 32 StringPrintf("prod=%s", OmahaQueryParams::GetProdIdString(prod_id)))); | 36 StringPrintf("prod=%s", OmahaQueryParams::GetProdIdString(prod_id)))); |
| 33 EXPECT_TRUE(Contains( | 37 if (extra_params) |
| 34 params, | 38 EXPECT_TRUE(Contains(params, "cat=dog")); |
| 35 StringPrintf("prodchannel=%s", OmahaQueryParams::GetChannelString()))); | |
| 36 EXPECT_TRUE(Contains( | |
| 37 params, | |
| 38 StringPrintf("prodversion=%s", chrome::VersionInfo().Version().c_str()))); | |
| 39 EXPECT_TRUE(Contains( | |
| 40 params, | |
| 41 StringPrintf("lang=%s", OmahaQueryParams::GetLang()))); | |
| 42 } | 39 } |
| 43 | 40 |
| 44 TEST(OmahaQueryParams, GetOmahaQueryParams) { | 41 TEST(OmahaQueryParamsTest, GetParams) { |
| 45 TestParams(OmahaQueryParams::CRX); | 42 TestParams(OmahaQueryParams::CRX, false); |
| 46 TestParams(OmahaQueryParams::CHROME); | 43 TestParams(OmahaQueryParams::CHROME, false); |
| 44 |
| 45 TestOmahaQueryParamsDelegate delegate; |
| 46 OmahaQueryParamsDelegate::Set(&delegate); |
| 47 |
| 48 TestParams(OmahaQueryParams::CRX, true); |
| 49 TestParams(OmahaQueryParams::CHROME, true); |
| 47 } | 50 } |
| 48 | |
| 49 } // namespace chrome | |
| OLD | NEW |