func (s *S) TestGetProcessesFromProcfile(c *check.C) {
tests := []struct {
procfile string
expected map[string][]string
}{
{procfile: "", expected: map[string][]string{}},
{procfile: "invalid", expected: map[string][]string{}},
{procfile: "web: a b c", expected: map[string][]string{
"web": {"a b c"},
}},
{procfile: "web: a b c\nworker: \t x y z \r ", expected: map[string][]string{
"web": {"a b c"},
"worker": {"x y z"},
}},
{procfile: "web:abc\nworker:xyz", expected: map[string][]string{
"web": {"abc"},
"worker": {"xyz"},
}},
{procfile: "web: a b c\r\nworker:x\r\nworker2: z\r\n", expected: map[string][]string{
"web": {"a b c"},
"worker": {"x"},
"worker2": {"z"},
}},
}
for i, t := range tests {
v := image.GetProcessesFromProcfile(t.procfile)
c.Check(v, check.DeepEquals, t.expected, check.Commentf("failed test %d", i))
}
}
func PrepareImageForDeploy(args PrepareImageArgs) (string, error) {
fmt.Fprintf(args.Out, "---- Inspecting image %q ----\n", args.ImageId)
procfile := image.GetProcessesFromProcfile(args.ProcfileRaw)
imageInspect, err := args.Client.InspectImage(args.ImageId)
if err != nil {
return "", err
}
if len(procfile) == 0 {
fmt.Fprintln(args.Out, " ---> Procfile not found, using entrypoint and cmd")
procfile["web"] = append(imageInspect.Config.Entrypoint, imageInspect.Config.Cmd...)
}
for k, v := range procfile {
fmt.Fprintf(args.Out, " ---> Process %q found with commands: %q\n", k, v)
}
newImage, err := image.AppNewImageName(args.App.GetName())
if err != nil {
return "", err
}
imageInfo := strings.Split(newImage, ":")
repo, tag := strings.Join(imageInfo[:len(imageInfo)-1], ":"), imageInfo[len(imageInfo)-1]
err = args.Client.TagImage(args.ImageId, docker.TagImageOptions{Repo: repo, Tag: tag, Force: true})
if err != nil {
return "", err
}
registry, err := config.GetString("docker:registry")
if err != nil {
return "", err
}
fmt.Fprintf(args.Out, "---- Pushing image %q to tsuru ----\n", newImage)
pushOpts := docker.PushImageOptions{
Name: repo,
Tag: tag,
Registry: registry,
OutputStream: args.Out,
InactivityTimeout: net.StreamInactivityTimeout,
RawJSONStream: true,
}
err = args.Client.PushImage(pushOpts, args.AuthConfig)
if err != nil {
return "", err
}
imageData := image.ImageMetadata{
Name: newImage,
Processes: procfile,
}
if len(imageInspect.Config.ExposedPorts) > 1 {
return "", errors.New("Too many ports. You should especify which one you want to.")
}
for k := range imageInspect.Config.ExposedPorts {
imageData.ExposedPort = string(k)
}
err = imageData.Save()
if err != nil {
return "", err
}
return newImage, nil
}