Function NewInstanceWithOptions

runtime/testruntime/testruntime.go:115–222

NewInstanceWithOptions creates a runtime and an instance for use in tests. The instance's repo is a temp directory that will be cleared when the tests finish.

(t TestingT, opts InstanceOptions)

// NewInstanceWithOptions creates a runtime and an instance for use in tests.
// The instance's repo is a temp directory that will be cleared when the tests finish.
func NewInstanceWithOptions(t TestingT, opts InstanceOptions) (*runtime.Runtime, string) {
	rt := New(t, !opts.DisableHostAccess)
	ctx := t.Context()

	olapDriver := os.Getenv("RILL_RUNTIME_TEST_OLAP_DRIVER")
	if olapDriver == "" {
		olapDriver = "duckdb"
	}
	olapDSN := os.Getenv("RILL_RUNTIME_TEST_OLAP_DSN")
	if olapDSN == "" {
		olapDSN = ":memory:"
	}

	vars := make(map[string]string)
	maps.Copy(vars, opts.Variables)
	if vars["rill.stage_changes"] == "" {
		vars["rill.stage_changes"] = strconv.FormatBool(opts.StageChanges)
	}
	if vars["rill.watch_repo"] == "" {
		vars["rill.watch_repo"] = strconv.FormatBool(opts.WatchRepo)
	}

	// Making LLM completions in tests is disabled by default.
	// If enabled, we skip the test in CI (short mode) to prevent running up costs.
	var aiConnector string
	if opts.AIConnector != "" {
		// Mark AI tests as expensive
		testmode.Expensive(t)

		// Add it to the test connectors if not already present.
		if !slices.Contains(opts.TestConnectors, opts.AIConnector) {
			opts.TestConnectors = append(opts.TestConnectors, opts.AIConnector)
		}

		// Set the AI test connector as the instance's default AI connector.
		// This enables LLM completions.
		aiConnector = opts.AIConnector
	}

	for _, conn := range opts.TestConnectors {
		acquire, ok := Connectors[conn]
		require.True(t, ok, "unknown test connector %q", conn)
		cfg := acquire(t)
		for k, v := range cfg {
			k = fmt.Sprintf("connector.%s.%s", conn, k)
			vars[k] = v
		}
	}

	tmpDir := t.TempDir()
	inst := &drivers.Instance{
		Environment:      "test",
		OLAPConnector:    olapDriver,
		RepoConnector:    "repo",
		AIConnector:      aiConnector,
		CatalogConnector: "catalog",
		AdminConnector:   "noop_admin",
		Connectors: []*runtimev1.Connector{