data_test.go

package stage

import (
	"fmt"

	. "github.com/onsi/gomega"

	"github.com/werf/werf/pkg/config"
	"github.com/werf/werf/pkg/util"
)

type TestDependencies struct {
	Dependencies   []*TestDependency
	ExpectedDigest string
}

type TestDependency struct {
	ImageName string

	TargetEnvImageName string
	TargetEnvImageRepo string
	TargetEnvImageTag  string
	TargetEnvImageID   string

	TargetBuildArgImageName string
	TargetBuildArgImageRepo string
	TargetBuildArgImageTag  string
	TargetBuildArgImageID   string

	DockerImageRepo string
	DockerImageTag  string
	DockerImageID   string
}

func (dep *TestDependency) GetDockerImageName() string {
	return fmt.Sprintf("%s:%s", dep.DockerImageRepo, dep.DockerImageTag)
}

func (dep *TestDependency) ToConfigDependency() *config.Dependency {
	depCfg := &config.Dependency{ImageName: dep.ImageName}

	if dep.TargetEnvImageName != "" {
		depCfg.Imports = append(depCfg.Imports, &config.DependencyImport{
			Type:      config.ImageNameImport,
			TargetEnv: dep.TargetEnvImageName,
		})
	}
	if dep.TargetEnvImageRepo != "" {
		depCfg.Imports = append(depCfg.Imports, &config.DependencyImport{
			Type:      config.ImageRepoImport,
			TargetEnv: dep.TargetEnvImageRepo,
		})
	}
	if dep.TargetEnvImageTag != "" {
		depCfg.Imports = append(depCfg.Imports, &config.DependencyImport{
			Type:      config.ImageTagImport,
			TargetEnv: dep.TargetEnvImageTag,
		})
	}
	if dep.TargetEnvImageID != "" {
		depCfg.Imports = append(depCfg.Imports, &config.DependencyImport{
			Type:      config.ImageIDImport,
			TargetEnv: dep.TargetEnvImageID,
		})
	}

	if dep.TargetBuildArgImageName != "" {
		depCfg.Imports = append(depCfg.Imports, &config.DependencyImport{
			Type:           config.ImageNameImport,
			TargetBuildArg: dep.TargetBuildArgImageName,
		})
	}
	if dep.TargetBuildArgImageRepo != "" {
		depCfg.Imports = append(depCfg.Imports, &config.DependencyImport{
			Type:           config.ImageRepoImport,
			TargetBuildArg: dep.TargetBuildArgImageRepo,
		})
	}
	if dep.TargetBuildArgImageTag != "" {
		depCfg.Imports = append(depCfg.Imports, &config.DependencyImport{
			Type:           config.ImageTagImport,
			TargetBuildArg: dep.TargetBuildArgImageTag,
		})
	}
	if dep.TargetBuildArgImageID != "" {
		depCfg.Imports = append(depCfg.Imports, &config.DependencyImport{
			Type:           config.ImageIDImport,
			TargetBuildArg: dep.TargetBuildArgImageID,
		})
	}

	return depCfg
}

func GetConfigDependencies(dependencies []*TestDependency) (res []*config.Dependency) {
	for _, dep := range dependencies {
		res = append(res, dep.ToConfigDependency())
	}

	return
}

func CheckImageDependenciesAfterPrepare(img *LegacyImageStub, dependencies []*TestDependency) {
	for _, dep := range dependencies {
		if dep.TargetEnvImageName != "" {
			Expect(img._Container._ServiceCommitChangeOptions.Env[dep.TargetEnvImageName]).To(Equal(dep.GetDockerImageName()))
		}
		if dep.TargetEnvImageRepo != "" {
			Expect(img._Container._ServiceCommitChangeOptions.Env[dep.TargetEnvImageRepo]).To(Equal(dep.DockerImageRepo))
		}
		if dep.TargetEnvImageTag != "" {
			Expect(img._Container._ServiceCommitChangeOptions.Env[dep.TargetEnvImageTag]).To(Equal(dep.DockerImageTag))
		}
		if dep.TargetEnvImageID != "" {
			Expect(img._Container._ServiceCommitChangeOptions.Env[dep.TargetEnvImageID]).To(Equal(dep.DockerImageID))
		}

		if dep.TargetBuildArgImageName != "" {
			Expect(util.IsStringsContainValue(img._DockerfileImageBuilder.BuildDockerfileOptions.BuildArgs, fmt.Sprintf("%s=%s", dep.TargetBuildArgImageName, dep.GetDockerImageName())))
		}
		if dep.TargetBuildArgImageRepo != "" {
			Expect(util.IsStringsContainValue(img._DockerfileImageBuilder.BuildDockerfileOptions.BuildArgs, fmt.Sprintf("%s=%s", dep.TargetBuildArgImageRepo, dep.DockerImageRepo)))
		}
		if dep.TargetBuildArgImageTag != "" {
			Expect(util.IsStringsContainValue(img._DockerfileImageBuilder.BuildDockerfileOptions.BuildArgs, fmt.Sprintf("%s=%s", dep.TargetBuildArgImageTag, dep.DockerImageTag)))
		}
		if dep.TargetBuildArgImageID != "" {
			Expect(util.IsStringsContainValue(img._DockerfileImageBuilder.BuildDockerfileOptions.BuildArgs, fmt.Sprintf("%s=%s", dep.TargetBuildArgImageID, dep.DockerImageID)))
		}
	}
}