[PATCH 5/6] msgb/test: Add tests for msgb_resize_area and msgb_copy

[Jacob Erlbeck]

Sponsored-by: On-Waves ehf
---
 tests/msgb/msgb_test.c  | 53 +++++++++++++++++++++++++++++++++++++++++++++++--
 tests/msgb/msgb_test.ok |  2 ++
 2 files changed, 53 insertions(+), 2 deletions(-)

diff --git a/tests/msgb/msgb_test.c b/tests/msgb/msgb_test.c
index 412e8bb..08d9857 100644
--- a/tests/msgb/msgb_test.c
+++ b/tests/msgb/msgb_test.c
@@ -87,6 +87,7 @@ static void test_msgb_api()
        msg->l3h = msg->head - 1;
        printf("Buffer: %s\n", msgb_hexdump(msg));

+
 #if 0
 extern void msgb_reset(struct msgb *m);
 #define msgb_l1(m)     ((void *)(MSGB_CHECK2(m)->l1h))
@@ -97,15 +98,61 @@ static inline unsigned int msgb_l1len(const struct msgb 
*msgb)
        static inline unsigned char *msgb_get(struct msgb *msgb, unsigned int 
len)
        static inline unsigned char *msgb_push(struct msgb *msgb, unsigned int 
len)
        static inline unsigned char *msgb_pull(struct msgb *msgb, unsigned int 
len)
-       
+
        static inline unsigned char *msgb_pull_to_l3(struct msgb *msg)
        static inline int msgb_trim(struct msgb *msg, int len)
        static inline int msgb_l3trim(struct msgb *msg, int l3len)
        uint8_t *msgb_data(const struct msgb *msg);
-       return;
+return;
 #endif
 }

+static void test_msgb_copy()
+{
+       struct msgb *msg = msgb_alloc_headroom(4096, 128, "data");
+       struct msgb *msg2;
+
+       printf("Testing msgb_copy\n");
+
+       msg->l1h = msgb_put(msg, 20);
+       msg->l2h = msgb_put(msg, 20);
+       msg->l3h = msgb_put(msg, 20);
+       msg->l4h = msgb_put(msg, 20);
+
+       msg2 = msgb_copy(msg, "copy");
+
+       OSMO_ASSERT(msgb_length(msg) == msgb_length(msg2));
+       OSMO_ASSERT(msgb_l1len(msg) == msgb_l1len(msg2));
+       OSMO_ASSERT(msgb_l2len(msg) == msgb_l2len(msg2));
+       OSMO_ASSERT(msgb_l3len(msg) == msgb_l3len(msg2));
+       OSMO_ASSERT(msg->tail - msg->l4h == msg2->tail - msg2->l4h);
+}
+
+static void test_msgb_resize_area()
+{
+       struct msgb *msg = msgb_alloc_headroom(4096, 128, "data");
+       int rc;
+
+       printf("Testing msgb_resize_area\n");
+
+       msg->l1h = msgb_put(msg, 20);
+       msg->l2h = msgb_put(msg, 20);
+       msg->l3h = msgb_put(msg, 20);
+       msg->l4h = msgb_put(msg, 20);
+
+       rc = msgb_resize_area(msg, msg->l2h, 20, 20 + 30);
+
+       OSMO_ASSERT(rc >= 0);
+       OSMO_ASSERT(msgb_length(msg) == 80 + 30);
+       OSMO_ASSERT(msgb_l1len(msg) == 80 + 30);
+       OSMO_ASSERT(msgb_l2len(msg) == 60 + 30);
+       OSMO_ASSERT(msgb_l3len(msg) == 40);
+       OSMO_ASSERT(msg->tail - msg->l4h == 20);
+
+       rc = msgb_resize_area(msg, msg->l2h, 50, 8000);
+       OSMO_ASSERT(rc == -1);
+}
+
 static struct log_info info = {};

 int main(int argc, char **argv)
@@ -113,6 +160,8 @@ int main(int argc, char **argv)
        osmo_init_logging(&info);

        test_msgb_api();
+       test_msgb_copy();
+       test_msgb_resize_area();

        printf("Success.\n");

diff --git a/tests/msgb/msgb_test.ok b/tests/msgb/msgb_test.ok
index f8de0cd..3ace5be 100644
--- a/tests/msgb/msgb_test.ok
+++ b/tests/msgb/msgb_test.ok
@@ -18,4 +18,6 @@ Test msgb_hexdump
 Buffer: (L1=data-124) 00 00 00 00 00 00 00 00 [L2]> 00 00 00 00 [L3]> 
(L4=tail+4) 
 Buffer: (L1=data-124) 00 00 00 00 00 00 00 00 [L2]> (L3+8) 00 00 00 00 
(L4=tail+4) 
 Buffer: (L1=data-124) 00 00 00 00 00 00 00 00 [L2]> 00 00 00 00 (L3 out of 
range) (L4=tail+4) 
+Testing msgb_copy
+Testing msgb_resize_area
 Success.
-- 
1.9.1