1 /*
2  * Copyright (C) 2023 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #include <sys/cdefs.h>
18 
19 #include <memory>
20 
21 #include "VirtualCameraRenderThread.h"
22 #include "VirtualCameraSessionContext.h"
23 #include "aidl/android/hardware/camera/common/CameraDeviceStatus.h"
24 #include "aidl/android/hardware/camera/common/TorchModeStatus.h"
25 #include "aidl/android/hardware/camera/device/BnCameraDeviceCallback.h"
26 #include "aidl/android/hardware/camera/device/BufferRequest.h"
27 #include "aidl/android/hardware/camera/device/BufferRequestStatus.h"
28 #include "aidl/android/hardware/camera/device/BufferStatus.h"
29 #include "aidl/android/hardware/camera/device/CaptureResult.h"
30 #include "aidl/android/hardware/camera/device/NotifyMsg.h"
31 #include "aidl/android/hardware/camera/device/StreamBuffer.h"
32 #include "aidl/android/hardware/camera/device/StreamBufferRet.h"
33 #include "android/binder_auto_utils.h"
34 #include "gmock/gmock.h"
35 #include "gtest/gtest.h"
36 #include "util/Util.h"
37 
38 namespace android {
39 namespace companion {
40 namespace virtualcamera {
41 namespace {
42 
43 using ::aidl::android::hardware::camera::common::CameraDeviceStatus;
44 using ::aidl::android::hardware::camera::common::TorchModeStatus;
45 using ::aidl::android::hardware::camera::device::BnCameraDeviceCallback;
46 using ::aidl::android::hardware::camera::device::BufferRequest;
47 using ::aidl::android::hardware::camera::device::BufferRequestStatus;
48 using ::aidl::android::hardware::camera::device::BufferStatus;
49 using ::aidl::android::hardware::camera::device::CaptureResult;
50 using ::aidl::android::hardware::camera::device::ErrorCode;
51 using ::aidl::android::hardware::camera::device::ErrorMsg;
52 using ::aidl::android::hardware::camera::device::NotifyMsg;
53 using ::aidl::android::hardware::camera::device::StreamBuffer;
54 using ::aidl::android::hardware::camera::device::StreamBufferRet;
55 using ::testing::AllOf;
56 using ::testing::ElementsAre;
57 using ::testing::Eq;
58 using ::testing::Field;
59 using ::testing::Matcher;
60 using ::testing::Property;
61 using ::testing::Return;
62 using ::testing::SizeIs;
63 
64 constexpr int kInputWidth = 640;
65 constexpr int kInputHeight = 480;
66 const Resolution kInputResolution(kInputWidth, kInputHeight);
67 
IsStreamBufferWithStatus(const int streamId,const int bufferId,const BufferStatus status)68 Matcher<StreamBuffer> IsStreamBufferWithStatus(const int streamId,
69                                                const int bufferId,
70                                                const BufferStatus status) {
71   return AllOf(Field(&StreamBuffer::streamId, Eq(streamId)),
72                Field(&StreamBuffer::bufferId, Eq(bufferId)),
73                Field(&StreamBuffer::status, Eq(status)));
74 }
75 
IsRequestErrorNotifyMsg(const int frameId)76 Matcher<NotifyMsg> IsRequestErrorNotifyMsg(const int frameId) {
77   return AllOf(Property(&NotifyMsg::getTag, Eq(NotifyMsg::error)),
78                Property(&NotifyMsg::get<NotifyMsg::error>,
79                         Field(&ErrorMsg::frameNumber, Eq(frameId))),
80                Property(&NotifyMsg::get<NotifyMsg::error>,
81                         Field(&ErrorMsg::errorStreamId, Eq(-1))),
82                Property(&NotifyMsg::get<NotifyMsg::error>,
83                         Field(&ErrorMsg::errorCode, Eq(ErrorCode::ERROR_REQUEST))));
84 }
85 
86 class MockCameraDeviceCallback : public BnCameraDeviceCallback {
87  public:
88   MOCK_METHOD(ndk::ScopedAStatus, notify, (const std::vector<NotifyMsg>&),
89               (override));
90   MOCK_METHOD(ndk::ScopedAStatus, processCaptureResult,
91               (const std::vector<CaptureResult>&), (override));
92   MOCK_METHOD(ndk::ScopedAStatus, requestStreamBuffers,
93               (const std::vector<BufferRequest>&, std::vector<StreamBufferRet>*,
94                BufferRequestStatus*),
95               (override));
96   MOCK_METHOD(ndk::ScopedAStatus, returnStreamBuffers,
97               (const std::vector<StreamBuffer>&), (override));
98 };
99 
100 class VirtualCameraRenderThreadTest : public ::testing::Test {
101  public:
SetUp()102   void SetUp() override {
103     mSessionContext = std::make_unique<VirtualCameraSessionContext>();
104     mMockCameraDeviceCallback =
105         ndk::SharedRefBase::make<MockCameraDeviceCallback>();
106     mRenderThread = std::make_unique<VirtualCameraRenderThread>(
107         *mSessionContext, kInputResolution,
108         /*reportedSensorSize*/ kInputResolution, mMockCameraDeviceCallback);
109   }
110 
111  protected:
112   std::unique_ptr<VirtualCameraSessionContext> mSessionContext;
113   std::unique_ptr<VirtualCameraRenderThread> mRenderThread;
114   std::shared_ptr<MockCameraDeviceCallback> mMockCameraDeviceCallback;
115 };
116 
TEST_F(VirtualCameraRenderThreadTest,FlushReturnsErrorForInFlightRequests)117 TEST_F(VirtualCameraRenderThreadTest, FlushReturnsErrorForInFlightRequests) {
118   const int frameNumber = 42;
119   const int firstStreamId = 1;
120   const int firstStreamBufferId = 1234;
121   const int secondStreamId = 7;
122   const int secondStreamBufferId = 4321;
123 
124   // Notify should be called with the error set to corresponding frame.
125   EXPECT_CALL(*mMockCameraDeviceCallback,
126               notify(ElementsAre(IsRequestErrorNotifyMsg(frameNumber))))
127       .WillOnce(Return(ndk::ScopedAStatus::ok()));
128 
129   // Process capture result should be called with all buffers in error state.
130   EXPECT_CALL(
131       *mMockCameraDeviceCallback,
132       processCaptureResult(ElementsAre(AllOf(
133           Field(&CaptureResult::frameNumber, frameNumber),
134           Field(&CaptureResult::outputBuffers,
135                 testing::UnorderedElementsAre(
136                     IsStreamBufferWithStatus(firstStreamId, firstStreamBufferId,
137                                              BufferStatus::ERROR),
138                     IsStreamBufferWithStatus(secondStreamId, secondStreamBufferId,
139                                              BufferStatus::ERROR)))))))
140       .WillOnce([]() { return ndk::ScopedAStatus::ok(); });
141 
142   mRenderThread->enqueueTask(std::make_unique<ProcessCaptureRequestTask>(
143       frameNumber,
144       std::vector<CaptureRequestBuffer>{
145           CaptureRequestBuffer(firstStreamId, firstStreamBufferId),
146           CaptureRequestBuffer(secondStreamId, secondStreamBufferId)}));
147 
148   mRenderThread->flush();
149 }
150 
151 }  // namespace
152 }  // namespace virtualcamera
153 }  // namespace companion
154 }  // namespace android
155