ct/go/util/gs_test.go

package util

import (
	"io/ioutil"
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/assert"
	"go.skia.org/infra/go/util"
)

// Will need a local valid google_storage_token.data file with read write access
// to run the below test.
func Auth_TestDownloadSwarmingArtifacts(t *testing.T) {
	testPagesetsDirName := filepath.Join("unit-tests", "util", "page_sets")

	gs, err := NewGcsUtil(nil)
	assert.NoError(t, err)

	localDir, err := ioutil.TempDir("", "util_test_")
	assert.NoError(t, err)
	defer util.RemoveAll(localDir)
	pageSetToIndex, err := gs.DownloadSwarmingArtifacts(localDir, testPagesetsDirName, "10k", 1, 2)
	if err != nil {
		t.Errorf("Unexpected error: %s", err)
	}

	// Examine contents of returned dictionary.
	assert.Equal(t, 2, len(pageSetToIndex))
	assert.Equal(t, 1, pageSetToIndex[filepath.Join(localDir, "1.py")])
	assert.Equal(t, 2, pageSetToIndex[filepath.Join(localDir, "2.py")])
	// Examine contents of the local directory.
	files, err := ioutil.ReadDir(localDir)
	assert.NoError(t, err)
	assert.Equal(t, 2, len(files))
	assert.Equal(t, "1.py", files[0].Name())
	assert.Equal(t, "2.py", files[1].Name())
}