From d9560c78b8e6e2799dcea64107917190cb7077ac Mon Sep 17 00:00:00 2001 From: Jack Christensen Date: Sat, 3 Jun 2023 07:59:28 -0500 Subject: [PATCH] Use tx instead of underlying conn in test Improves clarity --- copy_from_test.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/copy_from_test.go b/copy_from_test.go index e5c4cd82..92907d74 100644 --- a/copy_from_test.go +++ b/copy_from_test.go @@ -375,11 +375,11 @@ func TestConnCopyFromEnum(t *testing.T) { {nil, nil, nil, nil, nil, nil}, } - copyCount, err := conn.CopyFrom(ctx, pgx.Identifier{"foo"}, []string{"a", "b", "c", "d", "e", "f"}, pgx.CopyFromRows(inputRows)) + copyCount, err := tx.CopyFrom(ctx, pgx.Identifier{"foo"}, []string{"a", "b", "c", "d", "e", "f"}, pgx.CopyFromRows(inputRows)) require.NoError(t, err) require.EqualValues(t, len(inputRows), copyCount) - rows, err := conn.Query(ctx, "select * from foo") + rows, err := tx.Query(ctx, "select * from foo") require.NoError(t, err) var outputRows [][]any @@ -395,6 +395,9 @@ func TestConnCopyFromEnum(t *testing.T) { t.Errorf("Input rows and output rows do not equal: %v -> %v", inputRows, outputRows) } + err = tx.Rollback(ctx) + require.NoError(t, err) + ensureConnValid(t, conn) }