1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
// Copyright 2016 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 "chrome/browser/ui/ash/system_tray_client.h"
#include "ash/common/system/tray/system_tray.h"
#include "ash/common/system/update/tray_update.h"
#include "ash/common/wm_shell.h"
#include "ash/root_window_controller.h"
#include "chrome/browser/upgrade_detector.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "content/public/test/test_utils.h"
ash::TrayUpdate* GetTrayUpdate() {
return ash::WmShell::Get()
->GetPrimaryRootWindowController()
->GetSystemTray()
->tray_update();
}
using SystemTrayClientTest = InProcessBrowserTest;
// Test that a chrome update shows the update icon in the system menu.
IN_PROC_BROWSER_TEST_F(SystemTrayClientTest, UpdateTrayIcon) {
ash::TrayUpdate* tray_update = GetTrayUpdate();
// When no update is pending, the icon isn't visible.
EXPECT_FALSE(tray_update->tray_view()->visible());
// Simulate an upgrade. This sends a mojo message to ash.
UpgradeDetector::GetInstance()->NotifyUpgradeRecommended();
content::RunAllPendingInMessageLoop();
// Tray icon is now visible.
EXPECT_TRUE(tray_update->tray_view()->visible());
}
// Test that a flash update causes the update UI to show in the system menu.
IN_PROC_BROWSER_TEST_F(SystemTrayClientTest, FlashUpdateTrayIcon) {
ash::TrayUpdate* tray_update = GetTrayUpdate();
// When no update is pending, the icon isn't visible.
EXPECT_FALSE(tray_update->tray_view()->visible());
// Simulate a Flash update. This sends a mojo message to ash.
SystemTrayClient::Get()->SetFlashUpdateAvailable();
content::RunAllPendingInMessageLoop();
// Tray icon is now visible.
EXPECT_TRUE(tray_update->tray_view()->visible());
}
|
