func runCommand(t *testing.T, args []string, dff config.DeployFilterFlags) string {
out := &bytes.Buffer{}
state := makeTestState()
shc := sous.SourceHostChooser{}
rf, err := dff.BuildFilter(shc.ParseSourceLocation)
require.NoError(t, err)
deps, err := state.Deployments()
require.NoError(t, err)
smg := SousMetadataGet{
DeployFilterFlags: dff,
ResolveFilter: rf,
State: state,
CurrentGDM: graph.CurrentGDM{Deployments: deps},
OutWriter: graph.OutWriter(out),
}
res := smg.Execute(args)
assert.Equal(t, 0, res.ExitCode())
return out.String()
}
func newResolveFilter(sf *config.DeployFilterFlags, shc sous.SourceHostChooser) (*sous.ResolveFilter, error) {
return sf.BuildFilter(shc.ParseSourceLocation)
}
func TestPredicateBuilder(t *testing.T) {
assert := assert.New(t)
ds := make([]*sous.Deployment, 0, 8)
cs := []string{"cluster1", "cluster2"}
rs := []string{"github.com/ot/one", "github.com/ot/two"}
os := []string{"up", "down"}
for _, c := range cs {
for _, r := range rs {
for _, o := range os {
ds = append(ds, &sous.Deployment{
ClusterName: c,
SourceID: sous.SourceID{
Location: sous.SourceLocation{
Repo: r,
Dir: o,
},
},
})
}
}
}
parseSL := func(string) (sous.SourceLocation, error) {
return sous.SourceLocation{}, nil
}
// for i, d := range ds {
// fmt.Printf("%d: %#v\n", i, d)
// }
//
f := config.DeployFilterFlags{}
rf, err := f.BuildFilter(parseSL)
assert.NoError(err)
assert.True(rf.All())
f.Repo = string(rs[0])
pd, err := f.BuildPredicate(parseSL)
assert.NoError(err)
assert.NotNil(pd)
filtered := filter(ds, pd)
assert.Contains(filtered, ds[0])
assert.Contains(filtered, ds[1])
assert.Contains(filtered, ds[4])
assert.Contains(filtered, ds[5])
assert.Len(filtered, 4)
f.Offset = string(os[0])
pd, err = f.BuildPredicate(parseSL)
assert.NoError(err)
assert.NotNil(pd)
filtered = filter(ds, pd)
assert.Contains(filtered, ds[0])
assert.Contains(filtered, ds[4])
assert.Len(filtered, 2)
f.Cluster = cs[0]
pd, err = f.BuildPredicate(parseSL)
assert.NoError(err)
assert.NotNil(pd)
filtered = filter(ds, pd)
assert.Contains(filtered, ds[0])
assert.Len(filtered, 1)
f = config.DeployFilterFlags{Cluster: cs[1]}
pd, err = f.BuildPredicate(parseSL)
assert.NoError(err)
assert.NotNil(pd)
filtered = filter(ds, pd)
assert.Contains(filtered, ds[4])
assert.Contains(filtered, ds[5])
assert.Contains(filtered, ds[6])
assert.Contains(filtered, ds[7])
assert.Len(filtered, 4)
f = config.DeployFilterFlags{All: true}
pd, err = f.BuildPredicate(parseSL)
assert.NoError(err)
assert.NotNil(pd)
filtered = filter(ds, pd)
assert.Len(filtered, 8)
}