From 7986e2726d1679e78d3cce4c3df19e3f7bd3a866 Mon Sep 17 00:00:00 2001
From: Jack Christensen <jack@jackchristensen.com>
Date: Tue, 1 Jan 2019 16:55:48 -0600
Subject: [PATCH] pgx uses pgconn.CommandTag instead of own definition

---
 pgconn_test.go | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/pgconn_test.go b/pgconn_test.go
index 8b578d42..8f976d87 100644
--- a/pgconn_test.go
+++ b/pgconn_test.go
@@ -512,3 +512,26 @@ func TestConnCancelQuery(t *testing.T) {
 		t.Errorf("expected pgconn.PgError got %v", err)
 	}
 }
+
+func TestCommandTag(t *testing.T) {
+	t.Parallel()
+
+	var tests = []struct {
+		commandTag   pgconn.CommandTag
+		rowsAffected int64
+	}{
+		{commandTag: pgconn.CommandTag("INSERT 0 5"), rowsAffected: 5},
+		{commandTag: pgconn.CommandTag("UPDATE 0"), rowsAffected: 0},
+		{commandTag: pgconn.CommandTag("UPDATE 1"), rowsAffected: 1},
+		{commandTag: pgconn.CommandTag("DELETE 0"), rowsAffected: 0},
+		{commandTag: pgconn.CommandTag("DELETE 1"), rowsAffected: 1},
+		{commandTag: pgconn.CommandTag("CREATE TABLE"), rowsAffected: 0},
+		{commandTag: pgconn.CommandTag("ALTER TABLE"), rowsAffected: 0},
+		{commandTag: pgconn.CommandTag("DROP TABLE"), rowsAffected: 0},
+	}
+
+	for i, tt := range tests {
+		actual := tt.commandTag.RowsAffected()
+		assert.Equalf(t, tt.rowsAffected, actual, "%d. %v", i, tt.commandTag)
+	}
+}