forked from LaconicNetwork/kompose
Merge branch 'main' into feature-1846-network_mode-service
Signed-off-by: jose luis <2064537+sosan@users.noreply.github.com>
This commit is contained in:
@@ -663,7 +663,7 @@ func (k *Kubernetes) UpdateKubernetesObjects(name string, service kobject.Servic
|
||||
if serviceAccountName, ok := service.Labels[compose.LabelServiceAccountName]; ok {
|
||||
template.Spec.ServiceAccountName = serviceAccountName
|
||||
}
|
||||
|
||||
fillInitContainers(template, service)
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -995,6 +995,51 @@ func reformatSecretConfigUnderscoreWithDash(secretConfig types.ServiceSecretConf
|
||||
return newSecretConfig
|
||||
}
|
||||
|
||||
// fillInitContainers looks for an initContainer resources and its passed as labels
|
||||
// if there is no image, it does not fill the initContainer
|
||||
// https://kubernetes.io/docs/concepts/workloads/pods/init-containers/
|
||||
func fillInitContainers(template *api.PodTemplateSpec, service kobject.ServiceConfig) {
|
||||
resourceImage, exist := service.Labels[compose.LabelInitContainerImage]
|
||||
if !exist || resourceImage == "" {
|
||||
return
|
||||
}
|
||||
resourceName, exist := service.Labels[compose.LabelInitContainerName]
|
||||
if !exist || resourceName == "" {
|
||||
resourceName = "init-service"
|
||||
}
|
||||
|
||||
template.Spec.InitContainers = append(template.Spec.InitContainers, api.Container{
|
||||
Name: resourceName,
|
||||
Command: parseContainerCommandsFromStr(service.Labels[compose.LabelInitContainerCommand]),
|
||||
Image: resourceImage,
|
||||
})
|
||||
}
|
||||
|
||||
// parseContainerCommandsFromStr parses a string containing comma-separated commands
|
||||
// returns a slice of strings or a single command
|
||||
// example:
|
||||
// [ "bundle", "exec", "thin", "-p", "3000" ]
|
||||
//
|
||||
// example:
|
||||
// [ "bundle exec thin -p 3000" ]
|
||||
func parseContainerCommandsFromStr(line string) []string {
|
||||
if line == "" {
|
||||
return []string{}
|
||||
}
|
||||
var commands []string
|
||||
if strings.Contains(line, ",") {
|
||||
line = strings.TrimSpace(strings.Trim(line, "[]"))
|
||||
commands = strings.Split(line, ",")
|
||||
// remove space "'
|
||||
for i := range commands {
|
||||
commands[i] = strings.TrimSpace(strings.Trim(commands[i], `"' `))
|
||||
}
|
||||
} else {
|
||||
commands = append(commands, line)
|
||||
}
|
||||
return commands
|
||||
}
|
||||
|
||||
// fixNetworkModeToService is responsible for adjusting the network mode of services in docker compose (services:)
|
||||
// generate a mapping of deployments based on the network mode of each service
|
||||
// merging containers into the destination deployment, and removing transferred deployments
|
||||
|
||||
@@ -742,6 +742,212 @@ func TestRemoveEmptyInterfaces(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func Test_parseContainerCommandsFromStr(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
line string
|
||||
want []string
|
||||
}{
|
||||
{
|
||||
name: "line command without spaces in between",
|
||||
line: `[ "bundle", "exec", "thin", "-p", "3000" ]`,
|
||||
want: []string{
|
||||
"bundle", "exec", "thin", "-p", "3000",
|
||||
},
|
||||
},
|
||||
{
|
||||
name: `line command spaces inside ""`,
|
||||
line: `[ " bundle ", " exec ", " thin ", " -p ", "3000" ]`,
|
||||
want: []string{
|
||||
"bundle", "exec", "thin", "-p", "3000",
|
||||
},
|
||||
},
|
||||
{
|
||||
name: `more use cases for line command spaces inside ""`,
|
||||
line: `[ " bundle ", "exec ", " thin ", " -p ", "3000 " ]`,
|
||||
want: []string{
|
||||
"bundle", "exec", "thin", "-p", "3000",
|
||||
},
|
||||
},
|
||||
{
|
||||
name: `line command without [] and ""`,
|
||||
line: `bundle exec thin -p 3000`,
|
||||
want: []string{
|
||||
"bundle exec thin -p 3000",
|
||||
},
|
||||
},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
if got := parseContainerCommandsFromStr(tt.line); !reflect.DeepEqual(got, tt.want) {
|
||||
t.Errorf("parseContainerCommandsFromStr() = %v, want %v", got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func Test_fillInitContainers(t *testing.T) {
|
||||
type args struct {
|
||||
template *api.PodTemplateSpec
|
||||
service kobject.ServiceConfig
|
||||
}
|
||||
tests := []struct {
|
||||
name string
|
||||
args args
|
||||
want []corev1.Container
|
||||
}{
|
||||
{
|
||||
name: "Testing init container are generated from labels with ,",
|
||||
args: args{
|
||||
template: &api.PodTemplateSpec{},
|
||||
service: kobject.ServiceConfig{
|
||||
Labels: map[string]string{
|
||||
compose.LabelInitContainerName: "name",
|
||||
compose.LabelInitContainerImage: "image",
|
||||
compose.LabelInitContainerCommand: `[ "bundle", "exec", "thin", "-p", "3000" ]`,
|
||||
},
|
||||
},
|
||||
},
|
||||
want: []corev1.Container{
|
||||
{
|
||||
Name: "name",
|
||||
Image: "image",
|
||||
Command: []string{
|
||||
"bundle", "exec", "thin", "-p", "3000",
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Testing init container are generated from labels without ,",
|
||||
args: args{
|
||||
template: &api.PodTemplateSpec{},
|
||||
service: kobject.ServiceConfig{
|
||||
Labels: map[string]string{
|
||||
compose.LabelInitContainerName: "name",
|
||||
compose.LabelInitContainerImage: "image",
|
||||
compose.LabelInitContainerCommand: `bundle exec thin -p 3000`,
|
||||
},
|
||||
},
|
||||
},
|
||||
want: []corev1.Container{
|
||||
{
|
||||
Name: "name",
|
||||
Image: "image",
|
||||
Command: []string{
|
||||
`bundle exec thin -p 3000`,
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: `Testing init container with long command with vars inside and ''`,
|
||||
args: args{
|
||||
template: &api.PodTemplateSpec{},
|
||||
service: kobject.ServiceConfig{
|
||||
Labels: map[string]string{
|
||||
compose.LabelInitContainerName: "init-myservice",
|
||||
compose.LabelInitContainerImage: "busybox:1.28",
|
||||
compose.LabelInitContainerCommand: `['sh', '-c', "until nslookup myservice.$(cat /var/run/secrets/kubernetes.io/serviceaccount/namespace).svc.cluster.local; do echo waiting for myservice; sleep 2; done"]`,
|
||||
},
|
||||
},
|
||||
},
|
||||
want: []corev1.Container{
|
||||
{
|
||||
Name: "init-myservice",
|
||||
Image: "busybox:1.28",
|
||||
Command: []string{
|
||||
"sh", "-c", `until nslookup myservice.$(cat /var/run/secrets/kubernetes.io/serviceaccount/namespace).svc.cluster.local; do echo waiting for myservice; sleep 2; done`,
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: `without image`,
|
||||
args: args{
|
||||
template: &api.PodTemplateSpec{},
|
||||
service: kobject.ServiceConfig{
|
||||
Labels: map[string]string{
|
||||
compose.LabelInitContainerName: "init-myservice",
|
||||
compose.LabelInitContainerImage: "",
|
||||
compose.LabelInitContainerCommand: `['sh', '-c', "until nslookup myservice.$(cat /var/run/secrets/kubernetes.io/serviceaccount/namespace).svc.cluster.local; do echo waiting for myservice; sleep 2; done"]`,
|
||||
},
|
||||
},
|
||||
},
|
||||
want: nil,
|
||||
},
|
||||
{
|
||||
name: `Testing init container without name`,
|
||||
args: args{
|
||||
template: &api.PodTemplateSpec{},
|
||||
service: kobject.ServiceConfig{
|
||||
Labels: map[string]string{
|
||||
compose.LabelInitContainerName: "",
|
||||
compose.LabelInitContainerImage: "busybox:1.28",
|
||||
compose.LabelInitContainerCommand: `['sh', '-c', "until nslookup myservice.$(cat /var/run/secrets/kubernetes.io/serviceaccount/namespace).svc.cluster.local; do echo waiting for myservice; sleep 2; done"]`,
|
||||
},
|
||||
},
|
||||
},
|
||||
want: []corev1.Container{
|
||||
{
|
||||
Name: "init-service",
|
||||
Image: "busybox:1.28",
|
||||
Command: []string{
|
||||
"sh", "-c", `until nslookup myservice.$(cat /var/run/secrets/kubernetes.io/serviceaccount/namespace).svc.cluster.local; do echo waiting for myservice; sleep 2; done`,
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: `Testing init container without command`,
|
||||
args: args{
|
||||
template: &api.PodTemplateSpec{},
|
||||
service: kobject.ServiceConfig{
|
||||
Labels: map[string]string{
|
||||
compose.LabelInitContainerName: "init-service",
|
||||
compose.LabelInitContainerImage: "busybox:1.28",
|
||||
compose.LabelInitContainerCommand: ``,
|
||||
},
|
||||
},
|
||||
},
|
||||
want: []corev1.Container{
|
||||
{
|
||||
Name: "init-service",
|
||||
Image: "busybox:1.28",
|
||||
Command: []string{},
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: `Testing init container without command`,
|
||||
args: args{
|
||||
template: &api.PodTemplateSpec{},
|
||||
service: kobject.ServiceConfig{
|
||||
Labels: map[string]string{
|
||||
compose.LabelInitContainerName: "init-service",
|
||||
compose.LabelInitContainerImage: "busybox:1.28",
|
||||
},
|
||||
},
|
||||
},
|
||||
want: []corev1.Container{
|
||||
{
|
||||
Name: "init-service",
|
||||
Image: "busybox:1.28",
|
||||
Command: []string{},
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
fillInitContainers(tt.args.template, tt.args.service)
|
||||
if !reflect.DeepEqual(tt.args.template.Spec.InitContainers, tt.want) {
|
||||
t.Errorf("Test_fillInitContainers Fail got %v, want %v", tt.args.template.Spec.InitContainers, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func Test_removeFromSlice(t *testing.T) {
|
||||
type args struct {
|
||||
objects []runtime.Object
|
||||
|
||||
Reference in New Issue
Block a user