query_duration_test.go

package main

import (
	"testing"
)

func Test_queryDurationFuncs(t *testing.T) {
	ctx, conn, cleanup := setup(t)
	defer cleanup()

	for name, fn := range queryDurationFuncs {
		t.Run(name+" with planning", func(t *testing.T) {
			d, err := fn(ctx, conn, "SELECT 1", true)()
			if err != nil {
				t.Fatal(err)
			} else if d <= 0 {
				t.Fatalf("bad duration: %s", d)
			}
		})

		t.Run(name+" without planning", func(t *testing.T) {
			d, err := fn(ctx, conn, "SELECT 1", false)()
			if err != nil {
				t.Fatal(err)
			} else if d <= 0 {
				t.Fatalf("bad duration: %s", d)
			}
		})
	}
}