add Kind() method to convert *Rows to pgx.Rows interface

pashagolub · Nov 1, 2023 · b501ed2 · b501ed2
1 parent 5e0d622
commit b501ed2
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/rows.go b/rows.go
@@ -275,6 +275,14 @@ func (r *Rows) FromCSVString(s string) *Rows {
 	return r
 }
 
+// Kind returns rows corresponding to the interface pgx.Rows
+// useful for testing entities that implement an interface pgx.RowScanner
+func (r *Rows) Kind() pgx.Rows {
+	return &rowSets{
+		sets: []*Rows{r},
+	}
+}
+
 // NewRowsWithColumnDefinition return rows with columns metadata
 func NewRowsWithColumnDefinition(columns ...pgconn.FieldDescription) *Rows {
 	return &Rows{