v2: fix equal and add test for equal

hashicorp · Nov 4, 2023 · 4cfe145 · 4cfe145
1 parent ed09455
commit 4cfe145
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/collection.go b/collection.go
@@ -201,7 +201,7 @@ func equalSet[T any](a, b Collection[T]) bool {
 		}
 		return true
 	})
-	return missing
+	return !missing
 }
 
 func removeSet[T any](s, col Collection[T]) bool {

diff --git a/collection_test.go b/collection_test.go
@@ -221,3 +221,26 @@ func TestInsertSetFunc(t *testing.T) {
 		})
 	})
 }
+
+func TestEqualSet(t *testing.T) {
+	t.Run("equal ok", func(t *testing.T) {
+		a := From(ints(3))
+		b := From(ints(3))
+		must.True(t, a.EqualSet(b))
+	})
+	t.Run("none equal none", func(t *testing.T) {
+		a := New[int](0)
+		b := New[int](0)
+		must.True(t, a.EqualSet(b))
+	})
+	t.Run("size not equal", func(t *testing.T) {
+		a := From(ints(3))
+		b := From(ints(4))
+		must.False(t, a.EqualSet(b))
+	})
+	t.Run("items not equal", func(t *testing.T) {
+		a := From(ints(3))
+		b := From([]int{1, 2, 4})
+		must.False(t, a.EqualSet(b))
+	})
+}