337 lines
12 KiB
C++
337 lines
12 KiB
C++
#include "proto_test.h"
|
|
|
|
#include "flatbuffers/code_generator.h"
|
|
#include "idl_gen_fbs.h"
|
|
#include "test_assert.h"
|
|
|
|
namespace flatbuffers {
|
|
namespace tests {
|
|
|
|
void RunTest(const flatbuffers::IDLOptions &opts, const std::string &proto_path,
|
|
const std::string &proto_file, const std::string &golden_file,
|
|
const std::string import_proto_file) {
|
|
const char *include_directories[] = { proto_path.c_str(), nullptr };
|
|
|
|
// Parse proto.
|
|
flatbuffers::Parser parser(opts);
|
|
TEST_EQ(parser.Parse(proto_file.c_str(), include_directories), true);
|
|
|
|
// Generate fbs.
|
|
std::unique_ptr<CodeGenerator> fbs_generator = NewFBSCodeGenerator(true);
|
|
std::string fbs;
|
|
TEST_EQ(fbs_generator->GenerateCodeString(parser, "test", fbs),
|
|
CodeGenerator::Status::OK);
|
|
|
|
// Ensure generated file is parsable.
|
|
flatbuffers::Parser parser2;
|
|
|
|
if (!import_proto_file.empty()) {
|
|
// Generate fbs from import.proto
|
|
flatbuffers::Parser import_parser(opts);
|
|
TEST_EQ(import_parser.Parse(import_proto_file.c_str(), include_directories),
|
|
true);
|
|
std::string import_fbs;
|
|
TEST_EQ(
|
|
fbs_generator->GenerateCodeString(import_parser, "test", import_fbs),
|
|
CodeGenerator::Status::OK);
|
|
// auto import_fbs = flatbuffers::GenerateFBS(import_parser, "test", true);
|
|
// Since `imported.fbs` isn't in the filesystem AbsolutePath can't figure it
|
|
// out by itself. We manually construct it so Parser works.
|
|
std::string imported_fbs = flatbuffers::PosixPath(
|
|
flatbuffers::AbsolutePath(proto_path) + "/imported.fbs");
|
|
TEST_EQ(parser2.Parse(import_fbs.c_str(), include_directories,
|
|
imported_fbs.c_str()),
|
|
true);
|
|
}
|
|
|
|
TEST_EQ(parser2.Parse(fbs.c_str(), nullptr), true);
|
|
TEST_EQ_STR(fbs.c_str(), golden_file.c_str());
|
|
}
|
|
|
|
void proto_test(const std::string &proto_path, const std::string &proto_file) {
|
|
flatbuffers::IDLOptions opts;
|
|
opts.include_dependence_headers = false;
|
|
opts.proto_mode = true;
|
|
opts.proto_id_gap_action = IDLOptions::ProtoIdGapAction::NO_OP;
|
|
|
|
// load the .proto and the golden file from disk
|
|
std::string golden_file;
|
|
TEST_EQ(flatbuffers::LoadFile((proto_path + "test.golden.fbs").c_str(), false,
|
|
&golden_file),
|
|
true);
|
|
|
|
RunTest(opts, proto_path, proto_file, golden_file);
|
|
}
|
|
|
|
void proto_test_id(const std::string &proto_path,
|
|
const std::string &proto_file) {
|
|
flatbuffers::IDLOptions opts;
|
|
opts.include_dependence_headers = false;
|
|
opts.proto_mode = true;
|
|
opts.keep_proto_id = true;
|
|
opts.proto_id_gap_action = IDLOptions::ProtoIdGapAction::NO_OP;
|
|
|
|
// load the .proto and the golden file from disk
|
|
std::string golden_file;
|
|
TEST_EQ(flatbuffers::LoadFile((proto_path + "test_id.golden.fbs").c_str(),
|
|
false, &golden_file),
|
|
true);
|
|
|
|
RunTest(opts, proto_path, proto_file, golden_file);
|
|
}
|
|
|
|
void proto_test_union(const std::string &proto_path,
|
|
const std::string &proto_file) {
|
|
// Parse proto with --oneof-union option.
|
|
flatbuffers::IDLOptions opts;
|
|
opts.include_dependence_headers = false;
|
|
opts.proto_mode = true;
|
|
opts.proto_oneof_union = true;
|
|
opts.proto_id_gap_action = IDLOptions::ProtoIdGapAction::NO_OP;
|
|
|
|
std::string golden_file;
|
|
TEST_EQ(flatbuffers::LoadFile((proto_path + "test_union.golden.fbs").c_str(),
|
|
false, &golden_file),
|
|
true);
|
|
RunTest(opts, proto_path, proto_file, golden_file);
|
|
}
|
|
|
|
void proto_test_union_id(const std::string &proto_path,
|
|
const std::string &proto_file) {
|
|
// Parse proto with --oneof-union option.
|
|
flatbuffers::IDLOptions opts;
|
|
opts.include_dependence_headers = false;
|
|
opts.proto_mode = true;
|
|
opts.proto_oneof_union = true;
|
|
opts.keep_proto_id = true;
|
|
opts.proto_id_gap_action = IDLOptions::ProtoIdGapAction::NO_OP;
|
|
|
|
std::string golden_file;
|
|
TEST_EQ(
|
|
flatbuffers::LoadFile((proto_path + "test_union_id.golden.fbs").c_str(),
|
|
false, &golden_file),
|
|
true);
|
|
RunTest(opts, proto_path, proto_file, golden_file);
|
|
}
|
|
|
|
void proto_test_union_suffix(const std::string &proto_path,
|
|
const std::string &proto_file) {
|
|
flatbuffers::IDLOptions opts;
|
|
opts.include_dependence_headers = false;
|
|
opts.proto_mode = true;
|
|
opts.proto_namespace_suffix = "test_namespace_suffix";
|
|
opts.proto_oneof_union = true;
|
|
opts.proto_id_gap_action = IDLOptions::ProtoIdGapAction::NO_OP;
|
|
|
|
std::string golden_file;
|
|
TEST_EQ(flatbuffers::LoadFile(
|
|
(proto_path + "test_union_suffix.golden.fbs").c_str(), false,
|
|
&golden_file),
|
|
true);
|
|
RunTest(opts, proto_path, proto_file, golden_file);
|
|
}
|
|
|
|
void proto_test_union_suffix_id(const std::string &proto_path,
|
|
const std::string &proto_file) {
|
|
flatbuffers::IDLOptions opts;
|
|
opts.include_dependence_headers = false;
|
|
opts.proto_mode = true;
|
|
opts.proto_namespace_suffix = "test_namespace_suffix";
|
|
opts.proto_oneof_union = true;
|
|
opts.keep_proto_id = true;
|
|
opts.proto_id_gap_action = IDLOptions::ProtoIdGapAction::NO_OP;
|
|
|
|
std::string golden_file;
|
|
TEST_EQ(flatbuffers::LoadFile(
|
|
(proto_path + "test_union_suffix_id.golden.fbs").c_str(), false,
|
|
&golden_file),
|
|
true);
|
|
RunTest(opts, proto_path, proto_file, golden_file);
|
|
}
|
|
|
|
void proto_test_include(const std::string &proto_path,
|
|
const std::string &proto_file,
|
|
const std::string &import_proto_file) {
|
|
flatbuffers::IDLOptions opts;
|
|
opts.include_dependence_headers = true;
|
|
opts.proto_mode = true;
|
|
opts.proto_id_gap_action = IDLOptions::ProtoIdGapAction::NO_OP;
|
|
|
|
std::string golden_file;
|
|
TEST_EQ(
|
|
flatbuffers::LoadFile((proto_path + "test_include.golden.fbs").c_str(),
|
|
false, &golden_file),
|
|
true);
|
|
|
|
RunTest(opts, proto_path, proto_file, golden_file, import_proto_file);
|
|
}
|
|
|
|
void proto_test_include_id(const std::string &proto_path,
|
|
const std::string &proto_file,
|
|
const std::string &import_proto_file) {
|
|
flatbuffers::IDLOptions opts;
|
|
opts.include_dependence_headers = true;
|
|
opts.proto_mode = true;
|
|
opts.keep_proto_id = true;
|
|
opts.proto_id_gap_action = IDLOptions::ProtoIdGapAction::NO_OP;
|
|
|
|
std::string golden_file;
|
|
TEST_EQ(
|
|
flatbuffers::LoadFile((proto_path + "test_include_id.golden.fbs").c_str(),
|
|
false, &golden_file),
|
|
true);
|
|
|
|
RunTest(opts, proto_path, proto_file, golden_file, import_proto_file);
|
|
}
|
|
|
|
void proto_test_include_union(const std::string &proto_path,
|
|
const std::string &proto_file,
|
|
const std::string &import_proto_file) {
|
|
flatbuffers::IDLOptions opts;
|
|
opts.include_dependence_headers = true;
|
|
opts.proto_mode = true;
|
|
opts.proto_oneof_union = true;
|
|
opts.proto_id_gap_action = IDLOptions::ProtoIdGapAction::NO_OP;
|
|
|
|
std::string golden_file;
|
|
TEST_EQ(flatbuffers::LoadFile(
|
|
(proto_path + "test_union_include.golden.fbs").c_str(), false,
|
|
&golden_file),
|
|
true);
|
|
|
|
RunTest(opts, proto_path, proto_file, golden_file, import_proto_file);
|
|
}
|
|
|
|
void proto_test_include_union_id(const std::string &proto_path,
|
|
const std::string &proto_file,
|
|
const std::string &import_proto_file) {
|
|
flatbuffers::IDLOptions opts;
|
|
opts.include_dependence_headers = true;
|
|
opts.proto_mode = true;
|
|
opts.proto_oneof_union = true;
|
|
opts.keep_proto_id = true;
|
|
opts.proto_id_gap_action = IDLOptions::ProtoIdGapAction::NO_OP;
|
|
|
|
std::string golden_file;
|
|
TEST_EQ(flatbuffers::LoadFile(
|
|
(proto_path + "test_union_include_id.golden.fbs").c_str(), false,
|
|
&golden_file),
|
|
true);
|
|
|
|
RunTest(opts, proto_path, proto_file, golden_file, import_proto_file);
|
|
}
|
|
|
|
void ParseCorruptedProto(const std::string &proto_path) {
|
|
const char *include_directories[] = { proto_path.c_str(), nullptr };
|
|
|
|
flatbuffers::IDLOptions opts;
|
|
opts.include_dependence_headers = true;
|
|
opts.proto_mode = true;
|
|
opts.proto_oneof_union = true;
|
|
|
|
std::string proto_file;
|
|
|
|
std::unique_ptr<CodeGenerator> fbs_generator = NewFBSCodeGenerator(true);
|
|
|
|
// Parse proto with non positive id.
|
|
{
|
|
flatbuffers::Parser parser(opts);
|
|
TEST_EQ(
|
|
flatbuffers::LoadFile((proto_path + "non-positive-id.proto").c_str(),
|
|
false, &proto_file),
|
|
true);
|
|
TEST_EQ(parser.Parse(proto_file.c_str(), include_directories), true);
|
|
TEST_NE(fbs_generator->GenerateCode(parser, "temp.fbs", "test"),
|
|
CodeGenerator::Status::OK);
|
|
}
|
|
|
|
// Parse proto with twice id.
|
|
{
|
|
flatbuffers::Parser parser(opts);
|
|
TEST_EQ(flatbuffers::LoadFile((proto_path + "twice-id.proto").c_str(),
|
|
false, &proto_file),
|
|
true);
|
|
TEST_EQ(parser.Parse(proto_file.c_str(), include_directories), true);
|
|
TEST_NE(fbs_generator->GenerateCode(parser, "temp.fbs", "test"),
|
|
CodeGenerator::Status::OK);
|
|
}
|
|
|
|
// Parse proto with using reserved id.
|
|
{
|
|
flatbuffers::Parser parser(opts);
|
|
TEST_EQ(flatbuffers::LoadFile((proto_path + "twice-id.proto").c_str(),
|
|
false, &proto_file),
|
|
true);
|
|
TEST_EQ(parser.Parse(proto_file.c_str(), include_directories), true);
|
|
TEST_NE(fbs_generator->GenerateCode(parser, "temp.fbs", "test"),
|
|
CodeGenerator::Status::OK);
|
|
}
|
|
|
|
// Parse proto with error on gap.
|
|
{
|
|
opts.proto_id_gap_action = IDLOptions::ProtoIdGapAction::ERROR;
|
|
flatbuffers::Parser parser(opts);
|
|
TEST_EQ(flatbuffers::LoadFile((proto_path + "test.proto").c_str(), false,
|
|
&proto_file),
|
|
true);
|
|
TEST_EQ(parser.Parse(proto_file.c_str(), include_directories), true);
|
|
|
|
TEST_NE(fbs_generator->GenerateCode(parser, "temp.fbs", "test"),
|
|
CodeGenerator::Status::OK);
|
|
}
|
|
}
|
|
|
|
// Parse a .proto schema, output as .fbs
|
|
void ParseProtoTest(const std::string &tests_data_path) {
|
|
auto proto_path = tests_data_path + "prototest/";
|
|
std::string proto_file;
|
|
TEST_EQ(
|
|
flatbuffers::LoadFile((tests_data_path + "prototest/test.proto").c_str(),
|
|
false, &proto_file),
|
|
true);
|
|
|
|
std::string import_proto_file;
|
|
TEST_EQ(flatbuffers::LoadFile(
|
|
(tests_data_path + "prototest/imported.proto").c_str(), false,
|
|
&import_proto_file),
|
|
true);
|
|
|
|
proto_test(proto_path, proto_file);
|
|
proto_test_union(proto_path, proto_file);
|
|
proto_test_union_suffix(proto_path, proto_file);
|
|
proto_test_include(proto_path, proto_file, import_proto_file);
|
|
proto_test_include_union(proto_path, proto_file, import_proto_file);
|
|
|
|
proto_test_id(proto_path, proto_file);
|
|
proto_test_union_id(proto_path, proto_file);
|
|
proto_test_union_suffix_id(proto_path, proto_file);
|
|
proto_test_include_id(proto_path, proto_file, import_proto_file);
|
|
proto_test_include_union_id(proto_path, proto_file, import_proto_file);
|
|
|
|
ParseCorruptedProto(proto_path);
|
|
}
|
|
|
|
void ParseProtoBufAsciiTest() {
|
|
// We can put the parser in a mode where it will accept JSON that looks more
|
|
// like Protobuf ASCII, for users that have data in that format.
|
|
// This uses no "" for field names (which we already support by default,
|
|
// omits `,`, `:` before `{` and a couple of other features.
|
|
flatbuffers::Parser parser;
|
|
parser.opts.protobuf_ascii_alike = true;
|
|
TEST_EQ(
|
|
parser.Parse("table S { B:int; } table T { A:[int]; C:S; } root_type T;"),
|
|
true);
|
|
TEST_EQ(parser.Parse("{ A [1 2] C { B:2 }}"), true);
|
|
// Similarly, in text output, it should omit these.
|
|
std::string text;
|
|
auto err =
|
|
flatbuffers::GenText(
|
|
parser, parser.builder_.GetBufferPointer(), &text);
|
|
TEST_NULL(err);
|
|
TEST_EQ_STR(text.c_str(),
|
|
"{\n A [\n 1\n 2\n ]\n C {\n B: 2\n }\n}\n");
|
|
}
|
|
|
|
} // namespace tests
|
|
} // namespace flatbuffers
|