// NewConfigurationManager creates a new configuration manager
// and registers associated flags.
func NewConfigurationManager(name string) *ConfigurationManager {
flagSet := flag.NewFlagSet(name, flag.ContinueOnError)
m := &ConfigurationManager{
FlagSet: flagSet,
flagResolver: newFlagResolver(flagSet),
clientOptions: clientutil.NewClientOptions(flagSet),
}
// TODO: Support parallel mode
//flag.BoolVar(&m.parallel, "parallel", false, "Whether to run tests in parallel")
//flag.StringVar(&m.manager, "manager", "", "Image to use to manage test output")
return m
}
func main() {
name := filepath.Base(os.Args[0])
if name == "golem_runner" {
runnerMain()
return
}
var (
dockerBinary string
cacheDir string
buildCache string
)
co := clientutil.NewClientOptions()
cm := runner.NewConfigurationManager()
// Move Docker Specific options to separate type
flag.StringVar(&dockerBinary, "db", "", "Docker binary to test")
flag.StringVar(&cacheDir, "cache", "", "Cache directory")
flag.StringVar(&buildCache, "build-cache", "", "Build cache location, if outside of default cache directory")
// TODO: Add swarm flag and host option
flag.Parse()
// TODO: Allow quiet and verbose mode
logrus.SetLevel(logrus.DebugLevel)
if cacheDir == "" {
td, err := ioutil.TempDir("", "build-cache-")
if err != nil {
logrus.Fatalf("Error creating tempdir: %v", err)
}
cacheDir = td
defer os.RemoveAll(td)
}
if buildCache == "" {
buildCache = filepath.Join(cacheDir, "builds")
if err := os.MkdirAll(buildCache, 0755); err != nil {
logrus.Fatalf("Error creating build cache directory")
}
}
c := runner.CacheConfiguration{
ImageCache: runner.NewImageCache(filepath.Join(cacheDir, "images")),
BuildCache: buildutil.NewFSBuildCache(buildCache),
}
if dockerBinary != "" {
v, err := versionutil.BinaryVersion(dockerBinary)
if err != nil {
logrus.Fatalf("Error getting binary version of %s: %v", dockerBinary, err)
}
logrus.Debugf("Using local binary with version %s", v.String())
if err := c.BuildCache.PutVersion(v, dockerBinary); err != nil {
logrus.Fatalf("Error putting %s in cache as %s: %v", dockerBinary, v, err)
}
flag.Set("docker-version", v.String())
}
client, err := runner.NewDockerClient(co)
if err != nil {
logrus.Fatalf("Failed to create client: %v", err)
}
v, err := client.Version()
if err != nil {
logrus.Fatalf("Error getting version: %v", err)
}
serverVersion, err := versionutil.ParseVersion(v.Get("Version"))
if err != nil {
logrus.Fatalf("Unexpected version value: %s", v.Get("Version"))
}
// TODO: Support arbitrary load version instead of server version by
// starting up separate daemon for load
// TODO: Check cache here to ensure that load will not have issues
logrus.Debugf("Using docker daemon for image export, version %s", serverVersion)
r, err := cm.CreateRunner(serverVersion, c)
if err != nil {
logrus.Fatalf("Error creating runner: %v", err)
}
if err := r.Build(client); err != nil {
logrus.Fatalf("Error building test images: %v", err)
}
if err := r.Run(client); err != nil {
logrus.Fatalf("Error running tests: %v", err)
}
}