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
|
/*
* Copyright (C) 2021-2024 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "shared/source/os_interface/os_library.h"
#include <unordered_map>
class MockOsLibrary : public NEO::OsLibrary {
public:
MockOsLibrary(void *procAddress, bool isLoaded) : getProcAddressReturn{procAddress}, isLoadedReturn{isLoaded} {}
MockOsLibrary() {}
std::string lastRequestedProcName;
void *getProcAddressReturn = nullptr;
void *getProcAddress(const std::string &procName) override {
lastRequestedProcName = procName;
return getProcAddressReturn;
}
bool isLoadedReturn = false;
bool isLoaded() override {
return isLoadedReturn;
}
std::string getFullPath() override {
return std::string();
}
static OsLibrary *loadLibraryNewObject;
static OsLibrary *load(const std::string &name) {
OsLibrary *ptr = loadLibraryNewObject;
loadLibraryNewObject = nullptr;
return ptr;
}
};
class MockOsLibraryCustom : public MockOsLibrary {
public:
using MockOsLibrary::MockOsLibrary;
std::unordered_map<std::string, void *> procMap;
void *getProcAddress(const std::string &procName) override {
if (procMap.find(procName) != procMap.end()) {
return procMap[procName];
} else {
return getProcAddressReturn;
}
}
};
|
