Skip to content

Commit 0bdcf98

Browse files
Emit extensions in debug string in wire order
PiperOrigin-RevId: 699254023
1 parent 736672c commit 0bdcf98

File tree

2 files changed

+34
-25
lines changed

2 files changed

+34
-25
lines changed

upb/text/debug_string.c

+1-2
Original file line numberDiff line numberDiff line change
@@ -185,8 +185,7 @@ static void _upb_MessageDebugString(txtenc* e, const upb_Message* msg,
185185
const upb_MiniTableExtension* ext;
186186
upb_MessageValue val_ext;
187187
iter = kUpb_Message_ExtensionBegin;
188-
while (UPB_PRIVATE(_upb_Message_NextExtensionReverse)(msg, &ext, &val_ext,
189-
&iter)) {
188+
while (upb_Message_NextExtension(msg, &ext, &val_ext, &iter)) {
190189
const upb_MiniTableField* f = &ext->UPB_PRIVATE(field);
191190
// It is not sufficient to only pass |f| as we lose valuable information
192191
// about sub-messages. It is required that we pass |ext|.

upb/text/encode_debug_test.cc

+33-23
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
#include <string>
1111

1212
#include <gtest/gtest.h>
13-
#include "absl/log/absl_log.h"
1413
#include "upb/base/string_view.h"
1514
#include "upb/base/upcast.h"
1615
#include "upb/mem/arena.h"
@@ -20,44 +19,55 @@
2019
#include "upb/test/test.upb_minitable.h"
2120
#include "upb/text/debug_string.h"
2221

23-
TEST(TextNoReflection, Extensions) {
22+
std::string GetDebugString(const upb_Message* input,
23+
const upb_MiniTable* mt_main) {
24+
// Resizing/reallocation of the buffer is not necessary since we're only
25+
// testing that we get the expected debug string.
26+
char buf[100];
27+
int options =
28+
UPB_TXTENC_NOSORT; // Does not matter, but maps will not be sorted.
29+
size_t real_size = upb_DebugString(input, mt_main, options, buf, 100);
30+
EXPECT_EQ(buf[real_size], '\0');
31+
return std::string(buf);
32+
}
33+
34+
TEST(TextNoReflection, ExtensionsString) {
2435
const upb_MiniTable* mt_main = upb_0test__ModelWithExtensions_msg_init_ptr;
2536
upb_Arena* arena = upb_Arena_New();
2637

2738
upb_test_ModelExtension1* extension1 = upb_test_ModelExtension1_new(arena);
2839
upb_test_ModelExtension1_set_str(extension1,
2940
upb_StringView_FromString("Hello"));
3041

42+
upb_test_ModelWithExtensions* msg = upb_test_ModelWithExtensions_new(arena);
43+
44+
upb_test_ModelExtension1_set_model_ext(msg, extension1, arena);
45+
46+
std::string buf = GetDebugString(UPB_UPCAST(msg), mt_main);
47+
upb_Arena_Free(arena);
48+
std::string golden = R"([1547] {
49+
25: "Hello"
50+
}
51+
)";
52+
ASSERT_EQ(buf, golden);
53+
}
54+
55+
TEST(TextNoReflection, ExtensionsInt) {
56+
const upb_MiniTable* mt_main = upb_0test__ModelWithExtensions_msg_init_ptr;
57+
upb_Arena* arena = upb_Arena_New();
58+
3159
upb_test_ModelExtension2* extension2 = upb_test_ModelExtension2_new(arena);
3260
upb_test_ModelExtension2_set_i(extension2, 5);
3361

3462
upb_test_ModelWithExtensions* msg = upb_test_ModelWithExtensions_new(arena);
3563

36-
upb_test_ModelExtension1_set_model_ext(msg, extension1, arena);
3764
upb_test_ModelExtension2_set_model_ext(msg, extension2, arena);
3865

39-
// Convert to a type of upb_Message*
40-
upb_Message* input = UPB_UPCAST(msg);
41-
// Resizing/reallocation of the buffer is not necessary since we're only
42-
// testing that we get the expected debug string.
43-
char* buf = new char[100];
44-
int options =
45-
UPB_TXTENC_NOSORT; // Does not matter, but maps will not be sorted.
46-
size_t size = 100;
47-
size_t real_size = upb_DebugString(input, mt_main, options, buf, size);
48-
ABSL_LOG(INFO) << "Buffer: \n"
49-
<< buf << "\n"
50-
<< "Size:" << real_size << "\n";
66+
std::string buf = GetDebugString(UPB_UPCAST(msg), mt_main);
67+
upb_Arena_Free(arena);
5168
std::string golden = R"([4135] {
5269
9: 5
5370
}
54-
[1547] {
55-
25: "Hello"
56-
}
5771
)";
58-
ASSERT_EQ(buf[real_size], '\0');
59-
std::string str(buf);
6072
ASSERT_EQ(buf, golden);
61-
delete[] buf;
62-
upb_Arena_Free(arena);
63-
}
73+
}

0 commit comments

Comments
 (0)