45 lines
1.7 KiB
Go
45 lines
1.7 KiB
Go
|
package engine
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"testing/fstest"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMainRelativeAcquisitionClosesHandlesOnFailure(t *testing.T) {
|
||
|
|
for _, fail := range []string{"parse", "read"} {
|
||
|
|
t.Run(fail, func(t *testing.T) {
|
||
|
|
body := "items: [invalid\n"
|
||
|
|
if fail == "read" {
|
||
|
|
body = "items: {key: value}\n"
|
||
|
|
}
|
||
|
|
filesystem := &observedFilesystem{files: fstest.MapFS{"dir/main.yml": {Data: []byte("{}\n")}, "dir/child.yml": {Data: []byte(body)}}}
|
||
|
|
if fail == "read" {
|
||
|
|
filesystem.failRead = "dir/child.yml"
|
||
|
|
}
|
||
|
|
namespace := newNativeNamespace("bundle", filesystem)
|
||
|
|
main := Input{Source: filesystemFile(namespace, "dir/main.yml")}
|
||
|
|
underlay := Input{Source: Bytes("declarations", []byte("config: {imports: [{path: child.yml}]}\n")), Imports: true, MainRelative: &MainRelativeContext{}}
|
||
|
|
_, err := LoadLayered[importTestConfig](main, SecretBinding{}, LayeredOptions[importTestConfig]{Underlays: []Underlay{{Inputs: []Input{underlay}}}})
|
||
|
|
if err == nil {
|
||
|
|
t.Fatal("invalid selected child succeeded")
|
||
|
|
}
|
||
|
|
filesystem.assertClosed(t)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMainRelativeSharesAcquisitionBudget(t *testing.T) {
|
||
|
|
limits := defaultImportLimits()
|
||
|
|
limits.sources = 1
|
||
|
|
accounting := &importAccounting{}
|
||
|
|
state := &acquisitionState{accounting: accounting}
|
||
|
|
group := loadGroup{inputs: []loadInput{
|
||
|
|
{source: Bytes("one", []byte("items: {one: value}\n")), mainRelative: &MainRelativeContext{}},
|
||
|
|
{source: Bytes("two", []byte("items: {two: value}\n")), imports: true, mainRelative: &MainRelativeContext{}},
|
||
|
|
}}
|
||
|
|
_, diagnostics := acquireUnderlayGroupScopes(group, 1, nil, state, acquisitionPolicy{importLimits: limits}, nil)
|
||
|
|
if len(diagnostics) == 0 || diagnostics[0].Code != CodeImportLimitExceeded {
|
||
|
|
t.Fatalf("diagnostics = %#v", diagnostics)
|
||
|
|
}
|
||
|
|
}
|