func testContainerTxn(t *testing.T, conn db.Conn, spec string) {
compiled, err := stitch.FromJavascript(spec, stitch.DefaultImportGetter)
assert.Nil(t, err)
var containers []db.Container
conn.Transact(func(view db.Database) error {
updatePolicy(view, db.Master, compiled.String())
containers = view.SelectFromContainer(nil)
return nil
})
for _, e := range queryContainers(compiled) {
found := false
for i, c := range containers {
if e.Image == c.Image &&
reflect.DeepEqual(e.Command, c.Command) &&
util.EditDistance(c.Labels, e.Labels) == 0 {
containers = append(containers[:i], containers[i+1:]...)
found = true
break
}
}
assert.True(t, found)
}
assert.Empty(t, containers)
}
func testConnectionTxn(t *testing.T, conn db.Conn, spec string) {
compiled, err := stitch.FromJavascript(spec, stitch.DefaultImportGetter)
assert.Nil(t, err)
var connections []db.Connection
conn.Transact(func(view db.Database) error {
updatePolicy(view, db.Master, compiled.String())
connections = view.SelectFromConnection(nil)
return nil
})
exp := compiled.Connections
for _, e := range exp {
found := false
for i, c := range connections {
if e.From == c.From && e.To == c.To && e.MinPort == c.MinPort &&
e.MaxPort == c.MaxPort {
connections = append(
connections[:i], connections[i+1:]...)
found = true
break
}
}
assert.True(t, found)
}
assert.Empty(t, connections)
}
func prog(t *testing.T, code string) stitch.Stitch {
result, err := stitch.FromJavascript(code, stitch.DefaultImportGetter)
if err != nil {
t.Error(err.Error())
return stitch.Stitch{}
}
return result
}
func checkConfig(content string, quiltPath string) error {
oldHTTPGet := stitch.HTTPGet
defer func() {
stitch.HTTPGet = oldHTTPGet
}()
stitch.HTTPGet = func(url string) (*http.Response, error) {
resp := http.Response{
Body: ioutil.NopCloser(bytes.NewBufferString("")),
}
return &resp, nil
}
_, err := stitch.FromJavascript(content, stitch.ImportGetter{
Path: quiltPath,
})
if err != nil {
return err
}
return nil
}
func initSpec(src string) (stitch.Stitch, error) {
return stitch.FromJavascript(src, stitch.ImportGetter{
Path: "../specs",
})
}
func TestPlacementTxn(t *testing.T) {
conn := db.New()
checkPlacement := func(spec string, exp ...db.Placement) {
compiled, err := stitch.FromJavascript(spec,
stitch.DefaultImportGetter)
assert.Nil(t, err)
placements := map[db.Placement]struct{}{}
conn.Transact(func(view db.Database) error {
updatePolicy(view, db.Master, compiled.String())
res := view.SelectFromPlacement(nil)
// Set the ID to 0 so that we can use reflect.DeepEqual.
for _, p := range res {
p.ID = 0
placements[p] = struct{}{}
}
return nil
})
assert.Equal(t, len(exp), len(placements))
for _, p := range exp {
_, ok := placements[p]
assert.True(t, ok)
}
}
pre := `var foo = new Service("foo", [new Container("foo")]);
var bar = new Service("bar", [new Container("bar")]);
var baz = new Service("baz", [new Container("bar")]);
deployment.deploy([foo, bar, baz]);`
// Create an exclusive placement.
spec := pre + `bar.place(new LabelRule(true, foo));`
checkPlacement(spec,
db.Placement{
TargetLabel: "bar",
Exclusive: true,
OtherLabel: "foo",
},
)
// Change the placement from "exclusive" to "on".
spec = pre + `bar.place(new LabelRule(false, foo));`
checkPlacement(spec,
db.Placement{
TargetLabel: "bar",
Exclusive: false,
OtherLabel: "foo",
},
)
// Add another placement constraint.
spec = pre + `bar.place(new LabelRule(false, foo));
bar.place(new LabelRule(true, bar));`
checkPlacement(spec,
db.Placement{
TargetLabel: "bar",
Exclusive: false,
OtherLabel: "foo",
},
db.Placement{
TargetLabel: "bar",
Exclusive: true,
OtherLabel: "bar",
},
)
// Machine placement
spec = pre + `foo.place(new MachineRule(false, {size: "m4.large"}));`
checkPlacement(spec,
db.Placement{
TargetLabel: "foo",
Exclusive: false,
Size: "m4.large",
},
)
// Port placement
spec = pre + `publicInternet.connect(80, foo);
publicInternet.connect(81, foo);`
checkPlacement(spec,
db.Placement{
TargetLabel: "foo",
Exclusive: true,
OtherLabel: "foo",
},
)
spec = pre + `publicInternet.connect(80, foo);
publicInternet.connect(80, bar);
(function() {
publicInternet.connect(81, bar);
publicInternet.connect(81, baz);
})()`
checkPlacement(spec,
db.Placement{
TargetLabel: "foo",
Exclusive: true,
OtherLabel: "foo",
},
db.Placement{
TargetLabel: "bar",
Exclusive: true,
OtherLabel: "bar",
},
db.Placement{
TargetLabel: "foo",
Exclusive: true,
OtherLabel: "bar",
},
db.Placement{
TargetLabel: "bar",
Exclusive: true,
OtherLabel: "foo",
},
db.Placement{
TargetLabel: "baz",
Exclusive: true,
OtherLabel: "baz",
},
db.Placement{
TargetLabel: "bar",
Exclusive: true,
OtherLabel: "baz",
},
db.Placement{
TargetLabel: "baz",
Exclusive: true,
OtherLabel: "bar",
},
)
}