mirror of https://github.com/databricks/cli.git
35 lines
1012 B
Go
35 lines
1012 B
Go
|
package bundletest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/databricks/cli/bundle"
|
||
|
|
"github.com/databricks/cli/libs/dyn"
|
||
|
|
)
|
||
|
|
|
||
|
|
// SetLocation sets the location of all values in the bundle to the given path.
|
||
|
|
// This is useful for testing where we need to associate configuration
|
||
|
|
// with the path it is loaded from.
|
||
|
|
func SetLocation(b *bundle.Bundle, prefix string, filePath string) {
|
||
|
|
start := dyn.MustPathFromString(prefix)
|
||
|
|
b.Config.Mutate(func(root dyn.Value) (dyn.Value, error) {
|
||
|
|
return dyn.Walk(root, func(p dyn.Path, v dyn.Value) (dyn.Value, error) {
|
||
|
|
// If the path has the given prefix, set the location.
|
||
|
|
if p.HasPrefix(start) {
|
||
|
|
return v.WithLocation(dyn.Location{
|
||
|
|
File: filePath,
|
||
|
|
}), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// The path is not nested under the given prefix.
|
||
|
|
// If the path is a prefix of the prefix, keep traversing and return the node verbatim.
|
||
|
|
if start.HasPrefix(p) {
|
||
|
|
return v, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Return verbatim, but skip traversal.
|
||
|
|
return v, dyn.ErrSkip
|
||
|
|
})
|
||
|
|
})
|
||
|
|
|
||
|
|
b.Config.ConfigureConfigFilePath()
|
||
|
|
}
|