Function TestLoadExamples

pkg/teamloader/teamloader_test.go:87–128

(t *testing.T)

}

func TestLoadExamples(t *testing.T) {
	examples := collectExamples(t)

	// Set every env var referenced by the examples to a dummy value so model
	// and tool initialisation succeeds without real credentials.
	for env := range gatherExampleEnvVars(t, examples) {
		t.Setenv(env, "dummy")
	}

	for _, agentFilename := range examples {
		t.Run(agentFilename, func(t *testing.T) {
			t.Parallel()

			data, err := os.ReadFile(agentFilename)
			require.NoError(t, err)

			// Examples must not pin a version: they should always parse with
			// the latest config schema.
			var v struct {
				Version string `yaml:"version"`
			}
			require.NoError(t, yaml.Unmarshal(data, &v))
			require.Empty(t, v.Version, "example %s should not define a version", agentFilename)

			// Use a file source so the config's directory is known: examples
			// such as instruction_file.yaml reference sibling files that are
			// resolved relative to it at load time. A temp WorkingDir still
			// redirects toolsets that write to disk (memory, RAG, cache, ...)
			// away from the examples/ tree.
			agentSource := config.NewFileSource(agentFilename)
			runConfig := &config.RuntimeConfig{}
			runConfig.WorkingDir = t.TempDir()

			teams, err := Load(catalogContext(t), agentSource, runConfig, withTestProviderRegistry()...)
			if errors.Is(err, dmr.ErrNotInstalled) {
				t.Skipf("Skipping %s: Docker Model Runner not installed", agentFilename)
			}
			require.NoError(t, err)
			assert.NotEmpty(t, teams)
		})
	}
}

// gatherExampleEnvVars returns the union of env vars referenced by the given
// example files (both for models and toolsets). The set is collected up-front