Include unique option

DataDog · Jan 9, 2025 · bd2a3d5 · bd2a3d5
1 parent 14f31da
commit bd2a3d5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/marshal_test.go b/marshal_test.go
@@ -20,6 +20,7 @@ func TestMarshal(t *testing.T) {
 		given       any
 		expect      string
 		expectError error
+		opts        []MarshalOption
 	}{
 		{
 			description: "nil",
@@ -86,6 +87,7 @@ func TestMarshal(t *testing.T) {
 			given:       articlesAA,
 			expect:      "",
 			expectError: ErrNonuniqueResource,
+			opts:        []MarshalOption{MarshallCheckUniqueness()},
 		}, {
 			description: "[]*Article",
 			given:       articlesABPtr,
@@ -592,7 +594,7 @@ func TestMarshalRelationships(t *testing.T) {
 		}, {
 			description:    "with related nonunique comments",
 			given:          &articleRelatedNonuniqueComments,
-			marshalOptions: nil,
+			marshalOptions: []MarshalOption{MarshallCheckUniqueness()},
 			expect:         "",
 			expectError:    ErrNonuniqueResource,
 		}, {

diff --git a/unmarshal_test.go b/unmarshal_test.go
@@ -74,7 +74,7 @@ func TestUnmarshal(t *testing.T) {
 			given:       articlesABNonuniqueData,
 			do: func(body []byte) (any, error) {
 				var a []Article
-				err := Unmarshal(body, &a)
+				err := Unmarshal(body, &a, UnmarshalCheckUniqueness())
 				return a, err
 			},
 			expect:      ([]Article)(nil),
@@ -430,7 +430,7 @@ func TestUnmarshal(t *testing.T) {
 			given:       articleRelatedNonuniqueLinkage,
 			do: func(body []byte) (any, error) {
 				var a ArticleRelated
-				err := Unmarshal(body, &a)
+				err := Unmarshal(body, &a, UnmarshalCheckUniqueness())
 				return a, err
 			},
 			expect:      ArticleRelated{},