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#ifndef ASH_TEST_TEST_SHELF_DELEGATE_H_
6#define ASH_TEST_TEST_SHELF_DELEGATE_H_
7
8#include <map>
9#include <set>
10
11#include "ash/shelf/shelf_delegate.h"
12#include "base/compiler_specific.h"
13#include "ui/aura/window_observer.h"
14
15namespace ash {
16
17class ShelfModel;
18
19namespace test {
20
21// Test implementation of ShelfDelegate.
22// Tests may create icons for windows by calling AddShelfItem().
23class TestShelfDelegate : public ShelfDelegate, public aura::WindowObserver {
24 public:
25  explicit TestShelfDelegate(ShelfModel* model);
26  virtual ~TestShelfDelegate();
27
28  void AddShelfItem(aura::Window* window);
29  void AddShelfItem(aura::Window* window, ShelfItemStatus status);
30  void RemoveShelfItemForWindow(aura::Window* window);
31
32  static TestShelfDelegate* instance() { return instance_; }
33
34  // WindowObserver implementation
35  virtual void OnWindowDestroying(aura::Window* window) OVERRIDE;
36  virtual void OnWindowHierarchyChanging(
37      const HierarchyChangeParams& params) OVERRIDE;
38
39  // ShelfDelegate implementation.
40  virtual void OnShelfCreated(Shelf* shelf) OVERRIDE;
41  virtual void OnShelfDestroyed(Shelf* shelf) OVERRIDE;
42  virtual ShelfID GetShelfIDForAppID(const std::string& app_id) OVERRIDE;
43  virtual const std::string& GetAppIDForShelfID(ShelfID id) OVERRIDE;
44  virtual void PinAppWithID(const std::string& app_id) OVERRIDE;
45  virtual bool CanPin() const OVERRIDE;
46  virtual bool IsAppPinned(const std::string& app_id) OVERRIDE;
47  virtual void UnpinAppWithID(const std::string& app_id) OVERRIDE;
48
49 private:
50  static TestShelfDelegate* instance_;
51
52  ShelfModel* model_;
53
54  DISALLOW_COPY_AND_ASSIGN(TestShelfDelegate);
55};
56
57}  // namespace test
58}  // namespace ash
59
60#endif  // ASH_TEST_TEST_SHELF_DELEGATE_H_
61