#pragma once

#include "extensions/common/aws/credentials_provider.h"
#include "extensions/common/aws/signer.h"

#include "gmock/gmock.h"

namespace Envoy {
namespace Extensions {
namespace Common {
namespace Aws {

class MockCredentialsProvider : public CredentialsProvider {
public:
  MockCredentialsProvider();
  ~MockCredentialsProvider() override;

  MOCK_METHOD(Credentials, getCredentials, ());
};

class MockSigner : public Signer {
public:
  MockSigner();
  ~MockSigner() override;

  MOCK_METHOD(void, sign, (Http::RequestMessage&, bool));
  MOCK_METHOD(void, sign, (Http::RequestHeaderMap&));
  MOCK_METHOD(void, sign, (Http::RequestHeaderMap&, const std::string&));
};

class MockMetadataFetcher {
public:
  virtual ~MockMetadataFetcher() = default;

  MOCK_METHOD(absl::optional<std::string>, fetch,
              (const std::string&, const std::string&, const absl::optional<std::string>&),
              (const));
};

class DummyMetadataFetcher {
public:
  absl::optional<std::string> operator()(const std::string&, const std::string&,
                                         const absl::optional<std::string>&) {
    return absl::nullopt;
  }
};

} // namespace Aws
} // namespace Common
} // namespace Extensions
} // namespace Envoy