Esempio n. 1
0
func createTestTask(taskGuid string) *models.Task {
	task := model_helpers.NewValidTask(taskGuid)
	task.Domain = "test-domain"
	task.TaskDefinition.RootFs = "some:rootfs"
	task.TaskDefinition.Action = models.WrapAction(&models.RunAction{User: "******", Path: "/bin/true"})
	return task
}
			statusCodes = make(chan int)
			reqCount = make(chan struct{}, task_handler.POOL_SIZE)
			fakeServer.RouteToHandler("POST", "/the-callback/url", func(w http.ResponseWriter, req *http.Request) {
				reqCount <- struct{}{}
				w.WriteHeader(<-statusCodes)
			})

			callbackURL = fakeServer.URL() + "/the-callback/url"
		})

		AfterEach(func() {
			close(statusCodes)
		})

		simulateTaskCompleting := func() {
			task := model_helpers.NewValidTask("the-task-guid")
			task.CompletionCallbackUrl = callbackURL
			enqueue <- task
		}

		Context("when the task has a completion callback URL", func() {
			It("marks the task as resolving", func() {
				Expect(fakeBBS.ResolvingTaskCallCount()).To(Equal(0))

				simulateTaskCompleting()
				statusCodes <- 200

				Eventually(fakeBBS.ResolveTaskCallCount).Should(Equal(1))
				actualGuid := fakeBBS.ResolveTaskArgsForCall(0)
				Expect(actualGuid).To(Equal("the-task-guid"))
			})
			validCreateRequest = receptor.TaskCreateRequest{
				TaskGuid:   "task-guid-1",
				Domain:     "test-domain",
				RootFS:     "docker://docker",
				Action:     models.WrapAction(&models.RunAction{User: "******", Path: "/bin/bash", Args: []string{"echo", "hi"}}),
				MemoryMB:   24,
				DiskMB:     12,
				CPUWeight:  10,
				LogGuid:    "guid",
				LogSource:  "source-name",
				ResultFile: "result-file",
				Annotation: "some annotation",
				Privileged: true,
			}

			expectedTask = model_helpers.NewValidTask("task-guid-1")
			expectedTask.Domain = "test-domain"
			expectedTask.TaskDefinition.RootFs = "docker://docker"
			expectedTask.TaskDefinition.Action = models.WrapAction(&models.RunAction{User: "******", Path: "/bin/bash", Args: []string{"echo", "hi"}})
			expectedTask.TaskDefinition.MemoryMb = 24
			expectedTask.TaskDefinition.DiskMb = 12
			expectedTask.TaskDefinition.CpuWeight = 10
			expectedTask.TaskDefinition.LogGuid = "guid"
			expectedTask.TaskDefinition.LogSource = "source-name"
			expectedTask.TaskDefinition.ResultFile = "result-file"
			expectedTask.TaskDefinition.Annotation = "some annotation"
			expectedTask.TaskDefinition.Privileged = true
			expectedTask.TaskDefinition.EgressRules = nil
			expectedTask.TaskDefinition.MetricsGuid = ""
			expectedTask.TaskDefinition.EnvironmentVariables = nil
Esempio n. 4
0
import (
	"github.com/cloudfoundry-incubator/bbs/models"
	"github.com/cloudfoundry-incubator/bbs/models/internal/model_helpers"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
)

var _ = Describe("TaskDB", func() {
	Describe("Tasks", func() {
		Context("when there are tasks", func() {
			var expectedTasks []*models.Task

			BeforeEach(func() {
				expectedTasks = []*models.Task{
					model_helpers.NewValidTask("a-guid"), model_helpers.NewValidTask("b-guid"),
				}

				for _, t := range expectedTasks {
					etcdHelper.SetRawTask(t)
				}
			})

			It("returns all the tasks", func() {
				tasks, err := etcdDB.Tasks(logger, nil)
				Expect(err).NotTo(HaveOccurred())
				Expect(tasks.GetTasks()).To(ConsistOf(expectedTasks))
			})

			It("can filter", func() {
				tasks, err := etcdDB.Tasks(logger, func(t *models.Task) bool { return t.TaskGuid == "b-guid" })
		server.Close()
	})

	JustBeforeEach(func() {
		var err error

		response, err = http.Post(server.URL, "application/json", bytes.NewBuffer(payload))
		Expect(err).NotTo(HaveOccurred())
	})

	Describe("when the handler receives a task", func() {
		var task *models.Task

		BeforeEach(func() {
			var err error
			task = model_helpers.NewValidTask("guid")
			payload, err = json.Marshal([]*models.Task{task})
			Expect(err).NotTo(HaveOccurred())
		})

		It("enqueues the task on the worker", func() {
			Expect(enqueue).To(Receive(Equal(task)))
		})

		It("returns 202", func() {
			Expect(response.StatusCode).To(Equal(http.StatusAccepted))
		})
	})

	Describe("when the handler receives a bogus payload", func() {
		BeforeEach(func() {
Esempio n. 6
0
)

var _ = Describe("Task API", func() {
	var (
		actualTasks   []*models.Task
		expectedTasks []*models.Task

		filter db.TaskFilter

		getErr error
	)

	BeforeEach(func() {
		filter = nil
		actualTasks = nil
		expectedTasks = []*models.Task{model_helpers.NewValidTask("a-guid"), model_helpers.NewValidTask("b-guid")}
		expectedTasks[1].Domain = "b-domain"
		expectedTasks[1].CellId = "b-cell"
		for _, t := range expectedTasks {
			etcdHelper.SetRawTask(t)
		}
	})

	Describe("GET /v1/tasks", func() {
		Context("all tasks", func() {
			BeforeEach(func() {
				actualTasks, getErr = client.Tasks()
			})

			It("responds without error", func() {
				Expect(getErr).NotTo(HaveOccurred())