diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-09-24 18:24:47 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-09-27 15:48:18 +0000 |
commit | dfa1477f8144c8ae933f949f0cc6ab70b6ba372d (patch) | |
tree | 261804170b87e5d37633c276b7709f69868ff8e7 /src/profiling/test/ProfilingConnectionDumpToFileDecoratorTests.cpp | |
parent | 3a7d3a70d99cbe22f5e4711d5dbbea2a245da7ed (diff) | |
download | armnn-dfa1477f8144c8ae933f949f0cc6ab70b6ba372d.tar.gz |
IVGCVSW-3442 Add ProfilingConnectionDumpToFileDecorator
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I3a8313287c69268ae02b6a65103363e9fbba6b4a
Diffstat (limited to 'src/profiling/test/ProfilingConnectionDumpToFileDecoratorTests.cpp')
-rw-r--r-- | src/profiling/test/ProfilingConnectionDumpToFileDecoratorTests.cpp | 177 |
1 files changed, 177 insertions, 0 deletions
diff --git a/src/profiling/test/ProfilingConnectionDumpToFileDecoratorTests.cpp b/src/profiling/test/ProfilingConnectionDumpToFileDecoratorTests.cpp new file mode 100644 index 0000000000..3e06cb353b --- /dev/null +++ b/src/profiling/test/ProfilingConnectionDumpToFileDecoratorTests.cpp @@ -0,0 +1,177 @@ +// +// Copyright © 2019 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include "../ProfilingConnectionDumpToFileDecorator.hpp" + +#include <fstream> +#include <sstream> + +#include <boost/core/ignore_unused.hpp> +#include <boost/numeric/conversion/cast.hpp> +#include <boost/test/unit_test.hpp> + +#if defined(__ANDROID__) +#define ARMNN_PROFILING_CONNECTION_TEST_DUMP_DIR "/data/local/tmp" +#else +#define ARMNN_PROFILING_CONNECTION_TEST_DUMP_DIR "/tmp" +#endif + +using namespace armnn::profiling; + +namespace +{ + +const std::vector<char> g_Data = { 'd', 'u', 'm', 'm', 'y' }; +const uint32_t g_DataLength = boost::numeric_cast<uint32_t>(g_Data.size()); +const unsigned char* g_DataPtr = reinterpret_cast<const unsigned char*>(g_Data.data()); + +class DummyProfilingConnection : public IProfilingConnection +{ +public: + DummyProfilingConnection() + : m_Open(true) + , m_PacketData(std::make_unique<char[]>(g_DataLength)) + { + // populate packet data and construct packet + std::memcpy(m_PacketData.get(), g_DataPtr, g_DataLength); + m_Packet = std::make_unique<Packet>(0u, g_DataLength, m_PacketData); + } + + ~DummyProfilingConnection() = default; + + bool IsOpen() override + { + return m_Open; + } + + void Close() override + { + m_Open = false; + } + + bool WritePacket(const unsigned char* buffer, uint32_t length) override + { + boost::ignore_unused(buffer); + boost::ignore_unused(length); + return true; + } + + Packet ReadPacket(uint32_t timeout) override + { + boost::ignore_unused(timeout); + return std::move(*m_Packet); + } + +private: + bool m_Open; + std::unique_ptr<char[]> m_PacketData; + std::unique_ptr<Packet> m_Packet; +}; + +std::vector<char> ReadDumpFile(const std::string& dumpFileName) +{ + std::ifstream input(dumpFileName, std::ios::binary); + return std::vector<char>(std::istreambuf_iterator<char>(input), {}); +} + +} // anonymous namespace + +BOOST_AUTO_TEST_SUITE(ProfilingConnectionDumpToFileDecoratorTests) + +BOOST_AUTO_TEST_CASE(CheckSettings) +{ + ProfilingConnectionDumpToFileDecoratorSettings settings0("", ""); + BOOST_CHECK(settings0.m_DumpIncoming == false); + BOOST_CHECK(settings0.m_DumpOutgoing == false); + + ProfilingConnectionDumpToFileDecoratorSettings settings1("incomingDumpFile.dat", ""); + BOOST_CHECK(settings1.m_DumpIncoming == true); + BOOST_CHECK(settings1.m_DumpOutgoing == false); + + ProfilingConnectionDumpToFileDecoratorSettings settings2("", "outgoingDumpFile.dat"); + BOOST_CHECK(settings2.m_DumpIncoming == false); + BOOST_CHECK(settings2.m_DumpOutgoing == true); + + ProfilingConnectionDumpToFileDecoratorSettings settings3("incomingDumpFile.dat", "outgoingDumpFile.dat"); + BOOST_CHECK(settings3.m_DumpIncoming == true); + BOOST_CHECK(settings3.m_DumpOutgoing == true); +} + +BOOST_AUTO_TEST_CASE(DumpIncomingInvalidFile) +{ + ProfilingConnectionDumpToFileDecoratorSettings settings("/", "", false); + ProfilingConnectionDumpToFileDecorator decorator(std::make_unique<DummyProfilingConnection>(), settings); + BOOST_CHECK_THROW(decorator.ReadPacket(0), armnn::RuntimeException); +} + +BOOST_AUTO_TEST_CASE(DumpIncomingInvalidFileIgnoreErrors) +{ + ProfilingConnectionDumpToFileDecoratorSettings settings("/", "", true); + ProfilingConnectionDumpToFileDecorator decorator(std::make_unique<DummyProfilingConnection>(), settings); + BOOST_CHECK_NO_THROW(decorator.ReadPacket(0)); +} + +BOOST_AUTO_TEST_CASE(DumpIncomingValidFile) +{ + std::stringstream fileName; + fileName << ARMNN_PROFILING_CONNECTION_TEST_DUMP_DIR << "/test_dump_file_incoming.dat"; + + ProfilingConnectionDumpToFileDecoratorSettings settings(fileName.str(), "", false); + ProfilingConnectionDumpToFileDecorator decorator(std::make_unique<DummyProfilingConnection>(), settings); + + // NOTE: unique_ptr is needed here because operator=() is deleted for Packet + std::unique_ptr<Packet> packet; + BOOST_CHECK_NO_THROW(packet = std::make_unique<Packet>(decorator.ReadPacket(0))); + + decorator.Close(); + + std::vector<char> data = ReadDumpFile(settings.m_IncomingDumpFileName); + const char* packetData = packet->GetData(); + + // check if the data read back from the dump file matches the original + constexpr unsigned int bytesToSkip = 2u * sizeof(uint32_t); // skip header and packet length + int diff = std::strncmp(data.data() + bytesToSkip, packetData, g_DataLength); + BOOST_CHECK(diff == 0); +} + +BOOST_AUTO_TEST_CASE(DumpOutgoingInvalidFile) +{ + ProfilingConnectionDumpToFileDecoratorSettings settings("", "/", false); + ProfilingConnectionDumpToFileDecorator decorator(std::make_unique<DummyProfilingConnection>(), settings); + BOOST_CHECK_THROW(decorator.WritePacket(g_DataPtr, g_DataLength), armnn::RuntimeException); +} + +BOOST_AUTO_TEST_CASE(DumpOutgoingInvalidFileIgnoreErrors) +{ + ProfilingConnectionDumpToFileDecoratorSettings settings("", "/", true); + ProfilingConnectionDumpToFileDecorator decorator(std::make_unique<DummyProfilingConnection>(), settings); + BOOST_CHECK_NO_THROW(decorator.WritePacket(g_DataPtr, g_DataLength)); + + bool success = decorator.WritePacket(g_DataPtr, g_DataLength); + BOOST_CHECK(!success); +} + +BOOST_AUTO_TEST_CASE(DumpOutgoingValidFile) +{ + std::stringstream fileName; + fileName << ARMNN_PROFILING_CONNECTION_TEST_DUMP_DIR << "/test_dump_file.dat"; + + ProfilingConnectionDumpToFileDecoratorSettings settings("", fileName.str(), false); + ProfilingConnectionDumpToFileDecorator decorator(std::make_unique<DummyProfilingConnection>(), settings); + + bool success = false; + BOOST_CHECK_NO_THROW(success = decorator.WritePacket(g_DataPtr, g_DataLength)); + BOOST_CHECK(success); + + decorator.Close(); + + std::vector<char> data = ReadDumpFile(settings.m_OutgoingDumpFileName); + + // check if the data read back from the dump file matches the original + int diff = std::strncmp(data.data(), g_Data.data(), g_DataLength); + BOOST_CHECK(diff == 0); +} + +BOOST_AUTO_TEST_SUITE_END() |