package main

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"go.skia.org/infra/go/testutils"
	cloudbuild "google.golang.org/api/cloudbuild/v1"
)

func TestFindImages(t *testing.T) {
	testutils.SmallTest(t)
	*project = "skia-public"
	Init()
	buildInfo := cloudbuild.Build{
		Results: &cloudbuild.Results{
			Images: []*cloudbuild.BuiltImage{
				&cloudbuild.BuiltImage{Name: "testing-clang9"},
				&cloudbuild.BuiltImage{Name: "gcr.io/skia-public/fiddler:prod"},
				&cloudbuild.BuiltImage{Name: "gcr.io/skia-public/skottie:prod"},
			},
		},
	}
	images := imagesFromInfo([]string{"fiddler", "skottie"}, buildInfo)
	assert.Equal(t, "gcr.io/skia-public/fiddler:prod", images[0])
	assert.Equal(t, "gcr.io/skia-public/skottie:prod", images[1])

	images = imagesFromInfo([]string{"skottie"}, buildInfo)
	assert.Equal(t, "gcr.io/skia-public/skottie:prod", images[0])
}

func TestBaseImageName(t *testing.T) {
	testutils.SmallTest(t)
	assert.Equal(t, "", baseImageName(""))
	assert.Equal(t, "", baseImageName("debian"))
	assert.Equal(t, "fiddler", baseImageName("gcr.io/skia-public/fiddler:prod"))
	assert.Equal(t, "docserver", baseImageName("gcr.io/skia-public/docserver:123456"))
}