57 lines
1.7 KiB
C++
57 lines
1.7 KiB
C++
/*
|
|
* Copyright (C) 2017-2018 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "runtime/command_stream/aub_command_stream_receiver.h"
|
|
#include "gmock/gmock.h"
|
|
|
|
#if defined(__clang__)
|
|
#pragma clang diagnostic push
|
|
#pragma clang diagnostic ignored "-Winconsistent-missing-override"
|
|
#endif
|
|
|
|
namespace OCLRT {
|
|
|
|
struct MockAubFileStream : public AUBCommandStreamReceiver::AubFileStream {
|
|
bool init(uint32_t stepping, uint32_t device) override {
|
|
initCalledCnt++;
|
|
return true;
|
|
}
|
|
void flush() override {
|
|
flushCalled = true;
|
|
}
|
|
std::unique_lock<std::mutex> lockStream() override {
|
|
lockStreamCalled = true;
|
|
return AUBCommandStreamReceiver::AubFileStream::lockStream();
|
|
}
|
|
void expectMemory(uint64_t physAddress, const void *memory, size_t size, uint32_t addressSpace, uint32_t compareOperation) override {
|
|
physAddressCapturedFromExpectMemory = physAddress;
|
|
memoryCapturedFromExpectMemory = reinterpret_cast<uintptr_t>(memory);
|
|
sizeCapturedFromExpectMemory = size;
|
|
addressSpaceCapturedFromExpectMemory = addressSpace;
|
|
compareOperationFromExpectMemory = compareOperation;
|
|
}
|
|
uint32_t initCalledCnt = 0;
|
|
bool flushCalled = false;
|
|
bool lockStreamCalled = false;
|
|
uint64_t physAddressCapturedFromExpectMemory = 0;
|
|
uintptr_t memoryCapturedFromExpectMemory = 0;
|
|
size_t sizeCapturedFromExpectMemory = 0;
|
|
uint32_t addressSpaceCapturedFromExpectMemory = 0;
|
|
uint32_t compareOperationFromExpectMemory = 0;
|
|
};
|
|
|
|
struct GmockAubFileStream : public AUBCommandStreamReceiver::AubFileStream {
|
|
MOCK_METHOD1(addComment, bool(const char *message));
|
|
};
|
|
} // namespace OCLRT
|
|
|
|
#if defined(__clang__)
|
|
#pragma clang diagnostic pop
|
|
#endif
|