Joshua Haberman
5 years ago
parent
7f5fe52dfa
commit
2a85bef825
19 changed files with 509 additions and 409 deletions
@ -1,80 +0,0 @@ |
|||||||
|
|
||||||
-- Require "pb" first to ensure that the transitive require of "upb" is |
|
||||||
-- handled properly by the "pb" module. |
|
||||||
local pb = require "upb.pb" |
|
||||||
local upb = require "upb" |
|
||||||
local lunit = require "lunit" |
|
||||||
|
|
||||||
if _VERSION >= 'Lua 5.2' then |
|
||||||
_ENV = lunit.module("testupb_pb", "seeall") |
|
||||||
else |
|
||||||
module("testupb_pb", lunit.testcase, package.seeall) |
|
||||||
end |
|
||||||
|
|
||||||
local symtab = upb.SymbolTable{ |
|
||||||
upb.MessageDef{full_name = "TestMessage", fields = { |
|
||||||
upb.FieldDef{name = "i32", number = 1, type = upb.TYPE_INT32}, |
|
||||||
upb.FieldDef{name = "u32", number = 2, type = upb.TYPE_UINT32}, |
|
||||||
upb.FieldDef{name = "i64", number = 3, type = upb.TYPE_INT64}, |
|
||||||
upb.FieldDef{name = "u64", number = 4, type = upb.TYPE_UINT64}, |
|
||||||
upb.FieldDef{name = "dbl", number = 5, type = upb.TYPE_DOUBLE}, |
|
||||||
upb.FieldDef{name = "flt", number = 6, type = upb.TYPE_FLOAT}, |
|
||||||
upb.FieldDef{name = "bool", number = 7, type = upb.TYPE_BOOL}, |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
local factory = upb.MessageFactory(symtab); |
|
||||||
local TestMessage = factory:get_message_class("TestMessage") |
|
||||||
|
|
||||||
function test_parse_primitive() |
|
||||||
local binary_pb = |
|
||||||
"\008\128\128\128\128\002\016\128\128\128\128\004\024\128\128" |
|
||||||
.. "\128\128\128\128\128\002\032\128\128\128\128\128\128\128\001\041\000" |
|
||||||
.. "\000\000\000\000\000\248\063\053\000\000\096\064\056\001" |
|
||||||
local msg = TestMessage() |
|
||||||
pb.decode(msg, binary_pb) |
|
||||||
assert_equal(536870912, msg.i32) |
|
||||||
assert_equal(1073741824, msg.u32) |
|
||||||
assert_equal(1125899906842624, msg.i64) |
|
||||||
assert_equal(562949953421312, msg.u64) |
|
||||||
assert_equal(1.5, msg.dbl) |
|
||||||
assert_equal(3.5, msg.flt) |
|
||||||
assert_equal(true, msg.bool) |
|
||||||
|
|
||||||
local encoded = pb.encode(msg) |
|
||||||
local msg2 = TestMessage() |
|
||||||
pb.decode(msg2, encoded) |
|
||||||
assert_equal(536870912, msg.i32) |
|
||||||
assert_equal(1073741824, msg.u32) |
|
||||||
assert_equal(1125899906842624, msg.i64) |
|
||||||
assert_equal(562949953421312, msg.u64) |
|
||||||
assert_equal(1.5, msg.dbl) |
|
||||||
assert_equal(3.5, msg.flt) |
|
||||||
assert_equal(true, msg.bool) |
|
||||||
end |
|
||||||
|
|
||||||
function test_parse_string() |
|
||||||
local symtab = upb.SymbolTable{ |
|
||||||
upb.MessageDef{full_name = "TestMessage", fields = { |
|
||||||
upb.FieldDef{name = "str", number = 1, type = upb.TYPE_STRING}, |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
local factory = upb.MessageFactory(symtab); |
|
||||||
local TestMessage = factory:get_message_class("TestMessage") |
|
||||||
|
|
||||||
local binary_pb = "\010\005Hello" |
|
||||||
msg = TestMessage() |
|
||||||
pb.decode(msg, binary_pb) |
|
||||||
-- TODO(haberman): re-enable when this stuff works better. |
|
||||||
-- assert_equal("Hello", msg.str) |
|
||||||
end |
|
||||||
|
|
||||||
|
|
||||||
local stats = lunit.main() |
|
||||||
|
|
||||||
if stats.failed > 0 or stats.errors > 0 then |
|
||||||
error("One or more errors in test suite") |
|
||||||
end |
|
||||||
@ -1,89 +0,0 @@ |
|||||||
/*
|
|
||||||
** Functions for use by generated code. These are not public and users must |
|
||||||
** not call them directly. |
|
||||||
*/ |
|
||||||
|
|
||||||
#ifndef UPB_GENERATED_UTIL_H_ |
|
||||||
#define UPB_GENERATED_UTIL_H_ |
|
||||||
|
|
||||||
#include <stdint.h> |
|
||||||
#include "upb/msg.h" |
|
||||||
|
|
||||||
#include "upb/port_def.inc" |
|
||||||
|
|
||||||
#define PTR_AT(msg, ofs, type) (type*)((const char*)msg + ofs) |
|
||||||
|
|
||||||
UPB_INLINE const void *_upb_array_accessor(const void *msg, size_t ofs, |
|
||||||
size_t *size) { |
|
||||||
const upb_array *arr = *PTR_AT(msg, ofs, const upb_array*); |
|
||||||
if (arr) { |
|
||||||
if (size) *size = arr->len; |
|
||||||
return _upb_array_constptr(arr); |
|
||||||
} else { |
|
||||||
if (size) *size = 0; |
|
||||||
return NULL; |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
UPB_INLINE void *_upb_array_mutable_accessor(void *msg, size_t ofs, |
|
||||||
size_t *size) { |
|
||||||
upb_array *arr = *PTR_AT(msg, ofs, upb_array*); |
|
||||||
if (arr) { |
|
||||||
if (size) *size = arr->len; |
|
||||||
return _upb_array_ptr(arr); |
|
||||||
} else { |
|
||||||
if (size) *size = 0; |
|
||||||
return NULL; |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
UPB_INLINE void *_upb_array_resize_accessor(void *msg, size_t ofs, size_t size, |
|
||||||
upb_fieldtype_t type, |
|
||||||
upb_arena *arena) { |
|
||||||
upb_array **arr_ptr = PTR_AT(msg, ofs, upb_array*); |
|
||||||
upb_array *arr = *arr_ptr; |
|
||||||
if (!arr || arr->size < size) { |
|
||||||
return _upb_array_resize_fallback(arr_ptr, size, type, arena); |
|
||||||
} |
|
||||||
arr->len = size; |
|
||||||
return _upb_array_ptr(arr); |
|
||||||
} |
|
||||||
|
|
||||||
UPB_INLINE bool _upb_array_append_accessor(void *msg, size_t ofs, |
|
||||||
size_t elem_size, |
|
||||||
upb_fieldtype_t type, |
|
||||||
const void *value, |
|
||||||
upb_arena *arena) { |
|
||||||
upb_array **arr_ptr = PTR_AT(msg, ofs, upb_array*); |
|
||||||
upb_array *arr = *arr_ptr; |
|
||||||
void* ptr; |
|
||||||
if (!arr || arr->len == arr->size) { |
|
||||||
return _upb_array_append_fallback(arr_ptr, value, type, arena); |
|
||||||
} |
|
||||||
ptr = _upb_array_ptr(arr); |
|
||||||
memcpy(PTR_AT(ptr, arr->len * elem_size, char), value, elem_size); |
|
||||||
arr->len++; |
|
||||||
return true; |
|
||||||
} |
|
||||||
|
|
||||||
UPB_INLINE bool _upb_has_field(const void *msg, size_t idx) { |
|
||||||
return (*PTR_AT(msg, idx / 8, const char) & (1 << (idx % 8))) != 0; |
|
||||||
} |
|
||||||
|
|
||||||
UPB_INLINE bool _upb_sethas(const void *msg, size_t idx) { |
|
||||||
return (*PTR_AT(msg, idx / 8, char)) |= (char)(1 << (idx % 8)); |
|
||||||
} |
|
||||||
|
|
||||||
UPB_INLINE bool _upb_clearhas(const void *msg, size_t idx) { |
|
||||||
return (*PTR_AT(msg, idx / 8, char)) &= (char)(~(1 << (idx % 8))); |
|
||||||
} |
|
||||||
|
|
||||||
UPB_INLINE bool _upb_has_oneof_field(const void *msg, size_t case_ofs, int32_t num) { |
|
||||||
return *PTR_AT(msg, case_ofs, int32_t) == num; |
|
||||||
} |
|
||||||
|
|
||||||
#undef PTR_AT |
|
||||||
|
|
||||||
#include "upb/port_undef.inc" |
|
||||||
|
|
||||||
#endif /* UPB_GENERATED_UTIL_H_ */ |
|
||||||
Loading…
Reference in new issue