[GNA] Aligning GNA2 and GNA Unit testing (#952)

[platform/upstream/dldt.git] / inference-engine / tests_deprecated / unit / engines / gna / gna_mock_api.hpp

diff --git a/inference-engine/tests_deprecated/unit/engines/gna/gna_mock_api.hpp b/inference-engine/tests_deprecated/unit/engines/gna/gna_mock_api.hpp
index e716cf9..f93effc 100644 (file)
--- a/inference-engine/tests_deprecated/unit/engines/gna/gna_mock_api.hpp
+++ b/inference-engine/tests_deprecated/unit/engines/gna/gna_mock_api.hpp
@@ -4,7 +4,15 @@
 
 #pragma once
 #include <gmock/gmock-generated-function-mockers.h>
-
+#if GNA_LIB_VER == 1
+#include <gna-api.h>
+#include <gna-api-instrumentation.h>
+#include <gna-api-dumper.h>
+#else
+#include <gna2-instrumentation-api.h>
+#include <gna2-inference-api.h>
+#include <gna2-model-export-api.h>
+#endif
 #if defined(_WIN32)
     #ifdef libGNAStubs_EXPORTS
         #define GNA_STUBS_EXPORT __declspec(dllexport)
@@ -83,5 +91,96 @@ class GNACppApi {
         uint32_t sizeRequested,
         uint32_t * sizeGranted,
         void ** memoryAddress));
+
+    MOCK_METHOD1(Gna2DeviceOpen, Gna2Status (
+        uint32_t deviceIndex));
+
+    MOCK_METHOD2(Gna2DeviceSetNumberOfThreads, Gna2Status(
+        uint32_t deviceIndex,
+        uint32_t numberOfThreads));
+
+    MOCK_METHOD1(Gna2DeviceClose, Gna2Status (
+        uint32_t deviceIndex));
+
+    MOCK_METHOD1(Gna2MemoryFree, Gna2Status (
+        void * memory));
+
+    MOCK_METHOD3(Gna2StatusGetMessage, Gna2Status (
+        enum Gna2Status status,
+        char * messageBuffer,
+        uint32_t messageBufferSize));
+
+    MOCK_METHOD3(Gna2ModelCreate, Gna2Status (
+        uint32_t deviceIndex,
+        struct Gna2Model const * model,
+        uint32_t * modelId));
+
+    MOCK_METHOD1(Gna2ModelRelease, Gna2Status (
+        uint32_t modelId));
+
+    MOCK_METHOD1(Gna2ModelGetLastError, Gna2Status (
+        struct Gna2ModelError* error));
+
+    MOCK_METHOD2(Gna2RequestConfigCreate, Gna2Status (
+        uint32_t modelId,
+        uint32_t * requestConfigId));
+
+    MOCK_METHOD4(Gna2RequestConfigEnableActiveList, Gna2Status (
+        uint32_t requestConfigId,
+        uint32_t operationIndex,
+        uint32_t numberOfIndices,
+        uint32_t const * indices));
+
+    MOCK_METHOD2(Gna2RequestConfigEnableHardwareConsistency, Gna2Status (
+        uint32_t requestConfigId,
+        enum Gna2DeviceVersion deviceVersion));
+
+    MOCK_METHOD2(Gna2RequestConfigSetAccelerationMode, Gna2Status (
+        uint32_t requestConfigId,
+        enum Gna2AccelerationMode accelerationMode));
+
+    MOCK_METHOD2(Gna2RequestEnqueue, Gna2Status (
+        uint32_t requestConfigId,
+        uint32_t * requestId));
+
+    MOCK_METHOD2(Gna2RequestWait, Gna2Status (
+        uint32_t requestId,
+        uint32_t timeoutMilliseconds));
+
+    MOCK_METHOD2(Gna2ModelExportConfigCreate, Gna2Status (
+        Gna2UserAllocator userAllocator,
+        uint32_t * exportConfigId));
+
+    MOCK_METHOD1(Gna2ModelExportConfigRelease, Gna2Status (
+        uint32_t exportConfigId));
+
+    MOCK_METHOD3(Gna2ModelExportConfigSetSource, Gna2Status (
+        uint32_t exportConfigId,
+        uint32_t sourceDeviceIndex,
+        uint32_t sourceModelId));
+
+    MOCK_METHOD2(Gna2ModelExportConfigSetTarget, Gna2Status (
+        uint32_t exportConfigId,
+        enum Gna2DeviceVersion targetDeviceVersion));
+
+    MOCK_METHOD4(Gna2ModelExport, Gna2Status (
+        uint32_t exportConfigId,
+        enum Gna2ModelExportComponent componentType,
+        void ** exportBuffer,
+        uint32_t * exportBufferSize));
+
+    MOCK_METHOD2(Gna2DeviceGetVersion, Gna2Status (
+        uint32_t deviceIndex,
+        enum Gna2DeviceVersion * deviceVersion));
+
+    MOCK_METHOD4(Gna2InstrumentationConfigCreate, Gna2Status (
+        uint32_t numberOfInstrumentationPoints,
+        enum Gna2InstrumentationPoint* selectedInstrumentationPoints,
+        uint64_t * results,
+        uint32_t * instrumentationConfigId));
+
+    MOCK_METHOD2(Gna2InstrumentationConfigAssignToRequestConfig, Gna2Status (
+        uint32_t instrumentationConfigId,
+        uint32_t requestConfigId));
 #endif
 };