blob: 890a83152214a050fdf61e318161ac1acbdc1117 [file] [log] [blame]
#include "aos/flatbuffer_merge.h"
#include "gtest/gtest.h"
#include "aos/json_to_flatbuffer.h"
#include "aos/json_to_flatbuffer_generated.h"
#include "flatbuffers/minireflect.h"
namespace aos {
namespace testing {
class FlatbufferMerge : public ::testing::Test {
public:
FlatbufferMerge() {}
void ExpectMergedOutput(const flatbuffers::DetachedBuffer &fb_merged,
std::string_view expected_output) {
ASSERT_NE(fb_merged.size(), 0u);
const ::std::string merged_output =
FlatbufferToJson(fb_merged, ConfigurationTypeTable());
EXPECT_EQ(expected_output, merged_output);
aos::FlatbufferDetachedBuffer<Configuration> expected_message(
JsonToFlatbuffer(std::string(expected_output).c_str(),
ConfigurationTypeTable()));
EXPECT_TRUE(
CompareFlatBuffer(flatbuffers::GetRoot<Configuration>(fb_merged.data()),
&expected_message.message()));
}
void JsonMerge(const ::std::string in1, const ::std::string in2,
const ::std::string out) {
printf("Merging: %s\n", in1.c_str());
printf("Merging: %s\n", in2.c_str());
const flatbuffers::DetachedBuffer fb1 = JsonToFlatbuffer(
static_cast<const char *>(in1.c_str()), ConfigurationTypeTable());
const ::std::string in1_nested = "{ \"nested_config\": " + in1 + " }";
const flatbuffers::DetachedBuffer fb1_nested =
JsonToFlatbuffer(static_cast<const char *>(in1_nested.c_str()),
ConfigurationTypeTable());
const flatbuffers::DetachedBuffer fb2 = JsonToFlatbuffer(
static_cast<const char *>(in2.c_str()), ConfigurationTypeTable());
const ::std::string in2_nested = "{ \"nested_config\": " + in2 + " }";
const flatbuffers::DetachedBuffer fb2_nested =
JsonToFlatbuffer(static_cast<const char *>(in2_nested.c_str()),
ConfigurationTypeTable());
const ::std::string out_nested = "{ \"nested_config\": " + out + " }";
const flatbuffers::DetachedBuffer empty =
JsonToFlatbuffer("{ }", ConfigurationTypeTable());
ASSERT_NE(fb1.size(), 0u);
ASSERT_NE(fb2.size(), 0u);
ASSERT_NE(fb1_nested.size(), 0u);
ASSERT_NE(fb2_nested.size(), 0u);
// We now want to run 7 tests.
// in1 merged "" -> in1.
// in2 merged "" -> in2.
// "" merged in1 -> in1.
// "" merged in2 -> in2.
// nullptr merged in1 -> in1.
// in1 merged nullptr -> in1.
// in1 merged in2 -> out.
{
// in1 merged with "" => in1.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(fb1.data(), empty.data());
ASSERT_NE(fb_merged.size(), 0u);
EXPECT_EQ(in1, FlatbufferToJson(fb_merged, ConfigurationTypeTable()));
}
{
// in2 merged with "" => in2.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(fb2.data(), empty.data());
ASSERT_NE(fb_merged.size(), 0u);
EXPECT_EQ(in2, FlatbufferToJson(fb_merged, ConfigurationTypeTable()));
}
{
// "" merged with in1 => in1.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(empty.data(), fb1.data());
ASSERT_NE(fb_merged.size(), 0u);
EXPECT_EQ(in1, FlatbufferToJson(fb_merged, ConfigurationTypeTable()));
}
{
// "" merged with in2 => in2.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(empty.data(), fb2.data());
ASSERT_NE(fb_merged.size(), 0u);
EXPECT_EQ(in2, FlatbufferToJson(fb_merged, ConfigurationTypeTable()));
}
{
// nullptr merged with in1 => in1.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(nullptr, fb1.data());
ASSERT_NE(fb_merged.size(), 0u);
EXPECT_EQ(in1, FlatbufferToJson(fb_merged, ConfigurationTypeTable()));
}
{
// in1 merged with nullptr => in1.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(fb1.data(), nullptr);
ASSERT_NE(fb_merged.size(), 0u);
EXPECT_EQ(in1, FlatbufferToJson(fb_merged, ConfigurationTypeTable()));
}
{
// in1 merged with in2 => out.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(fb1.data(), fb2.data());
ExpectMergedOutput(fb_merged, out);
}
// Test all the different merge methods:
ExpectMergedOutput(
MergeFlatBuffers(ConfigurationTypeTable(), fb1.data(), fb2.data()),
out);
{
flatbuffers::FlatBufferBuilder fbb;
fbb.ForceDefaults(true);
fbb.Finish(MergeFlatBuffers(
ConfigurationTypeTable(),
flatbuffers::GetRoot<flatbuffers::Table>(fb1.data()),
flatbuffers::GetRoot<flatbuffers::Table>(fb2.data()), &fbb));
ExpectMergedOutput(fbb.Release(), out);
}
{
flatbuffers::FlatBufferBuilder fbb;
fbb.ForceDefaults(true);
fbb.Finish(MergeFlatBuffers<Configuration>(
flatbuffers::GetRoot<flatbuffers::Table>(fb1.data()),
flatbuffers::GetRoot<flatbuffers::Table>(fb2.data()), &fbb));
ExpectMergedOutput(fbb.Release(), out);
}
ExpectMergedOutput(MergeFlatBuffers<Configuration>(fb1.data(), fb2.data()),
out);
ExpectMergedOutput(MergeFlatBuffers<Configuration>(fb1, fb2), out);
ExpectMergedOutput(MergeFlatBuffers<Configuration>(
flatbuffers::GetRoot<Configuration>(fb1.data()),
flatbuffers::GetRoot<Configuration>(fb2.data()))
.buffer(),
out);
// Now, to make things extra exciting, nest a config inside a config. And
// run all the tests. This will exercise some fun nested merges and copies.
{
// in1_nested merged with "" => in1.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(fb1_nested.data(), empty.data());
ASSERT_NE(fb_merged.size(), 0u);
EXPECT_EQ(in1_nested,
FlatbufferToJson(fb_merged, ConfigurationTypeTable()));
}
{
// in2_nested merged with "" => in2_nested.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(fb2_nested.data(), empty.data());
ASSERT_NE(fb_merged.size(), 0u);
EXPECT_EQ(in2_nested,
FlatbufferToJson(fb_merged, ConfigurationTypeTable()));
}
{
// "" merged with in1_nested => in1_nested.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(empty.data(), fb1_nested.data());
ASSERT_NE(fb_merged.size(), 0u);
EXPECT_EQ(in1_nested,
FlatbufferToJson(fb_merged, ConfigurationTypeTable()));
}
{
// "" merged with in2_nested => in2_nested.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(empty.data(), fb2_nested.data());
ASSERT_NE(fb_merged.size(), 0u);
EXPECT_EQ(in2_nested,
FlatbufferToJson(fb_merged, ConfigurationTypeTable()));
}
{
// nullptr merged with in1_nested => in1_nested.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(nullptr, fb1_nested.data());
ASSERT_NE(fb_merged.size(), 0u);
EXPECT_EQ(in1_nested,
FlatbufferToJson(fb_merged, ConfigurationTypeTable()));
}
{
// nullptr merged with in1_nested => in1_nested.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(fb1_nested.data(), nullptr);
ASSERT_NE(fb_merged.size(), 0u);
EXPECT_EQ(in1_nested,
FlatbufferToJson(fb_merged, ConfigurationTypeTable()));
}
{
// in1_nested merged with in2_nested => out_nested.
flatbuffers::DetachedBuffer fb_merged =
MergeFlatBuffers<Configuration>(fb1_nested, fb2_nested);
ASSERT_NE(fb_merged.size(), 0u);
EXPECT_EQ(out_nested,
FlatbufferToJson(fb_merged, ConfigurationTypeTable()));
}
}
};
// Test the easy ones. Test every type, single, no nesting.
TEST_F(FlatbufferMerge, Basic) {
JsonMerge("{ }", "{ }", "{ }");
JsonMerge("{ \"foo_bool\": true }", "{ \"foo_bool\": false }",
"{ \"foo_bool\": false }");
JsonMerge("{ \"foo_bool\": false }", "{ \"foo_bool\": true }",
"{ \"foo_bool\": true }");
JsonMerge("{ \"foo_byte\": 5 }", "{ \"foo_byte\": -7 }",
"{ \"foo_byte\": -7 }");
JsonMerge("{ \"foo_ubyte\": 5 }", "{ \"foo_ubyte\": 7 }",
"{ \"foo_ubyte\": 7 }");
JsonMerge("{ \"foo_short\": 5 }", "{ \"foo_short\": 7 }",
"{ \"foo_short\": 7 }");
JsonMerge("{ \"foo_int\": 5 }", "{ \"foo_int\": 7 }", "{ \"foo_int\": 7 }");
JsonMerge("{ \"foo_uint\": 5 }", "{ \"foo_uint\": 7 }",
"{ \"foo_uint\": 7 }");
JsonMerge("{ \"foo_long\": 5 }", "{ \"foo_long\": 7 }",
"{ \"foo_long\": 7 }");
JsonMerge("{ \"foo_ulong\": 5 }", "{ \"foo_ulong\": 7 }",
"{ \"foo_ulong\": 7 }");
JsonMerge("{ \"foo_float\": 5.0 }", "{ \"foo_float\": 7.1 }",
"{ \"foo_float\": 7.1 }");
JsonMerge("{ \"foo_double\": 5.0 }", "{ \"foo_double\": 7.1 }",
"{ \"foo_double\": 7.1 }");
}
// Test arrays of simple types.
TEST_F(FlatbufferMerge, Array) {
JsonMerge("{ \"foo_string\": \"baz\" }", "{ \"foo_string\": \"bar\" }",
"{ \"foo_string\": \"bar\" }");
JsonMerge("{ \"vector_foo_bool\": [ true, true, false ] }",
"{ \"vector_foo_bool\": [ false, true, true, false ] }",
"{ \"vector_foo_bool\": [ true, true, false, false, "
"true, true, false ] }");
JsonMerge("{ \"vector_foo_byte\": [ 9, 7, 1 ] }",
"{ \"vector_foo_byte\": [ -3, 1, 3, 2 ] }",
"{ \"vector_foo_byte\": [ 9, 7, 1, -3, 1, 3, 2 ] }");
JsonMerge("{ \"vector_foo_ubyte\": [ 9, 7, 1 ] }",
"{ \"vector_foo_ubyte\": [ 3, 1, 3, 2 ] }",
"{ \"vector_foo_ubyte\": [ 9, 7, 1, 3, 1, 3, 2 ] }");
JsonMerge("{ \"vector_foo_short\": [ 9, 7, 1 ] }",
"{ \"vector_foo_short\": [ -3, 1, 3, 2 ] }",
"{ \"vector_foo_short\": [ 9, 7, 1, -3, 1, 3, 2 ] }");
JsonMerge("{ \"vector_foo_ushort\": [ 9, 7, 1 ] }",
"{ \"vector_foo_ushort\": [ 3, 1, 3, 2 ] }",
"{ \"vector_foo_ushort\": [ 9, 7, 1, 3, 1, 3, 2 ] }");
JsonMerge("{ \"vector_foo_int\": [ 9, 7, 1 ] }",
"{ \"vector_foo_int\": [ -3, 1, 3, 2 ] }",
"{ \"vector_foo_int\": [ 9, 7, 1, -3, 1, 3, 2 ] }");
JsonMerge("{ \"vector_foo_uint\": [ 9, 7, 1 ] }",
"{ \"vector_foo_uint\": [ 3, 1, 3, 2 ] }",
"{ \"vector_foo_uint\": [ 9, 7, 1, 3, 1, 3, 2 ] }");
JsonMerge("{ \"vector_foo_long\": [ 9, 7, 1 ] }",
"{ \"vector_foo_long\": [ -3, 1, 3, 2 ] }",
"{ \"vector_foo_long\": [ 9, 7, 1, -3, 1, 3, 2 ] }");
JsonMerge("{ \"vector_foo_ulong\": [ 9, 7, 1 ] }",
"{ \"vector_foo_ulong\": [ 3, 1, 3, 2 ] }",
"{ \"vector_foo_ulong\": [ 9, 7, 1, 3, 1, 3, 2 ] }");
JsonMerge("{ \"vector_foo_float\": [ 9.0, 7.0, 1.0 ] }",
"{ \"vector_foo_float\": [ -3.0, 1.3, 3.0, 2.0 ] }",
"{ \"vector_foo_float\": [ 9.0, 7.0, 1.0, -3.0, 1.3, 3.0, 2.0 ] }");
JsonMerge(
"{ \"vector_foo_string\": [ \"9\", \"7\", \"1 \" ] }",
"{ \"vector_foo_string\": [ \"31\", \"32\" ] }",
"{ \"vector_foo_string\": [ \"9\", \"7\", \"1 \", \"31\", \"32\" ] }");
}
// Test nested messages, and arrays of nested messages.
TEST_F(FlatbufferMerge, NestedStruct) {
JsonMerge(
"{ \"single_application\": { \"name\": \"woot\" } }",
"{ \"single_application\": { \"name\": \"wow\", \"priority\": 7 } }",
"{ \"single_application\": { \"name\": \"wow\", \"priority\": 7 } }");
JsonMerge(
"{ \"single_application\": { \"name\": \"wow\", \"priority\": 7 } }",
"{ \"single_application\": { \"name\": \"woot\" } }",
"{ \"single_application\": { \"name\": \"woot\", \"priority\": 7 } }");
JsonMerge("{ \"apps\": [ { \"name\": \"woot\" }, { \"name\": \"wow\" } ] }",
"{ \"apps\": [ { \"name\": \"woo2\" }, { \"name\": \"wo3\" } ] }",
"{ \"apps\": [ { \"name\": \"woot\" }, { \"name\": \"wow\" }, { "
"\"name\": \"woo2\" }, { \"name\": \"wo3\" } ] }");
}
// Tests a compare of 2 basic (different) messages.
TEST_F(FlatbufferMerge, CompareDifferent) {
aos::FlatbufferDetachedBuffer<Configuration> message1(JsonToFlatbuffer(
"{ \"single_application\": { \"name\": \"wow\", \"priority\": 7 } }",
ConfigurationTypeTable()));
aos::FlatbufferDetachedBuffer<Configuration> message2(JsonToFlatbuffer(
"{ \"single_application\": { \"name\": \"wow\", \"priority\": 8 } }",
ConfigurationTypeTable()));
EXPECT_FALSE(CompareFlatBuffer(&message1.message(), &message2.message()));
}
// TODO(austin): enums
// TODO(austin): unions
// TODO(austin): struct
} // namespace testing
} // namespace aos