gopkg/pgx
2
0
Fork 0
Code Activity

pgx uses pgconn.CommandTag instead of own definition

Browse Source
This commit is contained in:
Jack Christensen
2019-01-01 16:55:48 -06:00
parent fdbf2ba728
commit 7986e2726d
1 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pgconn_test.go
+23
View File
@@ -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)
}
}