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) }