login_manager/mock_device_identifier_generator.h

// 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 LOGIN_MANAGER_MOCK_DEVICE_IDENTIFIER_GENERATOR_H_
#define LOGIN_MANAGER_MOCK_DEVICE_IDENTIFIER_GENERATOR_H_

#include <gmock/gmock.h>

#include "login_manager/device_identifier_generator.h"

namespace login_manager {

class MockDeviceIdentifierGenerator : public DeviceIdentifierGenerator {
 public:
  MockDeviceIdentifierGenerator(SystemUtils* system_utils,
                                LoginMetrics* metrics)
      : DeviceIdentifierGenerator(system_utils, metrics) {}
  MockDeviceIdentifierGenerator(const MockDeviceIdentifierGenerator&) = delete;
  MockDeviceIdentifierGenerator& operator=(
      const MockDeviceIdentifierGenerator&) = delete;

  ~MockDeviceIdentifierGenerator() override {}

  MOCK_METHOD(void, RequestStateKeys, (const StateKeyCallback&), (override));
  MOCK_METHOD(void,
              RequestPsmDeviceActiveSecret,
              (const PsmDeviceActiveSecretCallback&),
              (override));
};

}  // namespace login_manager

#endif  // LOGIN_MANAGER_MOCK_DEVICE_IDENTIFIER_GENERATOR_H_