// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_ASH_ARC_TEST_ARC_DATA_REMOVED_WAITER_H_
#define CHROME_BROWSER_ASH_ARC_TEST_ARC_DATA_REMOVED_WAITER_H_

#include <memory>

#include "chrome/browser/ash/arc/session/arc_session_manager_observer.h"

namespace base {
class RunLoop;
}

namespace arc {

// Waits for ARC data has been removed.
class ArcDataRemovedWaiter : public ArcSessionManagerObserver {
 public:
  ArcDataRemovedWaiter();

  ArcDataRemovedWaiter(const ArcDataRemovedWaiter&) = delete;
  ArcDataRemovedWaiter& operator=(const ArcDataRemovedWaiter&) = delete;

  ~ArcDataRemovedWaiter() override;

  // Waits until ARC data is removed. Waiting is end once ArcSessionManager
  // sends OnArcDataRemoved notification.
  void Wait();

 private:
  // ArcSessionManagerObserver:
  void OnArcDataRemoved() override;

  std::unique_ptr<base::RunLoop> run_loop_;
};

}  // namespace arc

#endif  // CHROME_BROWSER_ASH_ARC_TEST_ARC_DATA_REMOVED_WAITER_H_