fieldpath.ManagedFields: Add Equals method

Author: Antoine Pelisse

fieldpath/managers.go

@@ -58,6 +58,28 @@ func (v versionedSet) Applied() bool {
 // what version).
 type ManagedFields map[string]VersionedSet
 
+// Equals returns true if the two managedfields are the same, false
+// otherwise.
+func (lhs ManagedFields) Equals(rhs ManagedFields) bool {
+	if len(lhs) != len(rhs) {
+		return false
+	}
+
+	for manager, left := range lhs {
+		right, ok := rhs[manager]
+		if !ok {
+			return false
+		}
+		if left.APIVersion() != right.APIVersion() || left.Applied() != right.Applied() {
+			return false
+		}
+		if !left.Set().Equals(right.Set()) {
+			return false
+		}
+	}
+	return true
+}
+
 // Difference returns a symmetric difference between two Managers. If a
 // given user's entry has version X in lhs and version Y in rhs, then
 // the return value for that user will be from rhs. If the difference for

fieldpath/managers_test.go

@@ -30,7 +30,7 @@ var (
 	_P  = fieldpath.MakePathOrDie
 )
 
-func TestManagersDifference(t *testing.T) {
+func TestManagersEquals(t *testing.T) {
 	tests := []struct {
 		name string
 		lhs  fieldpath.ManagedFields
@@ -164,3 +164,123 @@ func TestManagersDifference(t *testing.T) {
 		})
 	}
 }
+
+func TestManagersDifference(t *testing.T) {
+	tests := []struct {
+		name  string
+		lhs   fieldpath.ManagedFields
+		rhs   fieldpath.ManagedFields
+		equal bool
+	}{
+		{
+			name:  "Empty sets",
+			equal: true,
+		},
+		{
+			name: "Same everything",
+			lhs: fieldpath.ManagedFields{
+				"one": fieldpath.NewVersionedSet(
+					_NS(_P("numeric"), _P("string"), _P("bool")),
+					"v1",
+					false,
+				),
+			},
+			rhs: fieldpath.ManagedFields{
+				"one": fieldpath.NewVersionedSet(
+					_NS(_P("numeric"), _P("string"), _P("bool")),
+					"v1",
+					false,
+				),
+			},
+			equal: true,
+		},
+		{
+			name: "Empty RHS",
+			lhs: fieldpath.ManagedFields{
+				"default": fieldpath.NewVersionedSet(
+					_NS(_P("numeric"), _P("string"), _P("bool")),
+					"v1",
+					false,
+				),
+			},
+			equal: false,
+		},
+		{
+			name: "Empty LHS",
+			rhs: fieldpath.ManagedFields{
+				"default": fieldpath.NewVersionedSet(
+					_NS(_P("numeric"), _P("string"), _P("bool")),
+					"v1",
+					false,
+				),
+			},
+			equal: false,
+		},
+		{
+			name: "Different managers",
+			lhs: fieldpath.ManagedFields{
+				"one": fieldpath.NewVersionedSet(
+					_NS(_P("numeric"), _P("string"), _P("bool")),
+					"v1",
+					false,
+				),
+			},
+			rhs: fieldpath.ManagedFields{
+				"two": fieldpath.NewVersionedSet(
+					_NS(_P("numeric"), _P("string"), _P("bool")),
+					"v1",
+					false,
+				),
+			},
+			equal: false,
+		},
+		{
+			name: "Same manager, different version",
+			lhs: fieldpath.ManagedFields{
+				"one": fieldpath.NewVersionedSet(
+					_NS(_P("numeric"), _P("string"), _P("integer")),
+					"v1",
+					false,
+				),
+			},
+			rhs: fieldpath.ManagedFields{
+				"one": fieldpath.NewVersionedSet(
+					_NS(_P("numeric"), _P("string"), _P("bool")),
+					"v2",
+					false,
+				),
+			},
+			equal: false,
+		},
+		{
+			name: "Set difference",
+			lhs: fieldpath.ManagedFields{
+				"one": fieldpath.NewVersionedSet(
+					_NS(_P("numeric"), _P("string")),
+					"v1",
+					false,
+				),
+			},
+			rhs: fieldpath.ManagedFields{
+				"one": fieldpath.NewVersionedSet(
+					_NS(_P("string"), _P("bool")),
+					"v1",
+					false,
+				),
+			},
+			equal: false,
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(fmt.Sprintf(test.name), func(t *testing.T) {
+			equal := test.lhs.Equals(test.rhs)
+			if test.equal && !equal {
+				difference := test.lhs.Difference(test.rhs)
+				t.Errorf("should be equal, but are different: %v", difference)
+			} else if !test.equal && equal {
+				t.Errorf("should not be equal, but they are")
+			}
+		})
+	}
+}