Diff: libuavcan/test/transport/outgoing_transfer_registry.cpp

Revision:

0:dfe6edabb8ec

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/libuavcan/test/transport/outgoing_transfer_registry.cpp	Sat Apr 14 10:25:32 2018 +0000
@@ -0,0 +1,90 @@
+/*
+ * Copyright (C) 2014 Pavel Kirienko <pavel.kirienko@gmail.com>
+ */
+
+#include <algorithm>
+#include <gtest/gtest.h>
+#include <uavcan/transport/outgoing_transfer_registry.hpp>
+#include "../clock.hpp"
+#include "transfer_test_helpers.hpp"
+
+
+TEST(OutgoingTransferRegistry, Basic)
+{
+    using uavcan::OutgoingTransferRegistryKey;
+    uavcan::PoolAllocator<uavcan::MemPoolBlockSize * 2, uavcan::MemPoolBlockSize> poolmgr;
+    uavcan::OutgoingTransferRegistry otr(poolmgr);
+
+    otr.cleanup(tsMono(1000));
+
+    static const int NUM_KEYS = 5;
+    const OutgoingTransferRegistryKey keys[NUM_KEYS] =
+    {
+        OutgoingTransferRegistryKey(123, uavcan::TransferTypeServiceRequest,   42),
+        OutgoingTransferRegistryKey(321, uavcan::TransferTypeMessageBroadcast, 0),
+        OutgoingTransferRegistryKey(213, uavcan::TransferTypeServiceRequest,   2),
+        OutgoingTransferRegistryKey(312, uavcan::TransferTypeServiceRequest,   4),
+        OutgoingTransferRegistryKey(456, uavcan::TransferTypeServiceRequest,   2)
+    };
+
+    ASSERT_EQ(0, otr.accessOrCreate(keys[0], tsMono(1000000))->get());
+    ASSERT_EQ(0, otr.accessOrCreate(keys[1], tsMono(1000000))->get());
+    ASSERT_EQ(0, otr.accessOrCreate(keys[2], tsMono(1000000))->get());
+    ASSERT_EQ(0, otr.accessOrCreate(keys[3], tsMono(1000000))->get());
+    ASSERT_FALSE(otr.accessOrCreate(keys[4], tsMono(1000000)));        // OOM
+
+    /*
+     * Incrementing a little
+     */
+    otr.accessOrCreate(keys[0], tsMono(2000000))->increment();
+    otr.accessOrCreate(keys[0], tsMono(4000000))->increment();
+    otr.accessOrCreate(keys[0], tsMono(3000000))->increment();
+    ASSERT_EQ(3, otr.accessOrCreate(keys[0], tsMono(5000000))->get());
+
+    otr.accessOrCreate(keys[2], tsMono(2000000))->increment();
+    otr.accessOrCreate(keys[2], tsMono(3000000))->increment();
+    ASSERT_EQ(2, otr.accessOrCreate(keys[2], tsMono(6000000))->get());
+
+    otr.accessOrCreate(keys[3], tsMono(9000000))->increment();
+    ASSERT_EQ(1, otr.accessOrCreate(keys[3], tsMono(4000000))->get());
+
+    ASSERT_EQ(0, otr.accessOrCreate(keys[1], tsMono(4000000))->get());
+
+    ASSERT_FALSE(otr.accessOrCreate(keys[4], tsMono(1000000)));        // Still OOM
+
+    /*
+     * Checking existence
+     * Exist: 0, 1, 2, 3
+     * Does not exist: 4
+     */
+    ASSERT_TRUE(otr.exists(keys[1].getDataTypeID(), keys[1].getTransferType()));
+    ASSERT_TRUE(otr.exists(keys[0].getDataTypeID(), keys[0].getTransferType()));
+    ASSERT_TRUE(otr.exists(keys[3].getDataTypeID(), keys[3].getTransferType()));
+    ASSERT_TRUE(otr.exists(keys[2].getDataTypeID(), keys[2].getTransferType()));
+
+    ASSERT_FALSE(otr.exists(keys[1].getDataTypeID(), keys[2].getTransferType()));  // Invalid combination
+    ASSERT_FALSE(otr.exists(keys[0].getDataTypeID(), keys[1].getTransferType()));  // Invalid combination
+    ASSERT_FALSE(otr.exists(keys[4].getDataTypeID(), keys[4].getTransferType()));  // Plain missing
+
+    /*
+     * Cleaning up
+     */
+    otr.cleanup(tsMono(4000001));    // Kills 1, 3
+    ASSERT_EQ(0, otr.accessOrCreate(keys[1], tsMono(1000000))->get());
+    ASSERT_EQ(0, otr.accessOrCreate(keys[3], tsMono(1000000))->get());
+    otr.accessOrCreate(keys[1], tsMono(5000000))->increment();
+    otr.accessOrCreate(keys[3], tsMono(5000000))->increment();
+
+    ASSERT_EQ(3, otr.accessOrCreate(keys[0], tsMono(5000000))->get());
+    ASSERT_EQ(2, otr.accessOrCreate(keys[2], tsMono(6000000))->get());
+
+    otr.cleanup(tsMono(5000001));    // Kills 1, 3 (He needs a bath, Jud. He stinks of the ground you buried him in.), 0
+    ASSERT_EQ(0, otr.accessOrCreate(keys[0], tsMono(1000000))->get());
+    ASSERT_EQ(0, otr.accessOrCreate(keys[1], tsMono(1000000))->get());
+    ASSERT_EQ(0, otr.accessOrCreate(keys[3], tsMono(1000000))->get());
+
+    ASSERT_EQ(2, otr.accessOrCreate(keys[2], tsMono(1000000))->get());
+
+    otr.cleanup(tsMono(5000001));    // Frees some memory for 4
+    ASSERT_EQ(0, otr.accessOrCreate(keys[0], tsMono(1000000))->get());
+}