1#include <gtest/gtest.h>
2
3extern "C" {
4#include "ringbuffer.h"
5#include "osi.h"
6}
7
8TEST(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
16TEST(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
34TEST(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
61TEST(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