1
0
mirror of https://github.com/ko-build/ko.git synced 2024-12-15 09:04:09 +02:00
ko-build/pkg/publish/shared_test.go
2019-03-14 14:23:47 -04:00

89 lines
2.3 KiB
Go

// Copyright 2018 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 publish
import (
"testing"
"time"
"github.com/google/go-containerregistry/pkg/name"
v1 "github.com/google/go-containerregistry/pkg/v1"
"github.com/google/go-containerregistry/pkg/v1/random"
)
type slowpublish struct {
sleep time.Duration
}
// slowpublish implements Interface
var _ Interface = (*slowpublish)(nil)
func (sb *slowpublish) Publish(img v1.Image, ref string) (name.Reference, error) {
time.Sleep(sb.sleep)
return makeRef()
}
func TestCaching(t *testing.T) {
duration := 100 * time.Millisecond
ref := "foo"
sb := &slowpublish{duration}
cb, _ := NewCaching(sb)
previousDigest := "not-a-digest"
// Each iteration, we test that the first publish is slow and subsequent
// publishs are fast and return the same reference. For each of these
// iterations we use a new random image, which should invalidate the
// cached reference from previous iterations.
for idx := 0; idx < 3; idx++ {
img, _ := random.Image(256, 8)
start := time.Now()
ref1, err := cb.Publish(img, ref)
if err != nil {
t.Errorf("Publish() = %v", err)
}
end := time.Now()
elapsed := end.Sub(start)
if elapsed < duration {
t.Errorf("Elapsed time %v, wanted >= %s", elapsed, duration)
}
d1 := ref1.String()
if d1 == previousDigest {
t.Errorf("Got same digest as previous iteration, wanted different: %v", d1)
}
previousDigest = d1
start = time.Now()
ref2, err := cb.Publish(img, ref)
if err != nil {
t.Errorf("Publish() = %v", err)
}
end = time.Now()
elapsed = end.Sub(start)
if elapsed >= duration {
t.Errorf("Elapsed time %v, wanted < %s", elapsed, duration)
}
d2 := ref2.String()
if d1 != d2 {
t.Error("Got different references, wanted same")
}
}
}