chrome/elevation_service/scoped_mock_context.h

// Copyright 2021 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_ELEVATION_SERVICE_SCOPED_MOCK_CONTEXT_H_
#define CHROME_ELEVATION_SERVICE_SCOPED_MOCK_CONTEXT_H_

#include <unknwn.h>
#include <wrl/client.h>

#include "base/memory/raw_ptr_exclusion.h"

namespace elevation_service {

// Installs an implementation of IServerSecurity that allows code under test to
// successfully call ::CoImpersonateClient().
class ScopedMockContext {
 public:
  ScopedMockContext();
  ScopedMockContext(const ScopedMockContext&) = delete;
  ScopedMockContext& operator=(const ScopedMockContext&) = delete;
  ~ScopedMockContext();

  bool Succeeded() const { return mock_call_context_; }

 private:
  Microsoft::WRL::ComPtr<IUnknown> mock_call_context_;
  // This field is not a raw_ptr<> because it was filtered by the rewriter for:
  // #addr-of
  RAW_PTR_EXCLUSION IUnknown* original_call_context_ = nullptr;
};

}  // namespace elevation_service

#endif  // CHROME_ELEVATION_SERVICE_SCOPED_MOCK_CONTEXT_H_