| #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 |