Darin Petkov | f2065b4 | 2011-05-17 16:36:27 -0700 | [diff] [blame] | 1 | // Copyright (c) 2011 The Chromium OS Authors. All rights reserved. |
adlr@google.com | 3defe6a | 2009-12-04 20:57:17 +0000 | [diff] [blame] | 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 <stdio.h> |
Darin Petkov | fbb4009 | 2010-07-29 17:05:50 -0700 | [diff] [blame] | 6 | |
adlr@google.com | 3defe6a | 2009-12-04 20:57:17 +0000 | [diff] [blame] | 7 | #include <string> |
Darin Petkov | fbb4009 | 2010-07-29 17:05:50 -0700 | [diff] [blame] | 8 | |
| 9 | #include "base/file_util.h" |
| 10 | #include "gtest/gtest.h" |
adlr@google.com | 3defe6a | 2009-12-04 20:57:17 +0000 | [diff] [blame] | 11 | #include "update_engine/install_plan.h" |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 12 | #include "update_engine/omaha_request_params.h" |
adlr@google.com | 3defe6a | 2009-12-04 20:57:17 +0000 | [diff] [blame] | 13 | #include "update_engine/test_utils.h" |
| 14 | #include "update_engine/utils.h" |
| 15 | |
| 16 | using std::string; |
| 17 | |
| 18 | namespace chromeos_update_engine { |
| 19 | |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 20 | class OmahaRequestDeviceParamsTest : public ::testing::Test { |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 21 | protected: |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 22 | // Return true iff the OmahaRequestDeviceParams::Init succeeded. If |
| 23 | // out is non-NULL, it's set w/ the generated data. |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 24 | bool DoTest(OmahaRequestParams* out, const string& app_version, |
| 25 | const string& omaha_url); |
| 26 | |
| 27 | virtual void SetUp() { |
| 28 | ASSERT_EQ(0, System(string("mkdir -p ") + kTestDir + "/etc")); |
| 29 | ASSERT_EQ(0, System(string("mkdir -p ") + kTestDir + |
| 30 | utils::kStatefulPartition + "/etc")); |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 31 | params_.set_root(string("./") + kTestDir); |
Darin Petkov | 10d02dd | 2011-01-10 14:57:39 -0800 | [diff] [blame] | 32 | params_.SetLockDown(false); |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 33 | } |
| 34 | |
| 35 | virtual void TearDown() { |
| 36 | EXPECT_EQ(0, System(string("rm -rf ") + kTestDir)); |
| 37 | } |
| 38 | |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 39 | OmahaRequestDeviceParams params_; |
| 40 | |
adlr@google.com | 3defe6a | 2009-12-04 20:57:17 +0000 | [diff] [blame] | 41 | static const string kTestDir; |
| 42 | }; |
| 43 | |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 44 | const string OmahaRequestDeviceParamsTest::kTestDir = |
| 45 | "omaha_request_device_params-test"; |
adlr@google.com | 3defe6a | 2009-12-04 20:57:17 +0000 | [diff] [blame] | 46 | |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 47 | bool OmahaRequestDeviceParamsTest::DoTest(OmahaRequestParams* out, |
| 48 | const string& app_version, |
| 49 | const string& omaha_url) { |
Patrick Dubroy | 7fbbe8a | 2011-08-01 17:28:22 +0200 | [diff] [blame] | 50 | bool success = params_.Init(app_version, omaha_url, ""); |
adlr@google.com | 3defe6a | 2009-12-04 20:57:17 +0000 | [diff] [blame] | 51 | if (out) |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 52 | *out = params_; |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 53 | return success; |
adlr@google.com | 3defe6a | 2009-12-04 20:57:17 +0000 | [diff] [blame] | 54 | } |
| 55 | |
| 56 | namespace { |
adlr@google.com | 3defe6a | 2009-12-04 20:57:17 +0000 | [diff] [blame] | 57 | string GetMachineType() { |
| 58 | FILE* fp = popen("uname -m", "r"); |
| 59 | if (!fp) |
| 60 | return ""; |
| 61 | string ret; |
| 62 | for (;;) { |
| 63 | char buffer[10]; |
| 64 | size_t r = fread(buffer, 1, sizeof(buffer), fp); |
| 65 | if (r == 0) |
| 66 | break; |
| 67 | ret.insert(ret.begin(), buffer, buffer + r); |
| 68 | } |
| 69 | // strip trailing '\n' if it exists |
| 70 | if ((*ret.rbegin()) == '\n') |
| 71 | ret.resize(ret.size() - 1); |
| 72 | fclose(fp); |
| 73 | return ret; |
| 74 | } |
| 75 | } // namespace {} |
| 76 | |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 77 | TEST_F(OmahaRequestDeviceParamsTest, SimpleTest) { |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 78 | ASSERT_TRUE(WriteFileString( |
| 79 | kTestDir + "/etc/lsb-release", |
| 80 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 81 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 82 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 83 | "CHROMEOS_RELEASE_TRACK=footrack\n" |
| 84 | "CHROMEOS_AUSERVER=http://www.google.com")); |
| 85 | OmahaRequestParams out; |
| 86 | EXPECT_TRUE(DoTest(&out, "", "")); |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 87 | EXPECT_EQ("Chrome OS", out.os_platform); |
| 88 | EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp); |
| 89 | EXPECT_EQ("arm-generic", out.os_board); |
| 90 | EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.app_id); |
| 91 | EXPECT_EQ("0.2.2.3", out.app_version); |
| 92 | EXPECT_EQ("en-US", out.app_lang); |
Darin Petkov | fbb4009 | 2010-07-29 17:05:50 -0700 | [diff] [blame] | 93 | EXPECT_EQ("", out.hardware_class); |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 94 | EXPECT_TRUE(out.delta_okay); |
| 95 | EXPECT_EQ("footrack", out.app_track); |
| 96 | EXPECT_EQ("http://www.google.com", out.update_url); |
adlr@google.com | 3defe6a | 2009-12-04 20:57:17 +0000 | [diff] [blame] | 97 | } |
| 98 | |
Darin Petkov | d315dc7 | 2010-11-15 09:52:24 -0800 | [diff] [blame] | 99 | TEST_F(OmahaRequestDeviceParamsTest, AppIDTest) { |
| 100 | ASSERT_TRUE(WriteFileString( |
| 101 | kTestDir + "/etc/lsb-release", |
| 102 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 103 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 104 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 105 | "CHROMEOS_RELEASE_TRACK=footrack\n" |
| 106 | "CHROMEOS_RELEASE_APPID={58c35cef-9d30-476e-9098-ce20377d535d}\n" |
| 107 | "CHROMEOS_AUSERVER=http://www.google.com")); |
| 108 | OmahaRequestParams out; |
| 109 | EXPECT_TRUE(DoTest(&out, "", "")); |
| 110 | EXPECT_EQ("Chrome OS", out.os_platform); |
| 111 | EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp); |
| 112 | EXPECT_EQ("arm-generic", out.os_board); |
| 113 | EXPECT_EQ("{58c35cef-9d30-476e-9098-ce20377d535d}", out.app_id); |
| 114 | EXPECT_EQ("0.2.2.3", out.app_version); |
| 115 | EXPECT_EQ("en-US", out.app_lang); |
| 116 | EXPECT_EQ("", out.hardware_class); |
| 117 | EXPECT_TRUE(out.delta_okay); |
| 118 | EXPECT_EQ("footrack", out.app_track); |
| 119 | EXPECT_EQ("http://www.google.com", out.update_url); |
| 120 | } |
| 121 | |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 122 | TEST_F(OmahaRequestDeviceParamsTest, MissingTrackTest) { |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 123 | ASSERT_TRUE(WriteFileString( |
| 124 | kTestDir + "/etc/lsb-release", |
| 125 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 126 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 127 | "CHROMEOS_RELEASE_TRXCK=footrack")); |
| 128 | OmahaRequestParams out; |
| 129 | EXPECT_TRUE(DoTest(&out, "", "")); |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 130 | EXPECT_EQ("Chrome OS", out.os_platform); |
| 131 | EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp); |
| 132 | EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.app_id); |
| 133 | EXPECT_EQ("0.2.2.3", out.app_version); |
| 134 | EXPECT_EQ("en-US", out.app_lang); |
| 135 | EXPECT_EQ("", out.app_track); |
adlr@google.com | 3defe6a | 2009-12-04 20:57:17 +0000 | [diff] [blame] | 136 | } |
| 137 | |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 138 | TEST_F(OmahaRequestDeviceParamsTest, ConfusingReleaseTest) { |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 139 | ASSERT_TRUE(WriteFileString( |
| 140 | kTestDir + "/etc/lsb-release", |
| 141 | "CHROMEOS_RELEASE_FOO=CHROMEOS_RELEASE_VERSION=1.2.3.4\n" |
| 142 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 143 | "CHROMEOS_RELEASE_TRXCK=footrack")); |
| 144 | OmahaRequestParams out; |
| 145 | EXPECT_TRUE(DoTest(&out, "", "")); |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 146 | EXPECT_EQ("Chrome OS", out.os_platform); |
| 147 | EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp); |
| 148 | EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.app_id); |
| 149 | EXPECT_EQ("0.2.2.3", out.app_version); |
| 150 | EXPECT_EQ("en-US", out.app_lang); |
| 151 | EXPECT_EQ("", out.app_track); |
adlr@google.com | 3defe6a | 2009-12-04 20:57:17 +0000 | [diff] [blame] | 152 | } |
| 153 | |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 154 | TEST_F(OmahaRequestDeviceParamsTest, MissingVersionTest) { |
| 155 | ASSERT_TRUE(WriteFileString( |
| 156 | kTestDir + "/etc/lsb-release", |
| 157 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 158 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 159 | "CHROMEOS_RELEASE_TRACK=footrack")); |
| 160 | OmahaRequestParams out; |
| 161 | EXPECT_TRUE(DoTest(&out, "", "")); |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 162 | EXPECT_EQ("Chrome OS", out.os_platform); |
| 163 | EXPECT_EQ(string("_") + GetMachineType(), out.os_sp); |
| 164 | EXPECT_EQ("arm-generic", out.os_board); |
| 165 | EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.app_id); |
| 166 | EXPECT_EQ("", out.app_version); |
| 167 | EXPECT_EQ("en-US", out.app_lang); |
| 168 | EXPECT_TRUE(out.delta_okay); |
| 169 | EXPECT_EQ("footrack", out.app_track); |
| 170 | } |
| 171 | |
| 172 | TEST_F(OmahaRequestDeviceParamsTest, ForceVersionTest) { |
| 173 | ASSERT_TRUE(WriteFileString( |
| 174 | kTestDir + "/etc/lsb-release", |
| 175 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 176 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 177 | "CHROMEOS_RELEASE_TRACK=footrack")); |
| 178 | OmahaRequestParams out; |
| 179 | EXPECT_TRUE(DoTest(&out, "ForcedVersion", "")); |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 180 | EXPECT_EQ("Chrome OS", out.os_platform); |
| 181 | EXPECT_EQ(string("ForcedVersion_") + GetMachineType(), out.os_sp); |
| 182 | EXPECT_EQ("arm-generic", out.os_board); |
| 183 | EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.app_id); |
| 184 | EXPECT_EQ("ForcedVersion", out.app_version); |
| 185 | EXPECT_EQ("en-US", out.app_lang); |
| 186 | EXPECT_TRUE(out.delta_okay); |
| 187 | EXPECT_EQ("footrack", out.app_track); |
| 188 | } |
| 189 | |
| 190 | TEST_F(OmahaRequestDeviceParamsTest, ForcedURLTest) { |
| 191 | ASSERT_TRUE(WriteFileString( |
| 192 | kTestDir + "/etc/lsb-release", |
| 193 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 194 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 195 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 196 | "CHROMEOS_RELEASE_TRACK=footrack")); |
| 197 | OmahaRequestParams out; |
| 198 | EXPECT_TRUE(DoTest(&out, "", "http://forced.google.com")); |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 199 | EXPECT_EQ("Chrome OS", out.os_platform); |
| 200 | EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp); |
| 201 | EXPECT_EQ("arm-generic", out.os_board); |
| 202 | EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.app_id); |
| 203 | EXPECT_EQ("0.2.2.3", out.app_version); |
| 204 | EXPECT_EQ("en-US", out.app_lang); |
| 205 | EXPECT_TRUE(out.delta_okay); |
| 206 | EXPECT_EQ("footrack", out.app_track); |
| 207 | EXPECT_EQ("http://forced.google.com", out.update_url); |
| 208 | } |
| 209 | |
| 210 | TEST_F(OmahaRequestDeviceParamsTest, MissingURLTest) { |
| 211 | ASSERT_TRUE(WriteFileString( |
| 212 | kTestDir + "/etc/lsb-release", |
| 213 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 214 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 215 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 216 | "CHROMEOS_RELEASE_TRACK=footrack")); |
| 217 | OmahaRequestParams out; |
| 218 | EXPECT_TRUE(DoTest(&out, "", "")); |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 219 | EXPECT_EQ("Chrome OS", out.os_platform); |
| 220 | EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp); |
| 221 | EXPECT_EQ("arm-generic", out.os_board); |
| 222 | EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.app_id); |
| 223 | EXPECT_EQ("0.2.2.3", out.app_version); |
| 224 | EXPECT_EQ("en-US", out.app_lang); |
| 225 | EXPECT_TRUE(out.delta_okay); |
| 226 | EXPECT_EQ("footrack", out.app_track); |
| 227 | EXPECT_EQ(OmahaRequestParams::kUpdateUrl, out.update_url); |
Andrew de los Reyes | 970bb28 | 2009-12-09 16:34:04 -0800 | [diff] [blame] | 228 | } |
| 229 | |
Andrew de los Reyes | 3f0303a | 2010-07-15 22:35:35 -0700 | [diff] [blame] | 230 | TEST_F(OmahaRequestDeviceParamsTest, NoDeltasTest) { |
Andrew de los Reyes | 3f0303a | 2010-07-15 22:35:35 -0700 | [diff] [blame] | 231 | ASSERT_TRUE(WriteFileString( |
| 232 | kTestDir + "/etc/lsb-release", |
| 233 | "CHROMEOS_RELEASE_FOO=CHROMEOS_RELEASE_VERSION=1.2.3.4\n" |
| 234 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 235 | "CHROMEOS_RELEASE_TRXCK=footrack")); |
| 236 | ASSERT_TRUE(WriteFileString(kTestDir + "/.nodelta", "")); |
| 237 | OmahaRequestParams out; |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 238 | EXPECT_TRUE(DoTest(&out, "", "")); |
Andrew de los Reyes | 3f0303a | 2010-07-15 22:35:35 -0700 | [diff] [blame] | 239 | EXPECT_FALSE(out.delta_okay); |
Andrew de los Reyes | 3f0303a | 2010-07-15 22:35:35 -0700 | [diff] [blame] | 240 | } |
| 241 | |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 242 | TEST_F(OmahaRequestDeviceParamsTest, OverrideTest) { |
| 243 | ASSERT_TRUE(WriteFileString( |
| 244 | kTestDir + "/etc/lsb-release", |
| 245 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 246 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 247 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 248 | "CHROMEOS_RELEASE_TRACK=footrack\n" |
| 249 | "CHROMEOS_AUSERVER=http://www.google.com")); |
| 250 | ASSERT_TRUE(WriteFileString( |
| 251 | kTestDir + utils::kStatefulPartition + "/etc/lsb-release", |
| 252 | "CHROMEOS_RELEASE_BOARD=x86-generic\n" |
| 253 | "CHROMEOS_RELEASE_TRACK=bartrack\n" |
Darin Petkov | 10d02dd | 2011-01-10 14:57:39 -0800 | [diff] [blame] | 254 | "CHROMEOS_AUSERVER=https://www.google.com")); |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 255 | OmahaRequestParams out; |
| 256 | EXPECT_TRUE(DoTest(&out, "", "")); |
| 257 | EXPECT_EQ("Chrome OS", out.os_platform); |
| 258 | EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp); |
| 259 | EXPECT_EQ("x86-generic", out.os_board); |
| 260 | EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.app_id); |
| 261 | EXPECT_EQ("0.2.2.3", out.app_version); |
| 262 | EXPECT_EQ("en-US", out.app_lang); |
| 263 | EXPECT_EQ("", out.hardware_class); |
Darin Petkov | a3df55b | 2010-11-15 13:33:55 -0800 | [diff] [blame] | 264 | EXPECT_FALSE(out.delta_okay); |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 265 | EXPECT_EQ("bartrack", out.app_track); |
Darin Petkov | 10d02dd | 2011-01-10 14:57:39 -0800 | [diff] [blame] | 266 | EXPECT_EQ("https://www.google.com", out.update_url); |
| 267 | } |
| 268 | |
| 269 | TEST_F(OmahaRequestDeviceParamsTest, OverrideLockDownTest) { |
| 270 | ASSERT_TRUE(WriteFileString( |
| 271 | kTestDir + "/etc/lsb-release", |
| 272 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 273 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 274 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 275 | "CHROMEOS_RELEASE_TRACK=footrack\n" |
| 276 | "CHROMEOS_AUSERVER=https://www.google.com")); |
| 277 | ASSERT_TRUE(WriteFileString( |
| 278 | kTestDir + utils::kStatefulPartition + "/etc/lsb-release", |
| 279 | "CHROMEOS_RELEASE_BOARD=x86-generic\n" |
| 280 | "CHROMEOS_RELEASE_TRACK=dev-channel\n" |
| 281 | "CHROMEOS_AUSERVER=http://www.google.com")); |
| 282 | params_.SetLockDown(true); |
| 283 | OmahaRequestParams out; |
| 284 | EXPECT_TRUE(DoTest(&out, "", "")); |
| 285 | EXPECT_EQ("arm-generic", out.os_board); |
| 286 | EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.app_id); |
| 287 | EXPECT_EQ("0.2.2.3", out.app_version); |
| 288 | EXPECT_EQ("", out.hardware_class); |
| 289 | EXPECT_FALSE(out.delta_okay); |
| 290 | EXPECT_EQ("dev-channel", out.app_track); |
| 291 | EXPECT_EQ("https://www.google.com", out.update_url); |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 292 | } |
| 293 | |
Darin Petkov | a3df55b | 2010-11-15 13:33:55 -0800 | [diff] [blame] | 294 | TEST_F(OmahaRequestDeviceParamsTest, OverrideSameTrackTest) { |
| 295 | ASSERT_TRUE(WriteFileString( |
| 296 | kTestDir + "/etc/lsb-release", |
| 297 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 298 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 299 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 300 | "CHROMEOS_RELEASE_TRACK=footrack\n" |
| 301 | "CHROMEOS_AUSERVER=http://www.google.com")); |
| 302 | ASSERT_TRUE(WriteFileString( |
| 303 | kTestDir + utils::kStatefulPartition + "/etc/lsb-release", |
| 304 | "CHROMEOS_RELEASE_BOARD=x86-generic\n" |
| 305 | "CHROMEOS_RELEASE_TRACK=footrack")); |
| 306 | OmahaRequestParams out; |
| 307 | EXPECT_TRUE(DoTest(&out, "", "")); |
Darin Petkov | a3df55b | 2010-11-15 13:33:55 -0800 | [diff] [blame] | 308 | EXPECT_EQ("x86-generic", out.os_board); |
| 309 | EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.app_id); |
| 310 | EXPECT_EQ("0.2.2.3", out.app_version); |
Darin Petkov | a3df55b | 2010-11-15 13:33:55 -0800 | [diff] [blame] | 311 | EXPECT_EQ("", out.hardware_class); |
| 312 | EXPECT_TRUE(out.delta_okay); |
| 313 | EXPECT_EQ("footrack", out.app_track); |
| 314 | EXPECT_EQ("http://www.google.com", out.update_url); |
| 315 | } |
| 316 | |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 317 | TEST_F(OmahaRequestDeviceParamsTest, SetTrackSimpleTest) { |
| 318 | ASSERT_TRUE(WriteFileString( |
| 319 | kTestDir + "/etc/lsb-release", |
| 320 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 321 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 322 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 323 | "CHROMEOS_RELEASE_TRACK=footrack\n" |
| 324 | "CHROMEOS_AUSERVER=http://www.google.com")); |
| 325 | { |
| 326 | OmahaRequestDeviceParams params; |
| 327 | params.set_root(string("./") + kTestDir); |
Darin Petkov | 10d02dd | 2011-01-10 14:57:39 -0800 | [diff] [blame] | 328 | params.SetLockDown(false); |
Patrick Dubroy | 7fbbe8a | 2011-08-01 17:28:22 +0200 | [diff] [blame] | 329 | EXPECT_TRUE(params.Init("", "", "")); |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 330 | params.SetTrack("zootrack"); |
| 331 | } |
| 332 | OmahaRequestParams out; |
| 333 | EXPECT_TRUE(DoTest(&out, "", "")); |
| 334 | EXPECT_EQ("arm-generic", out.os_board); |
| 335 | EXPECT_EQ("zootrack", out.app_track); |
| 336 | } |
| 337 | |
| 338 | TEST_F(OmahaRequestDeviceParamsTest, SetTrackPreserveTest) { |
| 339 | ASSERT_TRUE(WriteFileString( |
| 340 | kTestDir + "/etc/lsb-release", |
| 341 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 342 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 343 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 344 | "CHROMEOS_RELEASE_TRACK=footrack\n" |
| 345 | "CHROMEOS_AUSERVER=http://www.google.com")); |
| 346 | ASSERT_TRUE(WriteFileString( |
| 347 | kTestDir + utils::kStatefulPartition + "/etc/lsb-release", |
| 348 | "CHROMEOS_RELEASE_BOARD=x86-generic\n" |
| 349 | "CHROMEOS_RELEASE_TRACK=bartrack")); |
| 350 | { |
| 351 | OmahaRequestDeviceParams params; |
| 352 | params.set_root(string("./") + kTestDir); |
Darin Petkov | 10d02dd | 2011-01-10 14:57:39 -0800 | [diff] [blame] | 353 | params.SetLockDown(false); |
Patrick Dubroy | 7fbbe8a | 2011-08-01 17:28:22 +0200 | [diff] [blame] | 354 | EXPECT_TRUE(params.Init("", "", "")); |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 355 | params.SetTrack("zootrack"); |
| 356 | } |
| 357 | OmahaRequestParams out; |
| 358 | EXPECT_TRUE(DoTest(&out, "", "")); |
| 359 | EXPECT_EQ("x86-generic", out.os_board); |
| 360 | EXPECT_EQ("zootrack", out.app_track); |
| 361 | } |
| 362 | |
| 363 | TEST_F(OmahaRequestDeviceParamsTest, SetTrackInvalidTest) { |
| 364 | ASSERT_TRUE(WriteFileString( |
| 365 | kTestDir + "/etc/lsb-release", |
| 366 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 367 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 368 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 369 | "CHROMEOS_RELEASE_TRACK=footrack\n" |
| 370 | "CHROMEOS_AUSERVER=http://www.google.com")); |
| 371 | { |
| 372 | OmahaRequestDeviceParams params; |
| 373 | params.set_root(string("./") + kTestDir); |
Darin Petkov | 10d02dd | 2011-01-10 14:57:39 -0800 | [diff] [blame] | 374 | params.SetLockDown(true); |
Patrick Dubroy | 7fbbe8a | 2011-08-01 17:28:22 +0200 | [diff] [blame] | 375 | EXPECT_TRUE(params.Init("", "", "")); |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 376 | params.SetTrack("zootrack"); |
| 377 | } |
| 378 | OmahaRequestParams out; |
| 379 | EXPECT_TRUE(DoTest(&out, "", "")); |
| 380 | EXPECT_EQ("arm-generic", out.os_board); |
| 381 | EXPECT_EQ("footrack", out.app_track); |
| 382 | } |
| 383 | |
| 384 | TEST_F(OmahaRequestDeviceParamsTest, IsValidTrackTest) { |
Darin Petkov | 10d02dd | 2011-01-10 14:57:39 -0800 | [diff] [blame] | 385 | params_.SetLockDown(true); |
Darin Petkov | 3f375c7 | 2010-11-05 09:42:26 -0700 | [diff] [blame] | 386 | EXPECT_TRUE(params_.IsValidTrack("canary-channel")); |
Andrew de los Reyes | dc92876 | 2011-04-12 13:15:25 -0700 | [diff] [blame] | 387 | EXPECT_TRUE(params_.IsValidTrack("stable-channel")); |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 388 | EXPECT_TRUE(params_.IsValidTrack("beta-channel")); |
| 389 | EXPECT_TRUE(params_.IsValidTrack("dev-channel")); |
Don Garrett | 0bf8960 | 2012-03-22 14:11:19 -0700 | [diff] [blame^] | 390 | EXPECT_TRUE(params_.IsValidTrack("dogfood-channel")); |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 391 | EXPECT_FALSE(params_.IsValidTrack("some-channel")); |
| 392 | EXPECT_FALSE(params_.IsValidTrack("")); |
Darin Petkov | 10d02dd | 2011-01-10 14:57:39 -0800 | [diff] [blame] | 393 | params_.SetLockDown(false); |
Darin Petkov | 3f375c7 | 2010-11-05 09:42:26 -0700 | [diff] [blame] | 394 | EXPECT_TRUE(params_.IsValidTrack("canary-channel")); |
Andrew de los Reyes | dc92876 | 2011-04-12 13:15:25 -0700 | [diff] [blame] | 395 | EXPECT_TRUE(params_.IsValidTrack("stable-channel")); |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 396 | EXPECT_TRUE(params_.IsValidTrack("beta-channel")); |
| 397 | EXPECT_TRUE(params_.IsValidTrack("dev-channel")); |
Don Garrett | 0bf8960 | 2012-03-22 14:11:19 -0700 | [diff] [blame^] | 398 | EXPECT_TRUE(params_.IsValidTrack("dogfood-channel")); |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 399 | EXPECT_TRUE(params_.IsValidTrack("some-channel")); |
| 400 | EXPECT_TRUE(params_.IsValidTrack("")); |
| 401 | } |
| 402 | |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 403 | TEST_F(OmahaRequestDeviceParamsTest, ValidTrackTest) { |
| 404 | ASSERT_TRUE(WriteFileString( |
| 405 | kTestDir + "/etc/lsb-release", |
| 406 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 407 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 408 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 409 | "CHROMEOS_RELEASE_TRACK=dev-channel\n" |
| 410 | "CHROMEOS_AUSERVER=http://www.google.com")); |
Darin Petkov | 10d02dd | 2011-01-10 14:57:39 -0800 | [diff] [blame] | 411 | params_.SetLockDown(true); |
Darin Petkov | 49d9132 | 2010-10-25 16:34:58 -0700 | [diff] [blame] | 412 | OmahaRequestParams out; |
| 413 | EXPECT_TRUE(DoTest(&out, "", "")); |
| 414 | EXPECT_EQ("Chrome OS", out.os_platform); |
| 415 | EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp); |
| 416 | EXPECT_EQ("arm-generic", out.os_board); |
| 417 | EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.app_id); |
| 418 | EXPECT_EQ("0.2.2.3", out.app_version); |
| 419 | EXPECT_EQ("en-US", out.app_lang); |
| 420 | EXPECT_EQ("", out.hardware_class); |
| 421 | EXPECT_TRUE(out.delta_okay); |
| 422 | EXPECT_EQ("dev-channel", out.app_track); |
| 423 | EXPECT_EQ("http://www.google.com", out.update_url); |
| 424 | } |
| 425 | |
Patrick Dubroy | 7fbbe8a | 2011-08-01 17:28:22 +0200 | [diff] [blame] | 426 | TEST_F(OmahaRequestDeviceParamsTest, ChannelSpecified) { |
| 427 | ASSERT_TRUE(WriteFileString( |
| 428 | kTestDir + "/etc/lsb-release", |
| 429 | "CHROMEOS_RELEASE_BOARD=arm-generic\n" |
| 430 | "CHROMEOS_RELEASE_FOO=bar\n" |
| 431 | "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" |
| 432 | "CHROMEOS_RELEASE_TRACK=dev-channel\n" |
| 433 | "CHROMEOS_AUSERVER=http://www.google.com")); |
| 434 | params_.SetLockDown(true); |
| 435 | // Passed-in value for release channel should be used. |
| 436 | params_.Init("", "", "beta-channel"); |
| 437 | EXPECT_EQ("beta-channel", params_.app_track); |
| 438 | |
| 439 | // When passed-in value is invalid, value from lsb-release should be used. |
| 440 | params_.Init("", "", "foo-channel"); |
| 441 | EXPECT_EQ("dev-channel", params_.app_track); |
| 442 | |
| 443 | // When passed-in value is empty, value from lsb-release should be used. |
| 444 | params_.Init("", "", ""); |
| 445 | EXPECT_EQ("dev-channel", params_.app_track); |
| 446 | } |
| 447 | |
Darin Petkov | 10d02dd | 2011-01-10 14:57:39 -0800 | [diff] [blame] | 448 | TEST_F(OmahaRequestDeviceParamsTest, ShouldLockDownTest) { |
| 449 | EXPECT_FALSE(params_.ShouldLockDown()); |
| 450 | } |
| 451 | |
adlr@google.com | 3defe6a | 2009-12-04 20:57:17 +0000 | [diff] [blame] | 452 | } // namespace chromeos_update_engine |