Function TestClassifyQuery

server/querylog_test.go:11–51

(t *testing.T)

	"unicode/utf8"
	"unsafe"

	"github.com/posthog/duckgres/server/observe"
	"github.com/posthog/duckgres/server/usersecrets"
)

func TestClassifyQuery(t *testing.T) {
	tests := []struct {
		cmdType string
		want    string
	}{
		{"SELECT", "Select"},
		{"SHOW", "Select"},
		{"TABLE", "Select"},
		{"VALUES", "Select"},
		{"EXPLAIN", "Select"},
		{"INSERT", "Insert"},
		{"UPDATE", "Update"},
		{"DELETE", "Delete"},
		{"CREATE", "DDL"},
		{"ALTER", "DDL"},
		{"DROP", "DDL"},
		{"TRUNCATE", "DDL"},
		{"COPY", "Copy"},
		{"BEGIN", "Utility"},
		{"COMMIT", "Utility"},
		{"ROLLBACK", "Utility"},
		{"SET", "Utility"},
		{"RESET", "Utility"},
		{"DISCARD", "Utility"},
		{"DEALLOCATE", "Utility"},
		{"LISTEN", "Utility"},
		{"NOTIFY", "Utility"},
		{"UNLISTEN", "Utility"},
		{"UNKNOWN", "Utility"},
		{"", "Utility"},
	}

	for _, tt := range tests {
		t.Run(tt.cmdType, func(t *testing.T) {
			got := classifyQuery(tt.cmdType)
			if got != tt.want {
				t.Errorf("classifyQuery(%q) = %q, want %q", tt.cmdType, got, tt.want)
			}
		})