mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-24 04:12:57 +08:00
aed890a219
unit_test/fixtures/mock_aub_center_fixture.h -> common/fixtures unit_test/helpers/raii_hw_helper.h -> common/helpers unit_test/helpers/static_size3.h -> common/helpers unit_test/helpers/ult_limits.h -> common/helpers unit_test/memory_manager/mock_prefetch_manager.h -> common/memory_manager common/mocks/mock_aub_stream.h -> unit_test/mocks common/mocks/mock_csr_simulated_common_hw.h -> unit_test/mocks common/mocks/mock_direct_submission_diagnostic_collector.h -> unit_test/mocks common/mocks/mock_lrca_helper.h -> unit_test/mocks common/mocks/mock_tbx_stream.h -> unit_test/mocks common/mocks/linux/mock_os_context_linux.h -> unit_test/mocks/linux common/mocks/windows/mock_wddm_direct_submission.h -> unit_test/mocks/windows Related-To: NEO-6524 Signed-off-by: Warchulski, Jaroslaw <jaroslaw.warchulski@intel.com>
40 lines
1.6 KiB
C++
40 lines
1.6 KiB
C++
/*
|
|
* Copyright (C) 2018-2022 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/command_stream/tbx_command_stream_receiver_hw.h"
|
|
#include "shared/source/tbx/tbx_proto.h"
|
|
#include "shared/test/common/mocks/mock_tbx_sockets.h"
|
|
#include "shared/test/unit_test/mocks/mock_tbx_stream.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace NEO;
|
|
|
|
TEST(TbxStreamTests, givenTbxStreamWhenWriteMemoryIsCalledThenMemTypeIsSetCorrectly) {
|
|
std::unique_ptr<TbxCommandStreamReceiver::TbxStream> mockTbxStream(new MockTbxStream());
|
|
MockTbxStream *mockTbxStreamPtr = static_cast<MockTbxStream *>(mockTbxStream.get());
|
|
|
|
MockTbxSockets *mockTbxSocket = new MockTbxSockets();
|
|
mockTbxStreamPtr->socket = mockTbxSocket;
|
|
|
|
uint32_t addressSpace = AubMemDump::AddressSpaceValues::TraceLocal;
|
|
mockTbxStream->writeMemory(0, nullptr, 0, addressSpace, 0);
|
|
EXPECT_EQ(mem_types::MEM_TYPE_LOCALMEM, mockTbxSocket->typeCapturedFromWriteMemory);
|
|
|
|
addressSpace = AubMemDump::AddressSpaceValues::TraceNonlocal;
|
|
mockTbxStream->writeMemory(0, nullptr, 0, addressSpace, 0);
|
|
EXPECT_EQ(mem_types::MEM_TYPE_SYSTEM, mockTbxSocket->typeCapturedFromWriteMemory);
|
|
|
|
addressSpace = AubMemDump::AddressSpaceValues::TraceLocal;
|
|
mockTbxStream->writePTE(0, 0, addressSpace);
|
|
EXPECT_EQ(mem_types::MEM_TYPE_LOCALMEM, mockTbxSocket->typeCapturedFromWriteMemory);
|
|
|
|
addressSpace = AubMemDump::AddressSpaceValues::TraceNonlocal;
|
|
mockTbxStream->writePTE(0, 0, addressSpace);
|
|
EXPECT_EQ(mem_types::MEM_TYPE_SYSTEM, mockTbxSocket->typeCapturedFromWriteMemory);
|
|
}
|