func ListNonTerminalTasks() ([]*types.EremeticTask, error) {
var tasks []*types.EremeticTask
err := ensureDB()
if err != nil {
return tasks, err
}
err = boltdb.View(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte("tasks"))
if b == nil {
return bolt.ErrBucketNotFound
}
b.ForEach(func(_, v []byte) error {
var task types.EremeticTask
json.Unmarshal(v, &task)
if !task.IsTerminated() {
tasks = append(tasks, &task)
}
return nil
})
return nil
})
return tasks, err
}
func createTaskInfo(task types.EremeticTask, offer *mesos.Offer) (types.EremeticTask, *mesos.TaskInfo) {
task.FrameworkId = *offer.FrameworkId.Value
task.SlaveId = *offer.SlaveId.Value
task.Hostname = *offer.Hostname
var environment []*mesos.Environment_Variable
for k, v := range task.Environment {
environment = append(environment, &mesos.Environment_Variable{
Name: proto.String(k),
Value: proto.String(v),
})
}
environment = append(environment, &mesos.Environment_Variable{
Name: proto.String("MESOS_TASK_ID"),
Value: proto.String(task.ID),
})
var volumes []*mesos.Volume
for _, v := range task.Volumes {
volumes = append(volumes, &mesos.Volume{
Mode: mesos.Volume_RW.Enum(),
ContainerPath: proto.String(v.ContainerPath),
HostPath: proto.String(v.HostPath),
})
}
var uris []*mesos.CommandInfo_URI
for _, v := range task.URIs {
uris = append(uris, &mesos.CommandInfo_URI{
Value: proto.String(v),
Extract: proto.Bool(isArchive(v)),
})
}
return task, &mesos.TaskInfo{
TaskId: &mesos.TaskID{
Value: proto.String(task.ID),
},
SlaveId: offer.SlaveId,
Name: proto.String(task.Name),
Command: &mesos.CommandInfo{
Value: proto.String(task.Command),
User: proto.String(task.User),
Environment: &mesos.Environment{
Variables: environment,
},
Uris: uris,
},
Container: &mesos.ContainerInfo{
Type: mesos.ContainerInfo_DOCKER.Enum(),
Docker: &mesos.ContainerInfo_DockerInfo{
Image: proto.String(task.Image),
},
Volumes: volumes,
},
Resources: []*mesos.Resource{
mesosutil.NewScalarResource("cpus", task.TaskCPUs),
mesosutil.NewScalarResource("mem", task.TaskMem),
},
}
}
func TestTask(t *testing.T) {
status := []types.Status{
types.Status{
Status: mesos.TaskState_TASK_RUNNING.String(),
Time: time.Now().Unix(),
},
}
Convey("createID", t, func() {
Convey("Given a string", func() {
Convey("It should build the appropriate ID", func() {
So(createID("1234"), ShouldEqual, "eremetic-task.1234")
})
})
})
Convey("createEremeticTask", t, func() {
request := types.Request{
TaskCPUs: 0.5,
TaskMem: 22.0,
DockerImage: "busybox",
Command: "echo hello",
}
Convey("No volume or environment specified", func() {
task, err := createEremeticTask(request)
So(err, ShouldBeNil)
So(task, ShouldNotBeNil)
So(task.Command, ShouldEqual, "echo hello")
So(task.User, ShouldEqual, "root")
So(task.Environment, ShouldBeEmpty)
So(task.Image, ShouldEqual, "busybox")
So(task.Volumes, ShouldBeEmpty)
So(task.Status[0].Status, ShouldEqual, "TASK_STAGING")
})
Convey("Given a volume and environment", func() {
var volumes []types.Volume
var environment = make(map[string]string)
environment["foo"] = "bar"
volumes = append(volumes, types.Volume{
ContainerPath: "/var/www",
HostPath: "/var/www",
})
request.Volumes = volumes
request.Environment = environment
task, err := createEremeticTask(request)
So(err, ShouldBeNil)
So(task.Environment, ShouldContainKey, "foo")
So(task.Environment["foo"], ShouldEqual, "bar")
So(task.Volumes[0].ContainerPath, ShouldEqual, volumes[0].ContainerPath)
So(task.Volumes[0].HostPath, ShouldEqual, volumes[0].HostPath)
})
Convey("Given uri to download", func() {
request.URIs = []string{"http://foobar.local/kitten.jpg"}
task, err := createEremeticTask(request)
So(err, ShouldBeNil)
So(task.URIs, ShouldHaveLength, 1)
So(task.URIs, ShouldContain, request.URIs[0])
})
})
Convey("createTaskInfo", t, func() {
eremeticTask := types.EremeticTask{
TaskCPUs: 0.2,
TaskMem: 0.5,
Command: "echo hello",
Image: "busybox",
Status: status,
ID: "eremetic-task.1234",
Name: "Eremetic task 17",
}
offer := mesos.Offer{
FrameworkId: &mesos.FrameworkID{
Value: proto.String("framework-id"),
},
SlaveId: &mesos.SlaveID{
Value: proto.String("slave-id"),
},
Hostname: proto.String("hostname"),
}
Convey("No volume or environment specified", func() {
net, taskInfo := createTaskInfo(eremeticTask, &offer)
So(taskInfo.TaskId.GetValue(), ShouldEqual, eremeticTask.ID)
So(taskInfo.GetName(), ShouldEqual, eremeticTask.Name)
So(taskInfo.GetResources()[0].GetScalar().GetValue(), ShouldEqual, eremeticTask.TaskCPUs)
So(taskInfo.GetResources()[1].GetScalar().GetValue(), ShouldEqual, eremeticTask.TaskMem)
So(taskInfo.Container.GetType().String(), ShouldEqual, "DOCKER")
So(taskInfo.Container.Docker.GetImage(), ShouldEqual, "busybox")
So(net.SlaveId, ShouldEqual, "slave-id")
})
Convey("Given a volume and environment", func() {
volumes := []types.Volume{types.Volume{
ContainerPath: "/var/www",
HostPath: "/var/www",
}}
environment := make(map[string]string)
environment["foo"] = "bar"
eremeticTask.Environment = environment
eremeticTask.Volumes = volumes
_, taskInfo := createTaskInfo(eremeticTask, &offer)
So(taskInfo.TaskId.GetValue(), ShouldEqual, eremeticTask.ID)
So(taskInfo.Container.Volumes[0].GetContainerPath(), ShouldEqual, volumes[0].ContainerPath)
So(taskInfo.Container.Volumes[0].GetHostPath(), ShouldEqual, volumes[0].HostPath)
So(taskInfo.Command.Environment.Variables[0].GetName(), ShouldEqual, "foo")
So(taskInfo.Command.Environment.Variables[0].GetValue(), ShouldEqual, "bar")
So(taskInfo.Command.Environment.Variables[1].GetName(), ShouldEqual, "MESOS_TASK_ID")
So(taskInfo.Command.Environment.Variables[1].GetValue(), ShouldEqual, eremeticTask.ID)
})
Convey("Given picture to download", func() {
eremeticTask.URIs = []string{"http://foobar.local/kitten.jpg"}
_, taskInfo := createTaskInfo(eremeticTask, &offer)
So(taskInfo.TaskId.GetValue(), ShouldEqual, eremeticTask.ID)
So(taskInfo.Command.Uris, ShouldHaveLength, 1)
So(taskInfo.Command.Uris[0].GetValue(), ShouldEqual, eremeticTask.URIs[0])
So(taskInfo.Command.Uris[0].GetExecutable(), ShouldBeFalse)
So(taskInfo.Command.Uris[0].GetExtract(), ShouldBeFalse)
So(taskInfo.Command.Uris[0].GetCache(), ShouldBeFalse)
})
Convey("Given archive to download", func() {
eremeticTask.URIs = []string{"http://foobar.local/cats.zip"}
_, taskInfo := createTaskInfo(eremeticTask, &offer)
So(taskInfo.TaskId.GetValue(), ShouldEqual, eremeticTask.ID)
So(taskInfo.Command.Uris, ShouldHaveLength, 1)
So(taskInfo.Command.Uris[0].GetValue(), ShouldEqual, eremeticTask.URIs[0])
So(taskInfo.Command.Uris[0].GetExecutable(), ShouldBeFalse)
So(taskInfo.Command.Uris[0].GetExtract(), ShouldBeTrue)
So(taskInfo.Command.Uris[0].GetCache(), ShouldBeFalse)
})
})
}
