Golang container.Container類代碼示例

func extendsOne(child, parent *container.Container, extendsChan map[string]chan bool) {
	// Waiting for dependencies to be extended
	<-extendsChan[parent.Name]

	// Extends the application
	child.ExtendsContainer(parent)
	close(extendsChan[child.Name])
}

func (maestro *Maestro) startContainer(currentContainer *container.Container, done map[string]chan bool) {
	// Waiting for dependencies to start
	for _, dependency := range currentContainer.Dependencies {
		<-done[dependency.Name]
	}

	currentContainer.Start()

	close(done[currentContainer.Name])
}

func (s *ContainerTestSuite) TestGetFirstPortShouldReturnTheFirstDeclaredPort(c *C) {
	container := container.Container{Name: "Test"}

	c.Check(container.GetFirstPort(), Equals, "")

	container.Ports = make(map[string]string)
	container.Ports["80"] = "8080"
	container.Ports["9000"] = "9000"

	c.Check(container.GetFirstPort(), Equals, "80")
}

func startOne(currentContainer *container.Container, rebuild bool, done map[string]chan bool) {
	// Waiting for dependencies to be started
	for _, dependency := range currentContainer.Dependencies {
		if dependency.Name == currentContainer.Name {
			util.LogError("Application " + currentContainer.Name + " can't be linked with itself.")
		}

		<-done[dependency.Name]
	}

	currentContainer.Start(rebuild)

	close(done[currentContainer.Name])
}

func (s *ContainerTestSuite) TestStartedApplicationShouldRetrieveItsIp(c *C) {
	// Create a gomock controller, and arrange for it's finish to be called
	ctrl := gomock.NewController(c)
	defer ctrl.Finish()

	// Setup the docker mock package
	docker.MOCK().SetController(ctrl)
	docker.EXPECT().Start(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return("123")
	docker.EXPECT().Inspect(gomock.Any()).Return([]byte("[{\"ID\": \"123\", \"State\":{\"Running\": true}, \"NetworkSettings\": {\"IPAddress\": \"172.17.0.10\"}}]"), nil)

	// @TODO : find a way to mock time.Sleep
	container := container.Container{Name: "Test"}
	container.Start(true)

	c.Check(container.IsRunning(), Equals, true)
	c.Check(container.Ip, Equals, "172.17.0.10")
}

func (s *ContainerTestSuite) TestCallCleanShouldStopTheContainer(c *C) {
	// Create a gomock controller, and arrange for it's finish to be called
	ctrl := gomock.NewController(c)
	defer ctrl.Finish()

	// Setup the docker mock package
	docker.MOCK().SetController(ctrl)
	docker.EXPECT().Inspect(gomock.Any()).Return([]byte("[{\"ID\": \"123\", \"State\":{\"Running\": false}, \"NetworkSettings\": {\"IPAddress\": \"172.17.0.10\"}}]"), nil)
	docker.EXPECT().Kill(gomock.Any()).Return()
	docker.EXPECT().Remove(gomock.Any()).Return()

	done := make(chan bool, 1)
	container := container.Container{Name: "Test"}
	container.Clean(done)
	<-done

	c.Check(container.IsRunning(), Equals, false)
}

func (s *ContainerTestSuite) TestContainerWithReadyDependenciesShouldBeReady(c *C) {
	dep1 := container.Container{Name: "abc"}
	dep1.Running = false

	dep2 := container.Container{Name: "abc"}
	dep2.Running = true

	mainContainer := container.Container{Name: "Test"}
	mainContainer.Dependencies = make([]*container.Container, 0)

	mainContainer.AddDependency(&dep1)
	mainContainer.AddDependency(&dep2)

	c.Check(mainContainer.IsReady(), Equals, false)

	dep1.Running = true
	c.Check(mainContainer.IsReady(), Equals, true)
}