mirror of https://github.com/databricks/cli.git
37 lines
689 B
Go
37 lines
689 B
Go
package root
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/databricks/databricks-sdk-go/useragent"
|
|
"github.com/spf13/cobra"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestWithCommandInUserAgent(t *testing.T) {
|
|
root := &cobra.Command{
|
|
Use: "root",
|
|
}
|
|
|
|
hello := &cobra.Command{
|
|
Use: "hello",
|
|
}
|
|
|
|
world := &cobra.Command{
|
|
Use: "world",
|
|
}
|
|
|
|
root.AddCommand(hello)
|
|
hello.AddCommand(world)
|
|
|
|
assert.Equal(t, "root", commandString(root))
|
|
assert.Equal(t, "hello", commandString(hello))
|
|
assert.Equal(t, "hello_world", commandString(world))
|
|
|
|
ctx := withCommandInUserAgent(context.Background(), world)
|
|
|
|
ua := useragent.FromContext(ctx)
|
|
assert.Contains(t, ua, "cmd/hello_world")
|
|
}
|