Function TestCustomQueriesSyncedRowScan

coderd/database/gentest/modelqueries_test.go:22–63

TestCustomQueriesSynced makes sure the manual custom queries in modelqueries.go are synced with the autogenerated queries.sql.go. This should probably be autogenerated, but it's not atm and this is easy to throw in to elevate a better error message. If this breaks, and is hard to fix, you can t.Ski

(t *testing.T)

// If this breaks, and is hard to fix, you can t.Skip() it. It is not a critical
// test. Ping @Emyrk to fix it again.
func TestCustomQueriesSyncedRowScan(t *testing.T) {
	t.Parallel()

	funcsToTrack := map[string]string{
		"GetTemplatesWithFilter": "GetAuthorizedTemplates",
		"GetWorkspaces":          "GetAuthorizedWorkspaces",
		"GetUsers":               "GetAuthorizedUsers",
		"GetChats":               "GetAuthorizedChats",
	}

	// Scan custom
	var custom []string
	for _, fn := range funcsToTrack {
		custom = append(custom, fn)
	}

	customFns := parseFile(t, "../modelqueries.go", func(name string) bool {
		return slices.Contains(custom, name)
	})
	generatedFns := parseFile(t, "../queries.sql.go", func(name string) bool {
		_, ok := funcsToTrack[name]
		return ok
	})
	merged := customFns
	for k, v := range generatedFns {
		merged[k] = v
	}

	for a, b := range funcsToTrack {
		a, b := a, b
		if !compareFns(t, a, b, merged[a], merged[b]) {
			//nolint:revive
			defer func() {
				// Run this at the end so the suggested fix is the last thing printed.
				t.Errorf("The functions %q and %q need to have identical 'rows.Scan()' "+
					"and 'db.QueryContext()' arguments in their function bodies. "+
					"Make sure to copy the function body from the autogenerated %q body. "+
					"Specifically the parameters for 'rows.Scan()' and 'db.QueryContext()'.", a, b, a)
			}()
		}
	}
}

type parsedFunc struct {
	RowScanArgs []ast.Expr