1 // Copyright 2013 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
7 #include "base/basictypes.h"
8 #include "base/command_line.h"
9 #include "base/compiler_specific.h"
10 #include "base/file_util.h"
11 #include "base/files/file_path.h"
12 #include "base/files/scoped_temp_dir.h"
13 #include "base/memory/scoped_ptr.h"
14 #include "base/path_service.h"
15 #include "base/prefs/pref_service.h"
16 #include "base/strings/string_number_conversions.h"
17 #include "base/test/scoped_path_override.h"
18 #include "base/threading/sequenced_worker_pool.h"
19 #include "base/time/time.h"
20 #include "chrome/browser/apps/app_browsertest_util.h"
21 #include "chrome/browser/browser_process.h"
22 #include "chrome/browser/browser_process_platform_part.h"
23 #include "chrome/browser/chromeos/app_mode/kiosk_app_update_service.h"
24 #include "chrome/browser/chromeos/system/automatic_reboot_manager.h"
25 #include "chrome/browser/extensions/extension_test_message_listener.h"
26 #include "chrome/browser/profiles/profile.h"
27 #include "chrome/common/chrome_switches.h"
28 #include "chrome/common/pref_names.h"
29 #include "chromeos/chromeos_paths.h"
30 #include "chromeos/dbus/update_engine_client.h"
31 #include "content/public/browser/browser_thread.h"
32 #include "content/public/test/test_utils.h"
33 #include "extensions/common/extension.h"
34 #include "testing/gtest/include/gtest/gtest.h"
35
36 namespace chromeos {
37
38 class KioskAppUpdateServiceTest : public extensions::PlatformAppBrowserTest {
39 public:
KioskAppUpdateServiceTest()40 KioskAppUpdateServiceTest() : app_(NULL), update_service_(NULL) {}
~KioskAppUpdateServiceTest()41 virtual ~KioskAppUpdateServiceTest() {}
42
43 // extensions::PlatformAppBrowserTest overrides:
SetUpOnMainThread()44 virtual void SetUpOnMainThread() OVERRIDE {
45 extensions::PlatformAppBrowserTest::SetUpOnMainThread();
46
47 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
48 const base::FilePath& temp_dir = temp_dir_.path();
49
50 const base::TimeDelta uptime = base::TimeDelta::FromHours(1);
51 const std::string uptime_seconds =
52 base::DoubleToString(uptime.InSecondsF());
53 const base::FilePath uptime_file = temp_dir.Append("uptime");
54 ASSERT_EQ(static_cast<int>(uptime_seconds.size()),
55 base::WriteFile(
56 uptime_file, uptime_seconds.c_str(), uptime_seconds.size()));
57 uptime_file_override_.reset(
58 new base::ScopedPathOverride(chromeos::FILE_UPTIME, uptime_file));
59
60 app_ = LoadExtension(
61 test_data_dir_.AppendASCII("api_test/runtime/on_restart_required"));
62
63 // Fake app mode command line.
64 CommandLine* command = CommandLine::ForCurrentProcess();
65 command->AppendSwitch(switches::kForceAppMode);
66 command->AppendSwitchASCII(switches::kAppId, app_->id());
67
68 update_service_ = KioskAppUpdateServiceFactory::GetForProfile(profile());
69 update_service_->set_app_id(app_->id());
70
71 content::BrowserThread::GetBlockingPool()->FlushForTesting();
72 content::RunAllPendingInMessageLoop();
73 }
74
FireAppUpdateAvailable()75 void FireAppUpdateAvailable() {
76 update_service_->OnAppUpdateAvailable(app_);
77 }
78
FireUpdatedNeedReboot()79 void FireUpdatedNeedReboot() {
80 UpdateEngineClient::Status status;
81 status.status = UpdateEngineClient::UPDATE_STATUS_UPDATED_NEED_REBOOT;
82 g_browser_process->platform_part()->automatic_reboot_manager()->
83 UpdateStatusChanged(status);
84 }
85
86 private:
87 base::ScopedTempDir temp_dir_;
88 scoped_ptr<base::ScopedPathOverride> uptime_file_override_;
89 const extensions::Extension* app_; // Not owned.
90 KioskAppUpdateService* update_service_; // Not owned.
91
92 DISALLOW_COPY_AND_ASSIGN(KioskAppUpdateServiceTest);
93 };
94
IN_PROC_BROWSER_TEST_F(KioskAppUpdateServiceTest,AppUpdate)95 IN_PROC_BROWSER_TEST_F(KioskAppUpdateServiceTest, AppUpdate) {
96 FireAppUpdateAvailable();
97
98 ExtensionTestMessageListener listener("app_update", false);
99 listener.WaitUntilSatisfied();
100 }
101
IN_PROC_BROWSER_TEST_F(KioskAppUpdateServiceTest,OsUpdate)102 IN_PROC_BROWSER_TEST_F(KioskAppUpdateServiceTest, OsUpdate) {
103 g_browser_process->local_state()->SetBoolean(prefs::kRebootAfterUpdate, true);
104 FireUpdatedNeedReboot();
105
106 ExtensionTestMessageListener listener("os_update", false);
107 listener.WaitUntilSatisfied();
108 }
109
IN_PROC_BROWSER_TEST_F(KioskAppUpdateServiceTest,Periodic)110 IN_PROC_BROWSER_TEST_F(KioskAppUpdateServiceTest, Periodic) {
111 g_browser_process->local_state()->SetInteger(
112 prefs::kUptimeLimit, base::TimeDelta::FromMinutes(30).InSeconds());
113
114 ExtensionTestMessageListener listener("periodic", false);
115 listener.WaitUntilSatisfied();
116 }
117
118 } // namespace chromeos
119