1 // Copyright (C) 2020 The Android Open Source Project
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 //      http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14 
15 #ifndef COMPUTEPIPE_RUNNER_STREAM_MOCK_ENGINE_H
16 #define COMPUTEPIPE_RUNNER_STREAM_MOCK_ENGINE_H
17 
18 #include <gmock/gmock.h>
19 #include <gtest/gtest.h>
20 
21 #include "StreamEngineInterface.h"
22 
23 namespace android {
24 namespace automotive {
25 namespace computepipe {
26 namespace runner {
27 namespace stream_manager {
28 
29 class MockEngine : public StreamEngineInterface {
30   public:
31     MockEngine() = default;
32     MOCK_METHOD(Status, dispatchPacket, (const std::shared_ptr<MemHandle>& data), (override));
33     MOCK_METHOD(void, notifyEndOfStream, (), (override));
34     MOCK_METHOD(void, notifyError, (std::string msg), (override));
35     void delegateToFake(const std::shared_ptr<StreamEngineInterface>& fake);
36 
37   private:
38     std::shared_ptr<StreamEngineInterface> mFake;
39 };
40 
41 }  // namespace stream_manager
42 }  // namespace runner
43 }  // namespace computepipe
44 }  // namespace automotive
45 }  // namespace android
46 
47 #endif
48