IVGCVSW-3442 Add ProfilingConnectionDumpToFileDecorator

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I3a8313287c69268ae02b6a65103363e9fbba6b4a

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()