| // Copyright (c) 2011 The Chromium OS 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 <stdio.h> | |
| #include <string> | |
| #include "base/files/file_util.h" | |
| #include "gtest/gtest.h" | |
| #include "update_engine/constants.h" | |
| #include "update_engine/fake_system_state.h" | |
| #include "update_engine/install_plan.h" | |
| #include "update_engine/omaha_request_params.h" | |
| #include "update_engine/test_utils.h" | |
| #include "update_engine/utils.h" | |
| using std::string; | |
| namespace chromeos_update_engine { | |
| class OmahaRequestParamsTest : public ::testing::Test { | |
| public: | |
| OmahaRequestParamsTest() : params_(&fake_system_state_) {} | |
| protected: | |
| // Return true iff the OmahaRequestParams::Init succeeded. If | |
| // out is non-null, it's set w/ the generated data. | |
| bool DoTest(OmahaRequestParams* out, const string& app_version, | |
| const string& omaha_url); | |
| virtual void SetUp() { | |
| // Create a uniquely named test directory. | |
| ASSERT_TRUE(utils::MakeTempDirectory(kTestDirTemplate, | |
| &test_dir_)); | |
| ASSERT_EQ(0, System(string("mkdir -p ") + test_dir_ + "/etc")); | |
| ASSERT_EQ(0, System(string("mkdir -p ") + test_dir_ + | |
| kStatefulPartition + "/etc")); | |
| // Create a fresh copy of the params for each test, so there's no | |
| // unintended reuse of state across tests. | |
| OmahaRequestParams new_params(&fake_system_state_); | |
| params_ = new_params; | |
| params_.set_root(test_dir_); | |
| params_.SetLockDown(false); | |
| } | |
| virtual void TearDown() { | |
| EXPECT_EQ(0, System(string("rm -rf ") + test_dir_)); | |
| } | |
| OmahaRequestParams params_; | |
| FakeSystemState fake_system_state_; | |
| static const char* kTestDirTemplate; | |
| string test_dir_; | |
| }; | |
| const char* OmahaRequestParamsTest::kTestDirTemplate = | |
| "omaha_request_params-test-XXXXXX"; | |
| bool OmahaRequestParamsTest::DoTest(OmahaRequestParams* out, | |
| const string& app_version, | |
| const string& omaha_url) { | |
| bool success = params_.Init(app_version, omaha_url, false); | |
| if (out) | |
| *out = params_; | |
| return success; | |
| } | |
| namespace { | |
| string GetMachineType() { | |
| string machine_type; | |
| if (!utils::ReadPipe("uname -m", &machine_type)) | |
| return ""; | |
| // Strip anything from the first newline char. | |
| size_t newline_pos = machine_type.find('\n'); | |
| if (newline_pos != string::npos) | |
| machine_type.erase(newline_pos); | |
| return machine_type; | |
| } | |
| } // namespace | |
| TEST_F(OmahaRequestParamsTest, SimpleTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel\n" | |
| "CHROMEOS_AUSERVER=http://www.google.com")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("Chrome OS", out.os_platform()); | |
| EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp()); | |
| EXPECT_EQ("arm-generic", out.os_board()); | |
| EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.GetAppId()); | |
| EXPECT_EQ("0.2.2.3", out.app_version()); | |
| EXPECT_EQ("en-US", out.app_lang()); | |
| EXPECT_EQ(fake_system_state_.hardware()->GetHardwareClass(), out.hwid()); | |
| EXPECT_TRUE(out.delta_okay()); | |
| EXPECT_EQ("dev-channel", out.target_channel()); | |
| EXPECT_EQ("http://www.google.com", out.update_url()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, AppIDTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel\n" | |
| "CHROMEOS_RELEASE_APPID={58c35cef-9d30-476e-9098-ce20377d535d}\n" | |
| "CHROMEOS_AUSERVER=http://www.google.com")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("Chrome OS", out.os_platform()); | |
| EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp()); | |
| EXPECT_EQ("arm-generic", out.os_board()); | |
| EXPECT_EQ("{58c35cef-9d30-476e-9098-ce20377d535d}", out.GetAppId()); | |
| EXPECT_EQ("0.2.2.3", out.app_version()); | |
| EXPECT_EQ("en-US", out.app_lang()); | |
| EXPECT_EQ(fake_system_state_.hardware()->GetHardwareClass(), out.hwid()); | |
| EXPECT_TRUE(out.delta_okay()); | |
| EXPECT_EQ("dev-channel", out.target_channel()); | |
| EXPECT_EQ("http://www.google.com", out.update_url()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, MissingChannelTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRXCK=dev-channel")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("Chrome OS", out.os_platform()); | |
| EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp()); | |
| EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.GetAppId()); | |
| EXPECT_EQ("0.2.2.3", out.app_version()); | |
| EXPECT_EQ("en-US", out.app_lang()); | |
| EXPECT_EQ("", out.target_channel()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, ConfusingReleaseTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_FOO=CHROMEOS_RELEASE_VERSION=1.2.3.4\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRXCK=dev-channel")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("Chrome OS", out.os_platform()); | |
| EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp()); | |
| EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.GetAppId()); | |
| EXPECT_EQ("0.2.2.3", out.app_version()); | |
| EXPECT_EQ("en-US", out.app_lang()); | |
| EXPECT_EQ("", out.target_channel()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, MissingVersionTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("Chrome OS", out.os_platform()); | |
| EXPECT_EQ(string("_") + GetMachineType(), out.os_sp()); | |
| EXPECT_EQ("arm-generic", out.os_board()); | |
| EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.GetAppId()); | |
| EXPECT_EQ("", out.app_version()); | |
| EXPECT_EQ("en-US", out.app_lang()); | |
| EXPECT_TRUE(out.delta_okay()); | |
| EXPECT_EQ("dev-channel", out.target_channel()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, ForceVersionTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "ForcedVersion", "")); | |
| EXPECT_EQ("Chrome OS", out.os_platform()); | |
| EXPECT_EQ(string("ForcedVersion_") + GetMachineType(), out.os_sp()); | |
| EXPECT_EQ("arm-generic", out.os_board()); | |
| EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.GetAppId()); | |
| EXPECT_EQ("ForcedVersion", out.app_version()); | |
| EXPECT_EQ("en-US", out.app_lang()); | |
| EXPECT_TRUE(out.delta_okay()); | |
| EXPECT_EQ("dev-channel", out.target_channel()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, ForcedURLTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "http://forced.google.com")); | |
| EXPECT_EQ("Chrome OS", out.os_platform()); | |
| EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp()); | |
| EXPECT_EQ("arm-generic", out.os_board()); | |
| EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.GetAppId()); | |
| EXPECT_EQ("0.2.2.3", out.app_version()); | |
| EXPECT_EQ("en-US", out.app_lang()); | |
| EXPECT_TRUE(out.delta_okay()); | |
| EXPECT_EQ("dev-channel", out.target_channel()); | |
| EXPECT_EQ("http://forced.google.com", out.update_url()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, MissingURLTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("Chrome OS", out.os_platform()); | |
| EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp()); | |
| EXPECT_EQ("arm-generic", out.os_board()); | |
| EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.GetAppId()); | |
| EXPECT_EQ("0.2.2.3", out.app_version()); | |
| EXPECT_EQ("en-US", out.app_lang()); | |
| EXPECT_TRUE(out.delta_okay()); | |
| EXPECT_EQ("dev-channel", out.target_channel()); | |
| EXPECT_EQ(kProductionOmahaUrl, out.update_url()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, NoDeltasTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_FOO=CHROMEOS_RELEASE_VERSION=1.2.3.4\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRXCK=dev-channel")); | |
| ASSERT_TRUE(WriteFileString(test_dir_ + "/.nodelta", "")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_FALSE(out.delta_okay()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, OverrideTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel\n" | |
| "CHROMEOS_AUSERVER=http://www.google.com")); | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + kStatefulPartition + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=x86-generic\n" | |
| "CHROMEOS_RELEASE_TRACK=beta-channel\n" | |
| "CHROMEOS_AUSERVER=https://www.google.com")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("Chrome OS", out.os_platform()); | |
| EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp()); | |
| EXPECT_EQ("x86-generic", out.os_board()); | |
| EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.GetAppId()); | |
| EXPECT_EQ("0.2.2.3", out.app_version()); | |
| EXPECT_EQ("en-US", out.app_lang()); | |
| EXPECT_EQ(fake_system_state_.hardware()->GetHardwareClass(), out.hwid()); | |
| EXPECT_FALSE(out.delta_okay()); | |
| EXPECT_EQ("beta-channel", out.target_channel()); | |
| EXPECT_EQ("https://www.google.com", out.update_url()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, OverrideLockDownTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel\n" | |
| "CHROMEOS_AUSERVER=https://www.google.com")); | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + kStatefulPartition + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=x86-generic\n" | |
| "CHROMEOS_RELEASE_TRACK=stable-channel\n" | |
| "CHROMEOS_AUSERVER=http://www.google.com")); | |
| params_.SetLockDown(true); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("arm-generic", out.os_board()); | |
| EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.GetAppId()); | |
| EXPECT_EQ("0.2.2.3", out.app_version()); | |
| EXPECT_EQ(fake_system_state_.hardware()->GetHardwareClass(), out.hwid()); | |
| EXPECT_FALSE(out.delta_okay()); | |
| EXPECT_EQ("stable-channel", out.target_channel()); | |
| EXPECT_EQ("https://www.google.com", out.update_url()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, OverrideSameChannelTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel\n" | |
| "CHROMEOS_AUSERVER=http://www.google.com")); | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + kStatefulPartition + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=x86-generic\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("x86-generic", out.os_board()); | |
| EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.GetAppId()); | |
| EXPECT_EQ("0.2.2.3", out.app_version()); | |
| EXPECT_EQ(fake_system_state_.hardware()->GetHardwareClass(), out.hwid()); | |
| EXPECT_TRUE(out.delta_okay()); | |
| EXPECT_EQ("dev-channel", out.target_channel()); | |
| EXPECT_EQ("http://www.google.com", out.update_url()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, SetTargetChannelTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel\n" | |
| "CHROMEOS_AUSERVER=http://www.google.com")); | |
| { | |
| OmahaRequestParams params(&fake_system_state_); | |
| params.set_root(test_dir_); | |
| params.SetLockDown(false); | |
| EXPECT_TRUE(params.Init("", "", false)); | |
| params.SetTargetChannel("canary-channel", false); | |
| EXPECT_FALSE(params.is_powerwash_allowed()); | |
| } | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("canary-channel", out.target_channel()); | |
| EXPECT_FALSE(out.is_powerwash_allowed()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, SetIsPowerwashAllowedTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel\n" | |
| "CHROMEOS_AUSERVER=http://www.google.com")); | |
| { | |
| OmahaRequestParams params(&fake_system_state_); | |
| params.set_root(test_dir_); | |
| params.SetLockDown(false); | |
| EXPECT_TRUE(params.Init("", "", false)); | |
| params.SetTargetChannel("canary-channel", true); | |
| EXPECT_TRUE(params.is_powerwash_allowed()); | |
| } | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("canary-channel", out.target_channel()); | |
| EXPECT_TRUE(out.is_powerwash_allowed()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, SetTargetChannelInvalidTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel\n" | |
| "CHROMEOS_AUSERVER=http://www.google.com")); | |
| { | |
| OmahaRequestParams params(&fake_system_state_); | |
| params.set_root(string("./") + test_dir_); | |
| params.SetLockDown(true); | |
| EXPECT_TRUE(params.Init("", "", false)); | |
| params.SetTargetChannel("dogfood-channel", true); | |
| EXPECT_FALSE(params.is_powerwash_allowed()); | |
| } | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("arm-generic", out.os_board()); | |
| EXPECT_EQ("dev-channel", out.target_channel()); | |
| EXPECT_FALSE(out.is_powerwash_allowed()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, IsValidChannelTest) { | |
| params_.SetLockDown(false); | |
| EXPECT_TRUE(params_.IsValidChannel("canary-channel")); | |
| EXPECT_TRUE(params_.IsValidChannel("stable-channel")); | |
| EXPECT_TRUE(params_.IsValidChannel("beta-channel")); | |
| EXPECT_TRUE(params_.IsValidChannel("dev-channel")); | |
| EXPECT_FALSE(params_.IsValidChannel("testimage-channel")); | |
| EXPECT_FALSE(params_.IsValidChannel("dogfood-channel")); | |
| EXPECT_FALSE(params_.IsValidChannel("some-channel")); | |
| EXPECT_FALSE(params_.IsValidChannel("")); | |
| } | |
| TEST_F(OmahaRequestParamsTest, ValidChannelTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel\n" | |
| "CHROMEOS_AUSERVER=http://www.google.com")); | |
| params_.SetLockDown(true); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("Chrome OS", out.os_platform()); | |
| EXPECT_EQ(string("0.2.2.3_") + GetMachineType(), out.os_sp()); | |
| EXPECT_EQ("arm-generic", out.os_board()); | |
| EXPECT_EQ("{87efface-864d-49a5-9bb3-4b050a7c227a}", out.GetAppId()); | |
| EXPECT_EQ("0.2.2.3", out.app_version()); | |
| EXPECT_EQ("en-US", out.app_lang()); | |
| EXPECT_EQ(fake_system_state_.hardware()->GetHardwareClass(), out.hwid()); | |
| EXPECT_TRUE(out.delta_okay()); | |
| EXPECT_EQ("dev-channel", out.target_channel()); | |
| EXPECT_EQ("http://www.google.com", out.update_url()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, SetTargetChannelWorks) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRACK=dev-channel\n" | |
| "CHROMEOS_AUSERVER=http://www.google.com")); | |
| params_.SetLockDown(false); | |
| // Check LSB value is used by default when SetTargetChannel is not called. | |
| params_.Init("", "", false); | |
| EXPECT_EQ("dev-channel", params_.target_channel()); | |
| // When an invalid value is set, it should be ignored and the | |
| // value from lsb-release should be used instead. | |
| params_.Init("", "", false); | |
| EXPECT_FALSE(params_.SetTargetChannel("invalid-channel", false)); | |
| EXPECT_EQ("dev-channel", params_.target_channel()); | |
| // When set to a valid value, it should take effect. | |
| params_.Init("", "", false); | |
| EXPECT_TRUE(params_.SetTargetChannel("beta-channel", true)); | |
| EXPECT_EQ("beta-channel", params_.target_channel()); | |
| // When set to the same value, it should be idempotent. | |
| params_.Init("", "", false); | |
| EXPECT_TRUE(params_.SetTargetChannel("beta-channel", true)); | |
| EXPECT_EQ("beta-channel", params_.target_channel()); | |
| // When set to a valid value while a change is already pending, it should | |
| // succeed. | |
| params_.Init("", "", false); | |
| EXPECT_TRUE(params_.SetTargetChannel("stable-channel", true)); | |
| EXPECT_EQ("stable-channel", params_.target_channel()); | |
| // Set a different channel in stateful LSB release. | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + kStatefulPartition + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_TRACK=stable-channel\n" | |
| "CHROMEOS_IS_POWERWASH_ALLOWED=true\n")); | |
| // When set to a valid value while a change is already pending, it should | |
| // succeed. | |
| params_.Init("", "", false); | |
| EXPECT_TRUE(params_.SetTargetChannel("beta-channel", true)); | |
| // The target channel should reflect the change, but the download channel | |
| // should continue to retain the old value ... | |
| EXPECT_EQ("beta-channel", params_.target_channel()); | |
| EXPECT_EQ("stable-channel", params_.download_channel()); | |
| // ... until we update the download channel explicitly. | |
| params_.UpdateDownloadChannel(); | |
| EXPECT_EQ("beta-channel", params_.download_channel()); | |
| EXPECT_EQ("beta-channel", params_.target_channel()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, ChannelIndexTest) { | |
| int canary = params_.GetChannelIndex("canary-channel"); | |
| int dev = params_.GetChannelIndex("dev-channel"); | |
| int beta = params_.GetChannelIndex("beta-channel"); | |
| int stable = params_.GetChannelIndex("stable-channel"); | |
| EXPECT_LE(canary, dev); | |
| EXPECT_LE(dev, beta); | |
| EXPECT_LE(beta, stable); | |
| // testimage-channel or other names are not recognized, so index will be -1. | |
| int testimage = params_.GetChannelIndex("testimage-channel"); | |
| int bogus = params_.GetChannelIndex("bogus-channel"); | |
| EXPECT_EQ(-1, testimage); | |
| EXPECT_EQ(-1, bogus); | |
| } | |
| TEST_F(OmahaRequestParamsTest, ToMoreStableChannelFlagTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=arm-generic\n" | |
| "CHROMEOS_RELEASE_FOO=bar\n" | |
| "CHROMEOS_RELEASE_VERSION=0.2.2.3\n" | |
| "CHROMEOS_RELEASE_TRACK=canary-channel\n" | |
| "CHROMEOS_AUSERVER=http://www.google.com")); | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + kStatefulPartition + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_BOARD=x86-generic\n" | |
| "CHROMEOS_RELEASE_TRACK=stable-channel\n" | |
| "CHROMEOS_AUSERVER=https://www.google.com")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("https://www.google.com", out.update_url()); | |
| EXPECT_FALSE(out.delta_okay()); | |
| EXPECT_EQ("stable-channel", out.target_channel()); | |
| EXPECT_TRUE(out.to_more_stable_channel()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, ShouldLockDownTest) { | |
| EXPECT_FALSE(params_.ShouldLockDown()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, BoardAppIdUsedForNonCanaryChannelTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_APPID=r\n" | |
| "CHROMEOS_BOARD_APPID=b\n" | |
| "CHROMEOS_CANARY_APPID=c\n" | |
| "CHROMEOS_RELEASE_TRACK=stable-channel\n")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("stable-channel", out.download_channel()); | |
| EXPECT_EQ("b", out.GetAppId()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, CanaryAppIdUsedForCanaryChannelTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_APPID=r\n" | |
| "CHROMEOS_BOARD_APPID=b\n" | |
| "CHROMEOS_CANARY_APPID=c\n" | |
| "CHROMEOS_RELEASE_TRACK=canary-channel\n")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("canary-channel", out.download_channel()); | |
| EXPECT_EQ("c", out.GetAppId()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, ReleaseAppIdUsedAsDefaultTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_APPID=r\n" | |
| "CHROMEOS_CANARY_APPID=c\n" | |
| "CHROMEOS_RELEASE_TRACK=stable-channel\n")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| EXPECT_TRUE(DoTest(&out, "", "")); | |
| EXPECT_EQ("stable-channel", out.download_channel()); | |
| EXPECT_EQ("r", out.GetAppId()); | |
| } | |
| TEST_F(OmahaRequestParamsTest, CollectECFWVersionsTest) { | |
| ASSERT_TRUE(WriteFileString( | |
| test_dir_ + "/etc/lsb-release", | |
| "CHROMEOS_RELEASE_APPID=r\n" | |
| "CHROMEOS_CANARY_APPID=c\n" | |
| "CHROMEOS_RELEASE_TRACK=stable-channel\n")); | |
| OmahaRequestParams out(&fake_system_state_); | |
| out.hwid_ = string("STUMPY ALEX 12345"); | |
| EXPECT_FALSE(out.CollectECFWVersions()); | |
| out.hwid_ = string("SNOW 12345"); | |
| EXPECT_TRUE(out.CollectECFWVersions()); | |
| out.hwid_ = string("SAMS ALEX 12345"); | |
| EXPECT_TRUE(out.CollectECFWVersions()); | |
| } | |
| } // namespace chromeos_update_engine |