mirror of
https://github.com/GoogleContainerTools/kaniko
synced 2026-10-04 05:58:59 +02:00
To allow contributors to run the integration tests with their own GCS buckets and image repos (since not all contributors will have accesss to the projects used by the kaniko maintainers) this updates the integration tests so that these can be provided on the command line. This allows tests to be run individually, without using `make integration-test`. Previously, part of the test setup was done in the shell script (creating the context tarball that is required for the tests that build images with context). Instead it will be done in the test iself, so we can use `go test` to run tests individually if we want to. If we are running only one individual test, we don't want to build all of the images, so this commit creates a builder which tracks which images it has built and can be used by a tests to check if it should build an image before running, or it will use the images that have already been built by a previous test. The name of the context tarball has also been made unique (it includes the unix timestamp) to avoid potential test flakes if two tests using the same GCS bucket run simultaneously.
74 lines
2.4 KiB
Go
74 lines
2.4 KiB
Go
/*
|
|
Copyright 2018 Google LLC
|
|
|
|
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 integration
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"time"
|
|
)
|
|
|
|
// CreateIntegrationTarball will take the contents of the integration directory and write
|
|
// them to a tarball in a temmporary dir. It will return a path to the tarball.
|
|
func CreateIntegrationTarball() (string, error) {
|
|
log.Println("Creating tarball of integration test files to use as build context")
|
|
dir, err := os.Getwd()
|
|
if err != nil {
|
|
return "", fmt.Errorf("Failed find path to integration dir: %s", err)
|
|
}
|
|
tempDir, err := ioutil.TempDir("", "")
|
|
if err != nil {
|
|
return "", fmt.Errorf("Failed to create temporary directoy to hold tarball: %s", err)
|
|
}
|
|
contextFile := fmt.Sprintf("%s/context_%d.tar.gz", tempDir, time.Now().UnixNano())
|
|
cmd := exec.Command("tar", "-C", dir, "-zcvf", contextFile, ".")
|
|
_, err = RunCommandWithoutTest(cmd)
|
|
if err != nil {
|
|
return "", fmt.Errorf("Failed to create build context tarball from integration dir: %s", err)
|
|
}
|
|
return contextFile, err
|
|
}
|
|
|
|
// UploadFileToBucket will upload the at filePath to gcsBucket. It will return the path
|
|
// of the file in gcsBucket.
|
|
func UploadFileToBucket(gcsBucket string, filePath string) (string, error) {
|
|
log.Printf("Uploading file at %s to GCS bucket at %s\n", filePath, gcsBucket)
|
|
|
|
cmd := exec.Command("gsutil", "cp", filePath, gcsBucket)
|
|
_, err := RunCommandWithoutTest(cmd)
|
|
if err != nil {
|
|
return "", fmt.Errorf("Failed to copy tarball to GCS bucket %s: %s", gcsBucket, err)
|
|
}
|
|
|
|
return filepath.Join(gcsBucket, filePath), err
|
|
}
|
|
|
|
// DeleteFromBucket will remove the content at path. path should be the full path
|
|
// to a file in GCS.
|
|
func DeleteFromBucket(path string) error {
|
|
cmd := exec.Command("gsutil", "rm", path)
|
|
_, err := RunCommandWithoutTest(cmd)
|
|
if err != nil {
|
|
return fmt.Errorf("Failed to delete file %s from GCS: %s", path, err)
|
|
}
|
|
return err
|
|
}
|