mirror of
https://github.com/intel/compute-runtime.git
synced 2025-09-15 13:01:45 +08:00
ca5f34133b
Related-To: NEO-3964 Change-Id: Ice978e582721498d7496f989767ce7d6f5f4caf4 Signed-off-by: Filip Hazubski <filip.hazubski@intel.com>
39 lines
781 B
C++
39 lines
781 B
C++
/*
|
|
* Copyright (C) 2018-2020 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "shared/source/aub/aub_center.h"
|
|
#include "shared/source/aub/aub_stream_provider.h"
|
|
|
|
#include "opencl/test/unit_test/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;
|
|
|
|
MockAubCenter() {
|
|
streamProvider.reset(new MockAubStreamProvider());
|
|
}
|
|
|
|
~MockAubCenter() override = default;
|
|
};
|
|
} // namespace NEO
|