1 #include <gtest/gtest.h> 2 3 extern "C" { 4 #include "ringbuffer.h" 5 #include "osi.h" 6 } 7 8 TEST(RingbufferTest, test_new_simple) { 9 ringbuffer_t *rb = ringbuffer_init(4096); 10 ASSERT_TRUE(rb != NULL); 11 EXPECT_EQ(4096, ringbuffer_available(rb)); 12 EXPECT_EQ(0, ringbuffer_size(rb)); 13 ringbuffer_free(rb); 14 } 15 16 TEST(RingbufferTest, test_insert_basic) { 17 ringbuffer_t *rb = ringbuffer_init(16); 18 19 uint8_t buffer[10] = {0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A}; 20 ringbuffer_insert(rb, buffer, 10); 21 EXPECT_EQ(10, ringbuffer_size(rb)); 22 EXPECT_EQ(6, ringbuffer_available(rb)); 23 24 uint8_t peek[10] = {0}; 25 size_t peeked = ringbuffer_peek(rb, peek, 10); 26 EXPECT_EQ(10, ringbuffer_size(rb)); // Ensure size doesn't change 27 EXPECT_EQ(6, ringbuffer_available(rb)); 28 EXPECT_EQ(10, peeked); 29 ASSERT_TRUE(0 == memcmp(buffer, peek, peeked)); 30 31 ringbuffer_free(rb); 32 } 33 34 TEST(RingbufferTest, test_insert_full) { 35 ringbuffer_t *rb = ringbuffer_init(5); 36 37 uint8_t aa[] = {0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA}; 38 uint8_t bb[] = {0xBB, 0xBB, 0xBB, 0xBB, 0xBB}; 39 uint8_t peek[5] = {0}; 40 41 size_t added = ringbuffer_insert(rb, aa, 7); 42 EXPECT_EQ(5, added); 43 EXPECT_EQ(0, ringbuffer_available(rb)); 44 EXPECT_EQ(5, ringbuffer_size(rb)); 45 46 added = ringbuffer_insert(rb, bb, 5); 47 EXPECT_EQ(0, added); 48 EXPECT_EQ(0, ringbuffer_available(rb)); 49 EXPECT_EQ(5, ringbuffer_size(rb)); 50 51 size_t peeked = ringbuffer_peek(rb, peek, 5); 52 EXPECT_EQ(5, peeked); 53 EXPECT_EQ(0, ringbuffer_available(rb)); 54 EXPECT_EQ(5, ringbuffer_size(rb)); 55 56 ASSERT_TRUE(0 == memcmp(aa, peek, peeked)); 57 58 ringbuffer_free(rb); 59 } 60 61 TEST(RingbufferTest, test_multi_insert_delete) { 62 ringbuffer_t *rb = ringbuffer_init(16); 63 EXPECT_EQ(16, ringbuffer_available(rb)); 64 EXPECT_EQ(0, ringbuffer_size(rb)); 65 66 // Insert some bytes 67 68 uint8_t aa[] = {0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA}; 69 size_t added = ringbuffer_insert(rb, aa, sizeof(aa)); 70 EXPECT_EQ(8, added); 71 EXPECT_EQ(8, ringbuffer_available(rb)); 72 EXPECT_EQ(8, ringbuffer_size(rb)); 73 74 uint8_t bb[] = {0xBB, 0xBB, 0xBB, 0xBB, 0xBB}; 75 ringbuffer_insert(rb, bb, sizeof(bb)); 76 EXPECT_EQ(3, ringbuffer_available(rb)); 77 EXPECT_EQ(13, ringbuffer_size(rb)); 78 79 uint8_t content[] = {0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xBB, 0xBB, 0xBB, 0xBB, 0xBB}; 80 uint8_t peek[16] = {0}; 81 size_t peeked = ringbuffer_peek(rb, peek, 16); 82 EXPECT_EQ(13, peeked); 83 ASSERT_TRUE(0 == memcmp(content, peek, peeked)); 84 85 // Delete some bytes 86 87 ringbuffer_delete(rb, sizeof(aa)); 88 EXPECT_EQ(11, ringbuffer_available(rb)); 89 EXPECT_EQ(5, ringbuffer_size(rb)); 90 91 // Add some more to wrap buffer 92 93 uint8_t cc[] = {0xCC, 0xCC, 0xCC, 0xCC, 0xCC, 0xCC, 0xCC, 0xCC, 0xCC}; 94 ringbuffer_insert(rb, cc, sizeof(cc)); 95 EXPECT_EQ(2, ringbuffer_available(rb)); 96 EXPECT_EQ(14, ringbuffer_size(rb)); 97 98 uint8_t content2[] = {0xBB, 0xBB, 0xBB, 0xBB, 0xBB, 0xCC, 0xCC}; 99 peeked = ringbuffer_peek(rb, peek, 7); 100 EXPECT_EQ(7, peeked); 101 ASSERT_TRUE(0 == memcmp(content2, peek, peeked)); 102 103 // Pop buffer 104 105 memset(peek, 0, 16); 106 size_t popped = ringbuffer_pop(rb, peek, 7); 107 EXPECT_EQ(7, popped); 108 EXPECT_EQ(9, ringbuffer_available(rb)); 109 ASSERT_TRUE(0 == memcmp(content2, peek, peeked)); 110 111 // Add more again to check head motion 112 113 uint8_t dd[] = { 0xDD, 0xDD, 0xDD, 0xDD, 0xDD, 0xDD, 0xDD, 0xDD }; 114 added = ringbuffer_insert(rb, dd, sizeof(dd)); 115 EXPECT_EQ(8, added); 116 EXPECT_EQ(1, ringbuffer_available(rb)); 117 118 // Delete everything 119 120 ringbuffer_delete(rb, 16); 121 EXPECT_EQ(16, ringbuffer_available(rb)); 122 EXPECT_EQ(0, ringbuffer_size(rb)); 123 124 // Add small token 125 126 uint8_t ae[] = { 0xAE, 0xAE, 0xAE }; 127 added = ringbuffer_insert(rb, ae, sizeof(ae)); 128 EXPECT_EQ(13, ringbuffer_available(rb)); 129 130 // Get everything 131 132 popped = ringbuffer_pop(rb, peek, 16); 133 EXPECT_EQ(added, popped); 134 EXPECT_EQ(16, ringbuffer_available(rb)); 135 EXPECT_EQ(0, ringbuffer_size(rb)); 136 ASSERT_TRUE(0 == memcmp(ae, peek, popped)); 137 138 ringbuffer_free(rb); 139 } 140
