Add id.Slice.Contains with tests (#2252)

fabric8-services · Aug 21, 2018 · b0ee723 · b0ee723
1 parent 69ce1ac
commit b0ee723
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/id/slice.go b/id/slice.go
@@ -105,6 +105,17 @@ func (s Slice) Swap(i, j int) {
 	s[i], s[j] = s[j], s[i]
 }
 
+// Contains returns true if the slice contains the given ID; otherwise false is
+// returned.
+func (s Slice) Contains(v uuid.UUID) bool {
+	for _, i := range s {
+		if i == v {
+			return true
+		}
+	}
+	return false
+}
+
 // Ensure Slice implements the sort.Interface
 var _ sort.Interface = Slice{}
 var _ sort.Interface = (*Slice)(nil)
diff --git a/id/slice_blackbox_test.go b/id/slice_blackbox_test.go
@@ -178,3 +178,11 @@ func TestToStringSlice(t *testing.T) {
 	s := id.Slice{a, b, c}
 	require.Equal(t, []string{a.String(), b.String(), c.String()}, s.ToStringSlice())
 }
+
+func TestSlice_Contains(t *testing.T) {
+	a := uuid.FromStringOrNil("cb4364bf-893c-4ac2-a35f-4ef74c212e88")
+	b := uuid.FromStringOrNil("d085a858-6b98-44ce-8777-b36d927b07dc")
+	require.True(t, id.Slice{a, b}.Contains(a))
+	require.False(t, id.Slice{}.Contains(a))
+	require.False(t, id.Slice{a}.Contains(b))
+}