blob: 0ed9af9929e0f63c5f38374f5afde71db1bda7eb [file] [log] [blame]
Lev Proleev900c28a2021-01-26 19:40:20 +00001/*
2 * Copyright (C) 2021 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
Michael Butler8414a6e2021-03-10 18:41:05 -080017#ifndef ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_AIDL_UTILS_TEST_MOCK_PREPARED_MODEL_H
18#define ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_AIDL_UTILS_TEST_MOCK_PREPARED_MODEL_H
Lev Proleev900c28a2021-01-26 19:40:20 +000019
Xusong Wang11f30c82021-10-05 14:10:41 -070020#include <aidl/android/hardware/neuralnetworks/BnExecution.h>
Lev Proleev900c28a2021-01-26 19:40:20 +000021#include <aidl/android/hardware/neuralnetworks/BnPreparedModel.h>
22#include <android/binder_interface_utils.h>
23#include <gmock/gmock.h>
24#include <gtest/gtest.h>
25#include <hidl/HidlSupport.h>
26#include <hidl/Status.h>
27
28namespace aidl::android::hardware::neuralnetworks::utils {
29
30class MockPreparedModel final : public BnPreparedModel {
31 public:
32 static std::shared_ptr<MockPreparedModel> create();
33
34 MOCK_METHOD(ndk::ScopedAStatus, executeSynchronously,
35 (const Request& request, bool measureTiming, int64_t deadline,
36 int64_t loopTimeoutDuration, ExecutionResult* executionResult),
37 (override));
38 MOCK_METHOD(ndk::ScopedAStatus, executeFenced,
39 (const Request& request, const std::vector<ndk::ScopedFileDescriptor>& waitFor,
40 bool measureTiming, int64_t deadline, int64_t loopTimeoutDuration,
41 int64_t duration, FencedExecutionResult* fencedExecutionResult),
42 (override));
Michael Butler7fc7e372021-03-10 22:51:53 -080043 MOCK_METHOD(ndk::ScopedAStatus, configureExecutionBurst, (std::shared_ptr<IBurst> * burst),
44 (override));
Xusong Wang11f30c82021-10-05 14:10:41 -070045 MOCK_METHOD(ndk::ScopedAStatus, createReusableExecution,
46 (const Request& request, bool measureTiming, int64_t loopTimeoutDuration,
47 std::shared_ptr<IExecution>* execution),
48 (override));
Lev Proleev900c28a2021-01-26 19:40:20 +000049};
50
51inline std::shared_ptr<MockPreparedModel> MockPreparedModel::create() {
52 return ndk::SharedRefBase::make<MockPreparedModel>();
53}
54
55} // namespace aidl::android::hardware::neuralnetworks::utils
56
Michael Butler8414a6e2021-03-10 18:41:05 -080057#endif // ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_AIDL_UTILS_TEST_MOCK_PREPARED_MODEL_H