|
|
|
@ -0,0 +1,436 @@ |
|
|
|
|
#region Copyright notice and license |
|
|
|
|
// Protocol Buffers - Google's data interchange format |
|
|
|
|
// Copyright 2015 Google Inc. All rights reserved. |
|
|
|
|
// https://developers.google.com/protocol-buffers/ |
|
|
|
|
// |
|
|
|
|
// Redistribution and use in source and binary forms, with or without |
|
|
|
|
// modification, are permitted provided that the following conditions are |
|
|
|
|
// met: |
|
|
|
|
// |
|
|
|
|
// * Redistributions of source code must retain the above copyright |
|
|
|
|
// notice, this list of conditions and the following disclaimer. |
|
|
|
|
// * Redistributions in binary form must reproduce the above |
|
|
|
|
// copyright notice, this list of conditions and the following disclaimer |
|
|
|
|
// in the documentation and/or other materials provided with the |
|
|
|
|
// distribution. |
|
|
|
|
// * Neither the name of Google Inc. nor the names of its |
|
|
|
|
// contributors may be used to endorse or promote products derived from |
|
|
|
|
// this software without specific prior written permission. |
|
|
|
|
// |
|
|
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
|
|
|
|
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
|
|
|
|
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
|
|
|
|
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
|
|
|
|
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
|
|
|
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
|
|
|
|
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
|
|
|
|
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
|
|
|
|
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
|
|
|
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
|
|
|
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
|
|
|
#endregion |
|
|
|
|
|
|
|
|
|
using System.Collections.Generic; |
|
|
|
|
using Google.Protobuf.Collections; |
|
|
|
|
using Google.Protobuf.TestProtos; |
|
|
|
|
using NUnit.Framework; |
|
|
|
|
using Google.Protobuf.WellKnownTypes; |
|
|
|
|
|
|
|
|
|
namespace Google.Protobuf |
|
|
|
|
{ |
|
|
|
|
public class FieldMaskTreeTest |
|
|
|
|
{ |
|
|
|
|
[Test] |
|
|
|
|
public void AddFieldPath() |
|
|
|
|
{ |
|
|
|
|
FieldMaskTree tree = new FieldMaskTree(); |
|
|
|
|
RepeatedField<string> paths = tree.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(0, paths.Count); |
|
|
|
|
|
|
|
|
|
tree.AddFieldPath(""); |
|
|
|
|
paths = tree.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(1, paths.Count); |
|
|
|
|
Assert.Contains("", paths); |
|
|
|
|
|
|
|
|
|
// New branch. |
|
|
|
|
tree.AddFieldPath("foo"); |
|
|
|
|
paths = tree.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(2, paths.Count); |
|
|
|
|
Assert.Contains("foo", paths); |
|
|
|
|
|
|
|
|
|
// Redundant path. |
|
|
|
|
tree.AddFieldPath("foo"); |
|
|
|
|
paths = tree.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(2, paths.Count); |
|
|
|
|
|
|
|
|
|
// New branch. |
|
|
|
|
tree.AddFieldPath("bar.baz"); |
|
|
|
|
paths = tree.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(3, paths.Count); |
|
|
|
|
Assert.Contains("bar.baz", paths); |
|
|
|
|
|
|
|
|
|
// Redundant sub-path. |
|
|
|
|
tree.AddFieldPath("foo.bar"); |
|
|
|
|
paths = tree.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(3, paths.Count); |
|
|
|
|
|
|
|
|
|
// New branch from a non-root node. |
|
|
|
|
tree.AddFieldPath("bar.quz"); |
|
|
|
|
paths = tree.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(4, paths.Count); |
|
|
|
|
Assert.Contains("bar.quz", paths); |
|
|
|
|
|
|
|
|
|
// A path that matches several existing sub-paths. |
|
|
|
|
tree.AddFieldPath("bar"); |
|
|
|
|
paths = tree.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(3, paths.Count); |
|
|
|
|
Assert.Contains("foo", paths); |
|
|
|
|
Assert.Contains("bar", paths); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
[Test] |
|
|
|
|
public void MergeFromFieldMask() |
|
|
|
|
{ |
|
|
|
|
FieldMaskTree tree = new FieldMaskTree(); |
|
|
|
|
tree.MergeFromFieldMask(new FieldMask |
|
|
|
|
{ |
|
|
|
|
Paths = {"foo", "bar.baz", "bar.quz"} |
|
|
|
|
}); |
|
|
|
|
RepeatedField<string> paths = tree.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(3, paths.Count); |
|
|
|
|
Assert.Contains("foo", paths); |
|
|
|
|
Assert.Contains("bar.baz", paths); |
|
|
|
|
Assert.Contains("bar.quz", paths); |
|
|
|
|
|
|
|
|
|
tree.MergeFromFieldMask(new FieldMask |
|
|
|
|
{ |
|
|
|
|
Paths = {"foo.bar", "bar"} |
|
|
|
|
}); |
|
|
|
|
paths = tree.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(2, paths.Count); |
|
|
|
|
Assert.Contains("foo", paths); |
|
|
|
|
Assert.Contains("bar", paths); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
[Test] |
|
|
|
|
public void IntersectFieldPath() |
|
|
|
|
{ |
|
|
|
|
FieldMaskTree tree = new FieldMaskTree(); |
|
|
|
|
FieldMaskTree result = new FieldMaskTree(); |
|
|
|
|
tree.MergeFromFieldMask(new FieldMask |
|
|
|
|
{ |
|
|
|
|
Paths = {"foo", "bar.baz", "bar.quz"} |
|
|
|
|
}); |
|
|
|
|
|
|
|
|
|
// Empty path. |
|
|
|
|
tree.IntersectFieldPath("", result); |
|
|
|
|
RepeatedField<string> paths = result.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(0, paths.Count); |
|
|
|
|
|
|
|
|
|
// Non-exist path. |
|
|
|
|
tree.IntersectFieldPath("quz", result); |
|
|
|
|
paths = result.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(0, paths.Count); |
|
|
|
|
|
|
|
|
|
// Sub-path of an existing leaf. |
|
|
|
|
tree.IntersectFieldPath("foo.bar", result); |
|
|
|
|
paths = result.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(1, paths.Count); |
|
|
|
|
Assert.Contains("foo.bar", paths); |
|
|
|
|
|
|
|
|
|
// Match an existing leaf node. |
|
|
|
|
tree.IntersectFieldPath("foo", result); |
|
|
|
|
paths = result.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(1, paths.Count); |
|
|
|
|
Assert.Contains("foo", paths); |
|
|
|
|
|
|
|
|
|
// Non-exist path. |
|
|
|
|
tree.IntersectFieldPath("bar.foo", result); |
|
|
|
|
paths = result.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(1, paths.Count); |
|
|
|
|
Assert.Contains("foo", paths); |
|
|
|
|
|
|
|
|
|
// Match a non-leaf node. |
|
|
|
|
tree.IntersectFieldPath("bar", result); |
|
|
|
|
paths = result.ToFieldMask().Paths; |
|
|
|
|
Assert.AreEqual(3, paths.Count); |
|
|
|
|
Assert.Contains("foo", paths); |
|
|
|
|
Assert.Contains("bar.baz", paths); |
|
|
|
|
Assert.Contains("bar.quz", paths); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
private void Merge(FieldMaskTree tree, IMessage source, IMessage destination, FieldMask.MergeOptions options, bool useDynamicMessage) |
|
|
|
|
{ |
|
|
|
|
if (useDynamicMessage) |
|
|
|
|
{ |
|
|
|
|
var newSource = source.Descriptor.Parser.CreateTemplate(); |
|
|
|
|
newSource.MergeFrom(source.ToByteString()); |
|
|
|
|
|
|
|
|
|
var newDestination = source.Descriptor.Parser.CreateTemplate(); |
|
|
|
|
newDestination.MergeFrom(destination.ToByteString()); |
|
|
|
|
|
|
|
|
|
tree.Merge(newSource, newDestination, options); |
|
|
|
|
|
|
|
|
|
// Clear before merging: |
|
|
|
|
foreach (var fieldDescriptor in destination.Descriptor.Fields.InFieldNumberOrder()) |
|
|
|
|
{ |
|
|
|
|
fieldDescriptor.Accessor.Clear(destination); |
|
|
|
|
} |
|
|
|
|
destination.MergeFrom(newDestination.ToByteString()); |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
tree.Merge(source, destination, options); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
[Test] |
|
|
|
|
[TestCase(true)] |
|
|
|
|
[TestCase(false)] |
|
|
|
|
public void Merge(bool useDynamicMessage) |
|
|
|
|
{ |
|
|
|
|
TestAllTypes value = new TestAllTypes |
|
|
|
|
{ |
|
|
|
|
SingleInt32 = 1234, |
|
|
|
|
SingleNestedMessage = new TestAllTypes.Types.NestedMessage {Bb = 5678}, |
|
|
|
|
RepeatedInt32 = {4321}, |
|
|
|
|
RepeatedNestedMessage = {new TestAllTypes.Types.NestedMessage {Bb = 8765}} |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
NestedTestAllTypes source = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Payload = value, |
|
|
|
|
Child = new NestedTestAllTypes {Payload = value} |
|
|
|
|
}; |
|
|
|
|
// Now we have a message source with the following structure: |
|
|
|
|
// [root] -+- payload -+- single_int32 |
|
|
|
|
// | +- single_nested_message |
|
|
|
|
// | +- repeated_int32 |
|
|
|
|
// | +- repeated_nested_message |
|
|
|
|
// | |
|
|
|
|
// +- child --- payload -+- single_int32 |
|
|
|
|
// +- single_nested_message |
|
|
|
|
// +- repeated_int32 |
|
|
|
|
// +- repeated_nested_message |
|
|
|
|
|
|
|
|
|
FieldMask.MergeOptions options = new FieldMask.MergeOptions(); |
|
|
|
|
|
|
|
|
|
// Test merging each individual field. |
|
|
|
|
NestedTestAllTypes destination = new NestedTestAllTypes(); |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("payload.single_int32"), |
|
|
|
|
source, destination, options, useDynamicMessage); |
|
|
|
|
NestedTestAllTypes expected = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Payload = new TestAllTypes |
|
|
|
|
{ |
|
|
|
|
SingleInt32 = 1234 |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
Assert.AreEqual(expected, destination); |
|
|
|
|
|
|
|
|
|
destination = new NestedTestAllTypes(); |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("payload.single_nested_message"), |
|
|
|
|
source, destination, options, useDynamicMessage); |
|
|
|
|
expected = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Payload = new TestAllTypes |
|
|
|
|
{ |
|
|
|
|
SingleNestedMessage = new TestAllTypes.Types.NestedMessage {Bb = 5678} |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
Assert.AreEqual(expected, destination); |
|
|
|
|
|
|
|
|
|
destination = new NestedTestAllTypes(); |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("payload.repeated_int32"), |
|
|
|
|
source, destination, options, useDynamicMessage); |
|
|
|
|
expected = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Payload = new TestAllTypes |
|
|
|
|
{ |
|
|
|
|
RepeatedInt32 = {4321} |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
Assert.AreEqual(expected, destination); |
|
|
|
|
|
|
|
|
|
destination = new NestedTestAllTypes(); |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("payload.repeated_nested_message"), |
|
|
|
|
source, destination, options, useDynamicMessage); |
|
|
|
|
expected = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Payload = new TestAllTypes |
|
|
|
|
{ |
|
|
|
|
RepeatedNestedMessage = {new TestAllTypes.Types.NestedMessage {Bb = 8765}} |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
Assert.AreEqual(expected, destination); |
|
|
|
|
|
|
|
|
|
destination = new NestedTestAllTypes(); |
|
|
|
|
Merge( |
|
|
|
|
new FieldMaskTree().AddFieldPath("child.payload.single_int32"), |
|
|
|
|
source, |
|
|
|
|
destination, |
|
|
|
|
options, |
|
|
|
|
useDynamicMessage); |
|
|
|
|
expected = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Child = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Payload = new TestAllTypes |
|
|
|
|
{ |
|
|
|
|
SingleInt32 = 1234 |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
Assert.AreEqual(expected, destination); |
|
|
|
|
|
|
|
|
|
destination = new NestedTestAllTypes(); |
|
|
|
|
Merge( |
|
|
|
|
new FieldMaskTree().AddFieldPath("child.payload.single_nested_message"), |
|
|
|
|
source, |
|
|
|
|
destination, |
|
|
|
|
options, |
|
|
|
|
useDynamicMessage); |
|
|
|
|
expected = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Child = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Payload = new TestAllTypes |
|
|
|
|
{ |
|
|
|
|
SingleNestedMessage = new TestAllTypes.Types.NestedMessage {Bb = 5678} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
Assert.AreEqual(expected, destination); |
|
|
|
|
|
|
|
|
|
destination = new NestedTestAllTypes(); |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("child.payload.repeated_int32"), |
|
|
|
|
source, destination, options, useDynamicMessage); |
|
|
|
|
expected = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Child = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Payload = new TestAllTypes |
|
|
|
|
{ |
|
|
|
|
RepeatedInt32 = {4321} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
Assert.AreEqual(expected, destination); |
|
|
|
|
|
|
|
|
|
destination = new NestedTestAllTypes(); |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("child.payload.repeated_nested_message"), |
|
|
|
|
source, destination, options, useDynamicMessage); |
|
|
|
|
expected = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Child = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Payload = new TestAllTypes |
|
|
|
|
{ |
|
|
|
|
RepeatedNestedMessage = {new TestAllTypes.Types.NestedMessage {Bb = 8765}} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
Assert.AreEqual(expected, destination); |
|
|
|
|
|
|
|
|
|
destination = new NestedTestAllTypes(); |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("child").AddFieldPath("payload"), |
|
|
|
|
source, destination, options, useDynamicMessage); |
|
|
|
|
Assert.AreEqual(source, destination); |
|
|
|
|
|
|
|
|
|
// Test repeated options. |
|
|
|
|
destination = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Payload = new TestAllTypes |
|
|
|
|
{ |
|
|
|
|
RepeatedInt32 = { 1000 } |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("payload.repeated_int32"), |
|
|
|
|
source, destination, options, useDynamicMessage); |
|
|
|
|
// Default behavior is to append repeated fields. |
|
|
|
|
Assert.AreEqual(2, destination.Payload.RepeatedInt32.Count); |
|
|
|
|
Assert.AreEqual(1000, destination.Payload.RepeatedInt32[0]); |
|
|
|
|
Assert.AreEqual(4321, destination.Payload.RepeatedInt32[1]); |
|
|
|
|
// Change to replace repeated fields. |
|
|
|
|
options.ReplaceRepeatedFields = true; |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("payload.repeated_int32"), |
|
|
|
|
source, destination, options, useDynamicMessage); |
|
|
|
|
Assert.AreEqual(1, destination.Payload.RepeatedInt32.Count); |
|
|
|
|
Assert.AreEqual(4321, destination.Payload.RepeatedInt32[0]); |
|
|
|
|
|
|
|
|
|
// Test message options. |
|
|
|
|
destination = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Payload = new TestAllTypes |
|
|
|
|
{ |
|
|
|
|
SingleInt32 = 1000, |
|
|
|
|
SingleUint32 = 2000 |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("payload"), |
|
|
|
|
source, destination, options, useDynamicMessage); |
|
|
|
|
// Default behavior is to merge message fields. |
|
|
|
|
Assert.AreEqual(1234, destination.Payload.SingleInt32); |
|
|
|
|
Assert.AreEqual(2000, destination.Payload.SingleUint32); |
|
|
|
|
|
|
|
|
|
// Test merging unset message fields. |
|
|
|
|
NestedTestAllTypes clearedSource = source.Clone(); |
|
|
|
|
clearedSource.Payload = null; |
|
|
|
|
destination = new NestedTestAllTypes(); |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("payload"), |
|
|
|
|
clearedSource, destination, options, useDynamicMessage); |
|
|
|
|
Assert.IsNull(destination.Payload); |
|
|
|
|
|
|
|
|
|
// Skip a message field if they are unset in both source and target. |
|
|
|
|
destination = new NestedTestAllTypes(); |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("payload.single_int32"), |
|
|
|
|
clearedSource, destination, options, useDynamicMessage); |
|
|
|
|
Assert.IsNull(destination.Payload); |
|
|
|
|
|
|
|
|
|
// Change to replace message fields. |
|
|
|
|
options.ReplaceMessageFields = true; |
|
|
|
|
destination = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Payload = new TestAllTypes |
|
|
|
|
{ |
|
|
|
|
SingleInt32 = 1000, |
|
|
|
|
SingleUint32 = 2000 |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("payload"), |
|
|
|
|
source, destination, options, useDynamicMessage); |
|
|
|
|
Assert.AreEqual(1234, destination.Payload.SingleInt32); |
|
|
|
|
Assert.AreEqual(0, destination.Payload.SingleUint32); |
|
|
|
|
|
|
|
|
|
// Test merging unset message fields. |
|
|
|
|
destination = new NestedTestAllTypes |
|
|
|
|
{ |
|
|
|
|
Payload = new TestAllTypes |
|
|
|
|
{ |
|
|
|
|
SingleInt32 = 1000, |
|
|
|
|
SingleUint32 = 2000 |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("payload"), |
|
|
|
|
clearedSource, destination, options, useDynamicMessage); |
|
|
|
|
Assert.IsNull(destination.Payload); |
|
|
|
|
|
|
|
|
|
// Test merging unset primitive fields. |
|
|
|
|
destination = source.Clone(); |
|
|
|
|
destination.Payload.SingleInt32 = 0; |
|
|
|
|
NestedTestAllTypes sourceWithPayloadInt32Unset = destination; |
|
|
|
|
destination = source.Clone(); |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("payload.single_int32"), |
|
|
|
|
sourceWithPayloadInt32Unset, destination, options, useDynamicMessage); |
|
|
|
|
Assert.AreEqual(0, destination.Payload.SingleInt32); |
|
|
|
|
|
|
|
|
|
// Change to clear unset primitive fields. |
|
|
|
|
options.ReplacePrimitiveFields = true; |
|
|
|
|
destination = source.Clone(); |
|
|
|
|
Merge(new FieldMaskTree().AddFieldPath("payload.single_int32"), |
|
|
|
|
sourceWithPayloadInt32Unset, destination, options, useDynamicMessage); |
|
|
|
|
Assert.IsNotNull(destination.Payload); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
} |