blob: 4c7bc935ffbe96de760ca478eb7a6a7ff1e78b32 (
plain)
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
// Copyright 2013 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.
#ifndef ASH_TEST_TEST_SHELF_DELEGATE_H_
#define ASH_TEST_TEST_SHELF_DELEGATE_H_
#include <map>
#include <set>
#include <string>
#include "ash/shelf/shelf_delegate.h"
#include "base/compiler_specific.h"
#include "base/macros.h"
#include "ui/aura/window_observer.h"
namespace ash {
class ShelfModel;
namespace test {
// Test implementation of ShelfDelegate.
// Tests may create icons for windows by calling AddShelfItem().
class TestShelfDelegate : public ShelfDelegate, public aura::WindowObserver {
public:
explicit TestShelfDelegate(ShelfModel* model);
~TestShelfDelegate() override;
// Adds a ShelfItem for the given |window|. The ShelfItem's status will be
// STATUS_CLOSED.
void AddShelfItem(aura::Window* window);
// Adds a ShelfItem for the given |window| and adds a mapping from the added
// ShelfItem's ShelfID to the given |app_id|. The ShelfItem's status will be
// STATUS_CLOSED.
void AddShelfItem(aura::Window* window, const std::string& app_id);
// Adds a ShelfItem for the given |window| with the specified |status|.
void AddShelfItem(aura::Window* window, ShelfItemStatus status);
// Removes the ShelfItem for the specified |window| and unpins it if it was
// pinned. The |window|'s ShelfID to app id mapping will be removed if it
// exists.
void RemoveShelfItemForWindow(aura::Window* window);
static TestShelfDelegate* instance() { return instance_; }
// WindowObserver implementation
void OnWindowDestroying(aura::Window* window) override;
void OnWindowHierarchyChanging(const HierarchyChangeParams& params) override;
// ShelfDelegate implementation.
void OnShelfCreated(Shelf* shelf) override;
void OnShelfDestroyed(Shelf* shelf) override;
ShelfID GetShelfIDForAppID(const std::string& app_id) override;
bool HasShelfIDToAppIDMapping(ShelfID id) const override;
const std::string& GetAppIDForShelfID(ShelfID id) override;
void PinAppWithID(const std::string& app_id) override;
bool IsAppPinned(const std::string& app_id) override;
void UnpinAppWithID(const std::string& app_id) override;
private:
// Adds a mapping from a ShelfID to an app id.
void AddShelfIDToAppIDMapping(ShelfID shelf_id, const std::string& app_id);
// Removes the mapping from a ShelfID to an app id.
void RemoveShelfIDToAppIDMapping(ShelfID shelf_id);
static TestShelfDelegate* instance_;
ShelfModel* model_;
std::set<std::string> pinned_apps_;
// Tracks the ShelfID to app id mappings.
std::map<ShelfID, std::string> shelf_id_to_app_id_map_;
DISALLOW_COPY_AND_ASSIGN(TestShelfDelegate);
};
} // namespace test
} // namespace ash
#endif // ASH_TEST_TEST_SHELF_DELEGATE_H_
|