mirror of
https://github.com/ko-build/ko.git
synced 2025-02-13 19:42:26 +02:00
d6b3a3cba3
- Export functions and a variable to enable embedding of ko's `publish` functionality to be embedded in other tools. See https://github.com/GoogleContainerTools/skaffold/pull/5611 - Remove DockerRepo PublishOption and flag. This removes the `DockerRepo` config option and `--docker-repo` flag from the PR. New PR with the extracted config option: https://github.com/google/ko/pull/351 - Fix copyright headers for boilerplate check. - Use DockerRepo PublishOption instead of env var. - Override defaultBaseImage using BuildOptions. Remove exported package global SetDefaultBaseImage and instead allow programmatic override of the default base image using the field `BaseImage` in `options.BuildOptions`. Also fix copyright header years. - Add BuildOptions parameter to getBaseImage This enables access to BaseImage for programmatically overriding the default base image from `.ko.yaml`. - Add UserAgent to BuildOptions and PublishOptions This enables programmatically overriding the `User-Agent` HTTP request header for both pulling the base image and pushing the built image. - Rename MakeBuilder to NewBuilder and MakePublisher to NewPublisher. For more idiomatic constructor function names.
101 lines
2.8 KiB
Go
101 lines
2.8 KiB
Go
/*
|
|
Copyright 2021 Google LLC All Rights Reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package commands
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/google/ko/pkg/build"
|
|
"github.com/google/ko/pkg/commands/options"
|
|
)
|
|
|
|
func TestPublishImages(t *testing.T) {
|
|
repo := "registry.example.com/repository"
|
|
sampleAppDir, err := sampleAppRelDir()
|
|
if err != nil {
|
|
t.Fatalf("sampleAppRelDir(): %v", err)
|
|
}
|
|
tests := []struct {
|
|
description string
|
|
publishArg string
|
|
importpath string
|
|
}{
|
|
{
|
|
description: "import path with ko scheme",
|
|
publishArg: "ko://github.com/google/ko/test",
|
|
importpath: "github.com/google/ko/test",
|
|
},
|
|
{
|
|
description: "import path without ko scheme",
|
|
publishArg: "github.com/google/ko/test",
|
|
importpath: "github.com/google/ko/test",
|
|
},
|
|
{
|
|
description: "file path",
|
|
publishArg: sampleAppDir,
|
|
importpath: "github.com/google/ko/test",
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
ctx := context.Background()
|
|
bo := &options.BuildOptions{
|
|
ConcurrentBuilds: 1,
|
|
}
|
|
builder, err := NewBuilder(ctx, bo)
|
|
if err != nil {
|
|
t.Fatalf("%s: MakeBuilder(): %v", test.description, err)
|
|
}
|
|
po := &options.PublishOptions{
|
|
DockerRepo: repo,
|
|
PreserveImportPaths: true,
|
|
}
|
|
publisher, err := NewPublisher(po)
|
|
if err != nil {
|
|
t.Fatalf("%s: MakePublisher(): %v", test.description, err)
|
|
}
|
|
importpathWithScheme := build.StrictScheme + test.importpath
|
|
refs, err := PublishImages(ctx, []string{test.publishArg}, publisher, builder)
|
|
if err != nil {
|
|
t.Fatalf("%s: PublishImages(): %v", test.description, err)
|
|
}
|
|
ref, exists := refs[importpathWithScheme]
|
|
if !exists {
|
|
t.Errorf("%s: could not find image for importpath %s", test.description, importpathWithScheme)
|
|
}
|
|
gotImageName := ref.Context().Name()
|
|
wantImageName := strings.ToLower(fmt.Sprintf("%s/%s", repo, test.importpath))
|
|
if gotImageName != wantImageName {
|
|
t.Errorf("%s: got %s, wanted %s", test.description, gotImageName, wantImageName)
|
|
}
|
|
}
|
|
}
|
|
|
|
func sampleAppRelDir() (string, error) {
|
|
_, filename, _, ok := runtime.Caller(0)
|
|
if !ok {
|
|
return "", fmt.Errorf("could not get current filename")
|
|
}
|
|
basepath := filepath.Dir(filename)
|
|
testAppDir := filepath.Join(basepath, "..", "..", "test")
|
|
return filepath.Rel(basepath, testAppDir)
|
|
}
|