GODRIVER-2443 Fix type and add want sanity check

prestonvasquez · Apr 30, 2024 · 926b83c · 926b83c
1 parent 0d80ea4
commit 926b83c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/internal/integration/collection_test.go b/internal/integration/collection_test.go
@@ -896,7 +896,7 @@ func TestCollection(t *testing.T) {
 				err := res.Decode(&got)
 				assert.NoError(t, err)
 
-				assert.EqualValues(mt, tc.want, got, "expected result %v, got %v", tc.want, res)
+				assert.EqualValues(mt, tc.want, got, "expected result %v, got %v", tc.want, got)
 			})
 		}
 	})

diff --git a/internal/integration/crud_helpers_test.go b/internal/integration/crud_helpers_test.go
@@ -1546,7 +1546,12 @@ func verifyDistinctResult(
 		return
 	}
 
-	for i, iwant := range want.(bson.A) {
+	assert.NotNil(mt, want, "expected want to be non-nil")
+
+	arr, ok := want.(bson.A)
+	assert.True(mt, ok, "expected want to be a BSON array")
+
+	for i, iwant := range arr {
 		gotRawValue := got.Index(uint(i))
 
 		iwantType, iwantBytes, err := bson.MarshalValue(iwant)