mirror of
https://github.com/intel/compute-runtime.git
synced 2025-09-15 13:01:45 +08:00
39 lines
808 B
C++
39 lines
808 B
C++
/*
|
|
* Copyright (C) 2018-2021 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "shared/source/aub/aub_center.h"
|
|
#include "shared/source/aub/aub_stream_provider.h"
|
|
#include "shared/test/common/mocks/mock_aub_file_stream.h"
|
|
|
|
namespace NEO {
|
|
class MockAubStreamProvider : public AubStreamProvider {
|
|
public:
|
|
AubMemDump::AubFileStream *getStream() override {
|
|
return &stream;
|
|
}
|
|
|
|
protected:
|
|
MockAubFileStream stream;
|
|
};
|
|
|
|
class MockAubCenter : public AubCenter {
|
|
public:
|
|
using AubCenter::AubCenter;
|
|
using AubCenter::aubManager;
|
|
using AubCenter::aubStreamMode;
|
|
using AubCenter::stepping;
|
|
|
|
MockAubCenter() {
|
|
streamProvider.reset(new MockAubStreamProvider());
|
|
}
|
|
|
|
~MockAubCenter() override = default;
|
|
};
|
|
} // namespace NEO
|