cryptohome/mock_le_credential_backend.h

// Copyright 2019 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_MOCK_LE_CREDENTIAL_BACKEND_H_
#define CRYPTOHOME_MOCK_LE_CREDENTIAL_BACKEND_H_

#include "cryptohome/le_credential_backend.h"

#include <map>
#include <string>
#include <vector>

#include <brillo/secure_blob.h>
#include <dbus/cryptohome/dbus-constants.h>
#include <gmock/gmock.h>

namespace cryptohome {

class MockLECredentialBackend : public LECredentialBackend {
 public:
  MockLECredentialBackend() = default;

  virtual ~MockLECredentialBackend() = default;

  MOCK_METHOD1(Reset, bool(std::vector<uint8_t>*));
  MOCK_METHOD0(IsSupported, bool());

  MOCK_METHOD10(InsertCredential,
                bool(const uint64_t,
                     const std::vector<std::vector<uint8_t>>&,
                     const brillo::SecureBlob&,
                     const brillo::SecureBlob&,
                     const brillo::SecureBlob&,
                     const std::map<uint32_t, uint32_t>&,
                     const ValidPcrCriteria&,
                     std::vector<uint8_t>*,
                     std::vector<uint8_t>*,
                     std::vector<uint8_t>*));

  MOCK_METHOD1(NeedsPCRBinding, bool(const std::vector<uint8_t>&));

  MOCK_METHOD10(CheckCredential,
                bool(const uint64_t,
                     const std::vector<std::vector<uint8_t>>&,
                     const std::vector<uint8_t>&,
                     const brillo::SecureBlob&,
                     std::vector<uint8_t>*,
                     std::vector<uint8_t>*,
                     brillo::SecureBlob*,
                     brillo::SecureBlob*,
                     LECredBackendError*,
                     std::vector<uint8_t>*));

  MOCK_METHOD8(ResetCredential,
               bool(const uint64_t,
                    const std::vector<std::vector<uint8_t>>&,
                    const std::vector<uint8_t>&,
                    const brillo::SecureBlob&,
                    std::vector<uint8_t>*,
                    std::vector<uint8_t>*,
                    LECredBackendError*,
                    std::vector<uint8_t>*));

  MOCK_METHOD4(RemoveCredential,
               bool(const uint64_t,
                    const std::vector<std::vector<uint8_t>>&,
                    const std::vector<uint8_t>&,
                    std::vector<uint8_t>*));

  MOCK_METHOD3(GetLog,
               bool(const std::vector<uint8_t>&,
                    std::vector<uint8_t>*,
                    std::vector<LELogEntry>*));

  MOCK_METHOD5(ReplayLogOperation,
               bool(const std::vector<uint8_t>&,
                    const std::vector<std::vector<uint8_t>>&,
                    const std::vector<uint8_t>&,
                    std::vector<uint8_t>*,
                    std::vector<uint8_t>*));
};

}  // namespace cryptohome

#endif  // CRYPTOHOME_MOCK_LE_CREDENTIAL_BACKEND_H_