1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <string.h>
4 #include <pb_encode.h>
5 #include <pb_decode.h>
6 #include "oneofmsg.pb.h"
7 #include "unittests.h"
8
main(int argc,char ** argv)9 int main(int argc, char **argv)
10 {
11 int status = 0;
12 uint8_t buffer[512];
13 pb_size_t msglen = 0;
14
15 {
16 pb_ostream_t stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
17 BodyMessage msg = BodyMessage_init_zero;
18
19 msg.which_body_type = BodyMessage_device_data_crypted_tag;
20 msg.body_type.device_data_crypted.size = 252;
21 memset(msg.body_type.device_data_crypted.bytes, 0xAA, 252);
22
23 TEST(pb_encode(&stream, BodyMessage_fields, &msg));
24
25 msglen = stream.bytes_written;
26 TEST(msglen > 252);
27 }
28
29 {
30 pb_istream_t stream = pb_istream_from_buffer(buffer, msglen);
31 BodyMessage msg = BodyMessage_init_zero;
32
33 TEST(pb_decode(&stream, BodyMessage_fields, &msg));
34
35 TEST(msg.which_body_type == BodyMessage_device_data_crypted_tag);
36 TEST(msg.body_type.device_data_crypted.size == 252);
37 TEST(msg.body_type.device_data_crypted.bytes[251] == 0xAA);
38 }
39
40 return status;
41 }
42
43