// Copyright 2017 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_TRAY_ACTION_TEST_TRAY_ACTION_CLIENT_H_
#define ASH_TRAY_ACTION_TEST_TRAY_ACTION_CLIENT_H_

#include <vector>

#include "ash/public/interfaces/tray_action.mojom.h"
#include "base/macros.h"
#include "mojo/public/cpp/bindings/binding.h"

namespace ash {

class TestTrayActionClient : public mojom::TrayActionClient {
 public:
  TestTrayActionClient();

  ~TestTrayActionClient() override;

  mojom::TrayActionClientPtr CreateInterfacePtrAndBind();

  void ClearRecordedRequests();

  const std::vector<mojom::LockScreenNoteOrigin>& note_origins() const {
    return note_origins_;
  }

  const std::vector<mojom::CloseLockScreenNoteReason>& close_note_reasons()
      const {
    return close_note_reasons_;
  }

  // mojom::TrayActionClient:
  void RequestNewLockScreenNote(mojom::LockScreenNoteOrigin origin) override;
  void CloseLockScreenNote(mojom::CloseLockScreenNoteReason reason) override;

 private:
  mojo::Binding<mojom::TrayActionClient> binding_;

  std::vector<mojom::LockScreenNoteOrigin> note_origins_;
  std::vector<mojom::CloseLockScreenNoteReason> close_note_reasons_;

  DISALLOW_COPY_AND_ASSIGN(TestTrayActionClient);
};

}  // namespace ash

#endif  // ASH_TRAY_ACTION_TEST_TRAY_ACTION_CLIENT_H_