mirror of
https://github.com/hanzoai/telemetry.git
synced 2026-10-30 20:01:49 +00:00
Damien MathieuandGitHub
d113af6e49
We are going to want to be able to initialize collectors with specific configurations (the `--from-config` option will allow building a new manifest out of an existing collector config). So this changes the manifest initialization, out of a template, and to using the actual `Config` struct. I had to setup manual omission of nil/empty values, as mapstructure's omitempty seems to only ignore structs, not maps or strings.
88 lines
2.2 KiB
Go
88 lines
2.2 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package internal // import "go.opentelemetry.io/collector/cmd/builder/internal"
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/knadh/koanf/parsers/yaml"
|
|
"github.com/knadh/koanf/providers/file"
|
|
"github.com/knadh/koanf/v2"
|
|
"github.com/spf13/cobra"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.opentelemetry.io/collector/cmd/builder/internal/builder"
|
|
)
|
|
|
|
func TestInitCommand(t *testing.T) {
|
|
cmd := initCommand()
|
|
|
|
assert.NotNil(t, cmd)
|
|
assert.IsType(t, &cobra.Command{}, cmd)
|
|
assert.Equal(t, "init", cmd.Use)
|
|
}
|
|
|
|
func TestRunInit(t *testing.T) {
|
|
for _, tt := range []struct {
|
|
name string
|
|
buildPath func(string) string
|
|
|
|
wantErr string
|
|
}{
|
|
{
|
|
name: "without a path",
|
|
buildPath: func(string) string { return "" },
|
|
wantErr: "argument must be a folder",
|
|
},
|
|
{
|
|
name: "with a relative path",
|
|
buildPath: func(string) string { return "./tmp/init" },
|
|
wantErr: "",
|
|
},
|
|
{
|
|
name: "with an absolute path",
|
|
buildPath: func(dir string) string { return dir },
|
|
wantErr: "",
|
|
},
|
|
} {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
tmpdir := filepath.Join(t.TempDir(), "init")
|
|
path := tt.buildPath(tmpdir)
|
|
err := run(path)
|
|
|
|
if tt.wantErr == "" {
|
|
require.NoError(t, err)
|
|
validateCollector(t, path)
|
|
} else {
|
|
require.ErrorContains(t, err, tt.wantErr)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func validateCollector(t *testing.T, path string) {
|
|
require.FileExists(t, filepath.Join(path, ".gitignore"))
|
|
require.FileExists(t, filepath.Join(path, "README.md"))
|
|
require.FileExists(t, filepath.Join(path, "manifest.yaml"))
|
|
require.FileExists(t, filepath.Join(path, "go.mod"))
|
|
require.FileExists(t, filepath.Join(path, "go.sum"))
|
|
require.FileExists(t, filepath.Join(path, "Makefile"))
|
|
require.FileExists(t, filepath.Join(path, "config.yaml"))
|
|
|
|
k := koanf.New(".")
|
|
err := k.Load(file.Provider(filepath.Join(path, "manifest.yaml")), yaml.Parser())
|
|
require.NoError(t, err)
|
|
|
|
cfg := builder.Config{}
|
|
err = k.UnmarshalWithConf("", &cfg, koanf.UnmarshalConf{
|
|
Tag: "mapstructure",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, "init", cfg.Distribution.Name)
|
|
assert.Equal(t, defaultDescription, cfg.Distribution.Description)
|
|
}
|