Import of the watch repository from Pebble

third_party/nanopb/tests/regression/issue_407/test_extensions.c

@@ -0,0 +1,68 @@
+#include <pb_encode.h>
+#include <pb_decode.h>
+#include "extensions.pb.h"
+#include "unittests.h"
+
+int main()
+{
+    uint8_t buffer[256];
+    int status = 0;
+    size_t msglen;
+    
+    {
+        pb_ostream_t stream;
+        SimpleMessage msg = SimpleMessage_init_zero;
+        ExtMessage extmsg = ExtMessage_init_zero;
+        pb_extension_t ext = pb_extension_init_zero;
+        
+        COMMENT("Encoding message");
+        msg.has_number = true;
+        msg.number = 1234;
+        extmsg.has_second_number = true;
+        extmsg.second_number = 5678;
+        
+        msg.extensions = &ext;
+        ext.type = &ExtMessage_ext_message_ext;
+        ext.dest = &extmsg;
+        ext.next = NULL;
+        
+        stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
+        TEST(pb_encode(&stream, SimpleMessage_fields, &msg));
+        msglen = stream.bytes_written;
+        TEST(msglen == 9); /* 3 for number, 3 for submsg tag+len, 3 for second_number */
+        TEST(buffer[0] == 0x08); /* varint 1 */
+        TEST(buffer[1] == 0xD2); /* number = 1234 */
+        TEST(buffer[2] == 0x09);
+        TEST(buffer[3] == 0xE2); /* submsg 1100 */
+        TEST(buffer[4] == 0x44);
+        TEST(buffer[5] == 0x03); /* length 3 */
+        TEST(buffer[6] == 0x08); /* varint 1 */
+        TEST(buffer[7] == 0xae); /* second_number = 5678 */
+        TEST(buffer[8] == 0x2c);
+    }
+
+    {
+        pb_istream_t stream;
+        SimpleMessage msg = SimpleMessage_init_zero;
+        ExtMessage extmsg = ExtMessage_init_zero;
+        pb_extension_t ext = pb_extension_init_zero;
+        
+        COMMENT("Decoding message");
+        
+        msg.extensions = &ext;
+        ext.type = &ExtMessage_ext_message_ext;
+        ext.dest = &extmsg;
+        ext.next = NULL;
+        ext.found = false;
+        
+        stream = pb_istream_from_buffer(buffer, msglen);
+        TEST(pb_decode(&stream, SimpleMessage_fields, &msg));
+        TEST(msg.has_number);
+        TEST(msg.number == 1234);
+        TEST(ext.found);
+        TEST(extmsg.has_second_number);
+        TEST(extmsg.second_number == 5678);
+    }
+    
+    return status;
+}