tests: unittests: add unittests for heatshrink package

tests/unittests/tests-heatshrink/Makefile

+include $(RIOTBASE)/Makefile.base

tests/unittests/tests-heatshrink/Makefile.include

+USEPKG += heatshrink

tests/unittests/tests-heatshrink/tests-heatshrink.c

+/*
+ * Copyright (C) 2016 Kaspar Schleiser <kaspar@schleiser.de>
+ *
+ * This file is subject to the terms and conditions of the GNU Lesser
+ * General Public License v2.1. See the file LICENSE in the top level
+ * directory for more details.
+ */
+
+/**
+ * @ingroup     unittests
+ * @{
+ * @file
+ * @brief       Tests for the heatshrink compression library package
+ * @author      Kaspar Schleiser <kaspar@schleiser.de>
+ * @}
+ */
+#include <string.h>
+#include <stdio.h>
+
+#include "embUnit.h"
+
+#include "heatshrink_encoder.h"
+#include "heatshrink_decoder.h"
+
+static const char *_data = "This is a test string TEST TEST TEST TEST!";
+
+#define BUFSIZE (256U)
+
+static uint8_t _buf[BUFSIZE];
+static uint8_t _buf_res[BUFSIZE];
+
+static heatshrink_encoder _encoder;
+static heatshrink_decoder _decoder;
+
+static int _comp_uncomp(const uint8_t *buf, size_t len)
+{
+    memset(_buf, 0, sizeof(_buf));
+    memset(_buf_res, 0, sizeof(_buf_res));
+
+    heatshrink_encoder_reset(&_encoder);
+    heatshrink_decoder_reset(&_decoder);
+
+    size_t n = len;
+    uint8_t *inpos = (uint8_t*)buf;
+    uint8_t *outpos = _buf;
+
+    /* compress */
+    while(1) {
+        unsigned n_sunk = 0;
+        if (n) {
+            heatshrink_encoder_sink(&_encoder, inpos, n, &n_sunk);
+            if (n_sunk) {
+                inpos += n_sunk;
+                n -= n_sunk;
+            }
+            unsigned written = 0;
+            heatshrink_encoder_poll(&_encoder, outpos, (_buf + sizeof(_buf) - outpos), &written);
+            outpos += written;
+        }
+        else {
+            while (heatshrink_encoder_finish(&_encoder) == HSER_FINISH_MORE) {
+                unsigned written = 0;
+                heatshrink_encoder_poll(&_encoder, outpos, (_buf + sizeof(_buf) - outpos), &written);
+                outpos += written;
+            }
+            break;
+        }
+    }
+
+    /* decompress */
+    n = (outpos - _buf);
+    inpos = (uint8_t*) _buf;
+    outpos = _buf_res;
+
+    while(1) {
+        unsigned n_sunk = 0;
+        if (n) {
+            heatshrink_decoder_sink(&_decoder, inpos, n, &n_sunk);
+            if (n_sunk) {
+                inpos += n_sunk;
+                n -= n_sunk;
+            }
+            unsigned written = 0;
+            heatshrink_decoder_poll(&_decoder, outpos, (_buf + sizeof(_buf) - outpos), &written);
+            outpos += written;
+        }
+        else {
+            while (heatshrink_decoder_finish(&_decoder) == HSDR_FINISH_MORE) {
+                unsigned written = 0;
+                heatshrink_decoder_poll(&_decoder, outpos, (_buf_res + sizeof(_buf_res) - outpos), &written);
+                outpos += written;
+            }
+            break;
+        }
+    }
+
+    if ((outpos - _buf_res) - len) {
+        return -1;
+    }
+
+    return memcmp(buf, _buf_res, len);
+}
+
+static void test_heatshrink(void)
+{
+    TEST_ASSERT_EQUAL_INT(_comp_uncomp((const uint8_t*)_data, strlen(_data)), 0);
+}
+
+TestRef test_heatshrink_all(void)
+{
+    EMB_UNIT_TESTFIXTURES(fixtures) {
+        new_TestFixture(test_heatshrink),
+    };
+
+    EMB_UNIT_TESTCALLER(HeatshrinkTest, 0, 0, fixtures);
+    return (TestRef) & HeatshrinkTest;
+}
+
+void tests_heatshrink(void)
+{
+    TESTS_RUN(test_heatshrink_all());
+}