Function TestS3RunExitOnError

upload/s3_test.go:309–353

(t *testing.T)

}

func TestS3RunExitOnError(t *testing.T) {
	defer testutil.DisableLogging()()

	tmpDir, fnames := prepareUploadS3TestFolder(t, 1)
	fname := fnames[0]

	stagingDir := t.TempDir()

	cfg := baker.UploadParams{
		ComponentParams: baker.ComponentParams{
			DecodedConfig: &S3Config{
				SourceBasePath: stagingDir,
				StagingPath:    tmpDir,
				Bucket:         "my-bucket",
				ExitOnError:    true,
			},
		},
	}
	iu, err := NewS3(cfg)
	if err != nil {
		t.Fatalf("NewS3Upload(%+v) = %q", cfg, err)
	}
	s, _, _ := mockS3Service(true)
	u := iu.(*S3)
	u.uploader = s3manager.NewUploaderWithClient(s)

	ch := make(chan string)
	go func() {
		ch <- fname
		close(ch)
	}()

	err = u.Run(ch)
	if err != nil {
		t.Fatal(err)
	}

	if int(u.totalerr) != 1 {
		t.Fatalf("totalerr: want: %d, got: %d", 1, int(u.totalerr))
	}

	if int(u.totaln) != 1 {
		t.Fatalf("totaln: want: %d, got: %d", 1, int(u.totaln))
	}
}

func TestS3RunNotExitOnError(t *testing.T) {
	defer testutil.DisableLogging()()