Пример #1
0
func TestDockerScheduler(t *testing.T) {
	d := NewScheduler(true, "unix:///var/run/docker.sock")
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
	}))

	d.client, _ = docker.NewClient(server.URL)

	var task demand.Task
	task.Demand = 5
	task.Image = "force12io/force12-demo:latest"

	d.InitScheduler("anything", &task)

	err := d.startTask("anything", &task)
	if err != nil {
		// We don't actually expect these to work locally
		// TODO! Some Docker tests that mock out the Docker client
		fmt.Printf("Error %v", err)
	}

	var tasks map[string]demand.Task
	tasks = make(map[string]demand.Task)
	tasks["anything"] = task
	d.CountAllTasks(tasks)
}
Пример #2
0
func TestDockerScheduler(t *testing.T) {
	d := NewScheduler()
	var task demand.Task
	task.Demand = 5
	task.Image = "force12io/force12-demo:latest"

	d.InitScheduler("anything", &task)
	fmt.Printf("%v\n", d)

	err := d.startTask("anything", &task)
	if err != nil {
		fmt.Printf("Error %v", err)
	}
	fmt.Printf("%v\n", d)

	var tasks map[string]demand.Task
	tasks = make(map[string]demand.Task)
	tasks["anything"] = task
	d.CountAllTasks(tasks)

	fmt.Println(tasks)
}
Пример #3
0
func TestDockerInitScheduler(t *testing.T) {
	tests := []struct {
		pullImages bool
	}{
		{
			pullImages: true,
		},
		{
			pullImages: false,
		},
	}

	for _, test := range tests {
		d := NewScheduler(test.pullImages, "unix:///var/run/docker.sock")
		log.Printf("Should I pull images? %v", test.pullImages)
		server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			log.Printf("Received something %v", r)

			// TODO!! Check that we're receiving what we expect
		}))

		log.Printf("Test server at %s", server.URL)
		d.client, _ = docker.NewClient(server.URL)
		log.Printf("Docker client %v", d.client)

		var task demand.Task
		task.Demand = 5
		task.Image = "force12io/force12-demo:latest"

		d.InitScheduler("anything", &task)
		err := d.startTask("anything", &task)
		if err != nil {
			fmt.Printf("Error %v", err)
		}
	}
}