Merge pull request #68 from getamis/link-and-entrypoint-option

test: add DB link and entrypoint option

test/docker.go

@@ -37,8 +37,10 @@ type Container struct {
 	exposedPorts map[docker.Port]struct{}
 
 	ports         []string
+	entrypoint    []string
 	runArgs       []string
 	envs          []string
+	links         []string
 	container     *docker.Container
 	healthChecker ContainerCallback
 	initializer   ContainerCallback
@@ -66,6 +68,7 @@ func NewDockerContainer(opts ...Option) *Container {
 		healthChecker: func(c *Container) error {
 			return nil
 		},
+		links: []string{},
 	}
 
 	for _, opt := range opts {
@@ -84,12 +87,14 @@ func NewDockerContainer(opts ...Option) *Container {
 		Name: c.name + generateNameSuffix(),
 		Config: &docker.Config{
 			Image:        c.imageRespository + ":" + c.imageTag,
+			Entrypoint:   c.entrypoint,
 			Cmd:          c.runArgs,
 			ExposedPorts: c.exposedPorts,
 			Env:          c.envs,
 		},
 		HostConfig: &docker.HostConfig{
 			PortBindings: c.portBindings,
+			Links:        c.links,
 		},
 		Context: context.TODO(),
 	})

test/docker_options.go

@@ -83,6 +83,12 @@ func Ports(ports ...int) Option {
 	}
 }
 
+func Entrypoint(entrypoint []string) Option {
+	return func(c *Container) {
+		c.entrypoint = entrypoint
+	}
+}
+
 func RunOptions(args []string) Option {
 	return func(c *Container) {
 		c.runArgs = args
@@ -95,6 +101,16 @@ func DockerEnv(env []string) Option {
 	}
 }
 
+func DockerRunLink(link string) Option {
+	return func(c *Container) {
+		if c.links == nil {
+			c.links = []string{link}
+		} else {
+			c.links = append(c.links, link)
+		}
+	}
+}
+
 func HealthChecker(checker ContainerCallback) Option {
 	return func(c *Container) {
 		c.healthChecker = checker

test/migration.go

@@ -14,6 +14,8 @@ type MigrationOptions struct {
 	// this command will override the default command.
 	// "bundle" "exec" "rake" "db:migrate"
 	Command []string
+	// override default entrypoint
+	Entrypoint []string
 }
 
 // RunMigrationContainer creates the migration container and connects to the
@@ -67,6 +69,8 @@ func RunMigrationContainer(dbContainer *SQLContainer, options MigrationOptions)
 			},
 		),
 		RunOptions(command),
+		DockerRunLink(dbContainer.container.ID+":DB"),
+		Entrypoint(options.Entrypoint),
 	)
 
 	if err := container.Start(); err != nil {
@@ -128,6 +132,8 @@ func RunGoMigrationContainer(dbContainer *SQLContainer, options MigrationOptions
 		ImageRepository(options.ImageRepository),
 		ImageTag(options.ImageTag),
 		RunOptions(command),
+		DockerRunLink(dbContainer.container.ID+":DB"),
+		Entrypoint(options.Entrypoint),
 	)
 
 	if err := container.Start(); err != nil {