1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85
   86
   87
   88
   89
   90
   91
   92
   93
   94
   95


media / gpu / windows / d3d11_video_processor_proxy_unittest.cc [blame]

// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <random>
#include <utility>

#include "base/functional/callback_helpers.h"
#include "media/base/win/d3d11_mocks.h"
#include "media/gpu/windows/d3d11_copying_texture_wrapper.h"
#include "media/gpu/windows/d3d11_texture_wrapper.h"
#include "media/gpu/windows/d3d11_video_processor_proxy.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

using ::testing::_;
using ::testing::DoAll;
using ::testing::Return;
using ::testing::SetArgPointee;

namespace media {

using MockD3D11VideoDevice = Microsoft::WRL::ComPtr<D3D11VideoDeviceMock>;
using MockD3D11DeviceContext = Microsoft::WRL::ComPtr<D3D11DeviceContextMock>;
using MockD3D11VideoContext = Microsoft::WRL::ComPtr<D3D11VideoContextMock>;
using MockD3D11VideoProcessor = Microsoft::WRL::ComPtr<D3D11VideoProcessorMock>;
using MockD3D11VideoProcessorEnumerator =
    Microsoft::WRL::ComPtr<D3D11VideoProcessorEnumeratorMock>;

class D3D11VideoProcessorProxyUnittest : public ::testing::Test {
 public:
  MockD3D11VideoDevice dev_;
  MockD3D11DeviceContext ctx_;
  MockD3D11VideoContext vctx_;
  MockD3D11VideoProcessorEnumerator enumerator_;
  MockD3D11VideoProcessor proc_;

  scoped_refptr<VideoProcessorProxy> CreateProxy() {
    dev_ = MakeComPtr<D3D11VideoDeviceMock>();
    ctx_ = MakeComPtr<D3D11DeviceContextMock>();
    vctx_ = MakeComPtr<D3D11VideoContextMock>();
    proc_ = MakeComPtr<D3D11VideoProcessorMock>();
    enumerator_ = MakeComPtr<D3D11VideoProcessorEnumeratorMock>();

    EXPECT_CALL(*dev_.Get(), CreateVideoProcessorEnumerator(_, _))
        .WillOnce(SetComPointeeAndReturnOk<1>(enumerator_.Get()));

    EXPECT_CALL(*dev_.Get(), CreateVideoProcessor(_, _, _))
        .WillOnce(SetComPointeeAndReturnOk<2>(proc_.Get()));

    EXPECT_CALL(*ctx_.Get(), QueryInterface(_, _))
        .WillOnce(SetComPointeeAndReturnOk<1>(vctx_.Get()));

    return base::MakeRefCounted<VideoProcessorProxy>(dev_, ctx_);
  }

  // Pull a random pointer off the stack, rather than relying on nullptrs.
  template <typename T>
  T* CreateGarbagePtr() {
    int foo;
    void* local = &foo;
    return static_cast<T*>(local);
  }
};

// The processor proxy wraps the VideoDevice/VideoContext and stores some of the
// d3d11 types. Make sure that the arguments we give these methods are passed
// through correctly.
TEST_F(D3D11VideoProcessorProxyUnittest, EnsureMethodPassthrough) {
  auto proxy = CreateProxy();

  // Garbage pointers are used because the proxy just passes them along and does
  // absolutely nothing with them.
  auto* texture = CreateGarbagePtr<ID3D11Texture2D>();
  auto* out_desc = CreateGarbagePtr<D3D11_VIDEO_PROCESSOR_OUTPUT_VIEW_DESC>();
  auto* in_desc = CreateGarbagePtr<D3D11_VIDEO_PROCESSOR_INPUT_VIEW_DESC>();
  auto* out_view = CreateGarbagePtr<ID3D11VideoProcessorOutputView>();
  auto* streams = CreateGarbagePtr<D3D11_VIDEO_PROCESSOR_STREAM>();

  EXPECT_CALL(*dev_.Get(), CreateVideoProcessorOutputView(
                               texture, enumerator_.Get(), out_desc, nullptr));

  EXPECT_CALL(*dev_.Get(), CreateVideoProcessorInputView(
                               texture, enumerator_.Get(), in_desc, nullptr));

  EXPECT_CALL(*vctx_.Get(),
              VideoProcessorBlt(proc_.Get(), out_view, 6, 7, streams));

  EXPECT_TRUE(proxy->Init(0, 0).is_ok());
  proxy->CreateVideoProcessorOutputView(texture, out_desc, nullptr);
  proxy->CreateVideoProcessorInputView(texture, in_desc, nullptr);
  proxy->VideoProcessorBlt(out_view, 6, 7, streams);
}

}  // namespace media