// Copyright 2021 The Chromium OS 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 CRYPTOHOME_CLEANUP_MOCK_LOW_DISK_SPACE_HANDLER_H_
#define CRYPTOHOME_CLEANUP_MOCK_LOW_DISK_SPACE_HANDLER_H_

#include "cryptohome/cleanup/low_disk_space_handler.h"

#include <cstdint>
#include <string>

#include <base/bind.h>
#include <base/time/time.h>
#include <gmock/gmock.h>

namespace cryptohome {

class MockLowDiskSpaceHandler : public LowDiskSpaceHandler {
 public:
  MockLowDiskSpaceHandler() : LowDiskSpaceHandler(nullptr, nullptr, nullptr) {}
  virtual ~MockLowDiskSpaceHandler() = default;

  MOCK_METHOD(
      bool,
      Init,
      (base::RepeatingCallback<bool(
           const base::Location&, base::OnceClosure, const base::TimeDelta&)>),
      (override));
  MOCK_METHOD(void,
              SetLowDiskSpaceCallback,
              (const base::Callback<void(uint64_t)>&),
              (override));
  MOCK_METHOD(void,
              SetUpdateUserActivityTimestampCallback,
              (const base::Callback<void()>&),
              (override));
  MOCK_METHOD(void, Stop, (), (override));
};

}  // namespace cryptohome

#endif  // CRYPTOHOME_CLEANUP_MOCK_LOW_DISK_SPACE_HANDLER_H_