diff options
Diffstat (limited to 'src/profiling/test')
-rw-r--r-- | src/profiling/test/BufferTests.cpp | 8 | ||||
-rw-r--r-- | src/profiling/test/SendCounterPacketTests.hpp | 22 |
2 files changed, 14 insertions, 16 deletions
diff --git a/src/profiling/test/BufferTests.cpp b/src/profiling/test/BufferTests.cpp index 7a06843ab3..cba209c9a2 100644 --- a/src/profiling/test/BufferTests.cpp +++ b/src/profiling/test/BufferTests.cpp @@ -17,7 +17,7 @@ BOOST_AUTO_TEST_SUITE(BufferTests) BOOST_AUTO_TEST_CASE(PacketBufferTest0) { - std::unique_ptr<IPacketBuffer> packetBuffer = std::make_unique<PacketBuffer>(512); + IPacketBufferPtr packetBuffer = std::make_unique<PacketBuffer>(512); BOOST_TEST(packetBuffer->GetSize() == 0); @@ -55,7 +55,7 @@ BOOST_AUTO_TEST_CASE(PacketBufferTest0) BOOST_AUTO_TEST_CASE(PacketBufferTest1) { - std::unique_ptr<IPacketBuffer> packetBuffer = std::make_unique<PacketBuffer>(512); + IPacketBufferPtr packetBuffer = std::make_unique<PacketBuffer>(512); BOOST_TEST(packetBuffer->GetSize() == 0); @@ -88,7 +88,7 @@ BOOST_AUTO_TEST_CASE(PacketBufferTest1) } BOOST_AUTO_TEST_CASE(PacketBufferReleaseTest) { - std::unique_ptr<IPacketBuffer> packetBuffer = std::make_unique<PacketBuffer>(512); + IPacketBufferPtr packetBuffer = std::make_unique<PacketBuffer>(512); BOOST_TEST(packetBuffer->GetSize() == 0); @@ -107,7 +107,7 @@ BOOST_AUTO_TEST_CASE(PacketBufferReleaseTest) { BOOST_AUTO_TEST_CASE(PacketBufferCommitErrorTest) { - std::unique_ptr<IPacketBuffer> packetBuffer = std::make_unique<PacketBuffer>(8); + IPacketBufferPtr packetBuffer = std::make_unique<PacketBuffer>(8); // Cannot commit data bigger than the max size of the buffer BOOST_CHECK_THROW(packetBuffer->Commit(16);, armnn::RuntimeException); diff --git a/src/profiling/test/SendCounterPacketTests.hpp b/src/profiling/test/SendCounterPacketTests.hpp index 73fc39b437..d70239ef1a 100644 --- a/src/profiling/test/SendCounterPacketTests.hpp +++ b/src/profiling/test/SendCounterPacketTests.hpp @@ -145,7 +145,7 @@ public: ~MockBufferManager() {} - std::unique_ptr<IPacketBuffer> Reserve(unsigned int requestedSize, unsigned int& reservedSize) override + IPacketBufferPtr Reserve(unsigned int requestedSize, unsigned int& reservedSize) override { if (requestedSize > m_BufferSize) { @@ -159,24 +159,24 @@ public: return std::move(m_Buffer); } - void Commit(std::unique_ptr<IPacketBuffer>& packetBuffer, unsigned int size) override + void Commit(IPacketBufferPtr& packetBuffer, unsigned int size) override { packetBuffer->Commit(size); m_Buffer = std::move(packetBuffer); } - std::unique_ptr<IPacketBuffer> GetReadableBuffer() override + IPacketBufferPtr GetReadableBuffer() override { return std::move(m_Buffer); } - void Release(std::unique_ptr<IPacketBuffer>& packetBuffer) override + void Release(IPacketBufferPtr& packetBuffer) override { packetBuffer->Release(); m_Buffer = std::move(packetBuffer); } - void MarkRead(std::unique_ptr<IPacketBuffer>& packetBuffer) override + void MarkRead(IPacketBufferPtr& packetBuffer) override { packetBuffer->MarkRead(); m_Buffer = std::move(packetBuffer); @@ -184,14 +184,12 @@ public: private: unsigned int m_BufferSize; - std::unique_ptr<IPacketBuffer> m_Buffer; + IPacketBufferPtr m_Buffer; }; class MockStreamCounterBuffer : public IBufferManager { public: - using IPacketBufferPtr = std::unique_ptr<IPacketBuffer>; - MockStreamCounterBuffer(unsigned int maxBufferSize = 4096) : m_MaxBufferSize(maxBufferSize) , m_BufferList() @@ -286,7 +284,7 @@ public: { std::string message("SendStreamMetaDataPacket"); unsigned int reserved = 0; - std::unique_ptr<IPacketBuffer> buffer = m_BufferManager.Reserve(1024, reserved); + IPacketBufferPtr buffer = m_BufferManager.Reserve(1024, reserved); memcpy(buffer->GetWritableData(), message.c_str(), static_cast<unsigned int>(message.size()) + 1); m_BufferManager.Commit(buffer, reserved); } @@ -295,7 +293,7 @@ public: { std::string message("SendCounterDirectoryPacket"); unsigned int reserved = 0; - std::unique_ptr<IPacketBuffer> buffer = m_BufferManager.Reserve(1024, reserved); + IPacketBufferPtr buffer = m_BufferManager.Reserve(1024, reserved); memcpy(buffer->GetWritableData(), message.c_str(), static_cast<unsigned int>(message.size()) + 1); m_BufferManager.Commit(buffer, reserved); } @@ -305,7 +303,7 @@ public: { std::string message("SendPeriodicCounterCapturePacket"); unsigned int reserved = 0; - std::unique_ptr<IPacketBuffer> buffer = m_BufferManager.Reserve(1024, reserved); + IPacketBufferPtr buffer = m_BufferManager.Reserve(1024, reserved); memcpy(buffer->GetWritableData(), message.c_str(), static_cast<unsigned int>(message.size()) + 1); m_BufferManager.Commit(buffer, reserved); } @@ -315,7 +313,7 @@ public: { std::string message("SendPeriodicCounterSelectionPacket"); unsigned int reserved = 0; - std::unique_ptr<IPacketBuffer> buffer = m_BufferManager.Reserve(1024, reserved); + IPacketBufferPtr buffer = m_BufferManager.Reserve(1024, reserved); memcpy(buffer->GetWritableData(), message.c_str(), static_cast<unsigned int>(message.size()) + 1); m_BufferManager.Commit(buffer, reserved); } |