mirror of
https://github.com/hashicorp/packer.git
synced 2026-09-20 23:11:40 -04:00
hashicorp-copywrite[bot]
19055df3ec
* Updating the license from MPL to Business Source License Going forward, this project will be licensed under the Business Source License v1.1. Please see our blog post for more details at https://hashi.co/bsl-blog, FAQ at https://hashi.co/license-faq, and details of the license at www.hashicorp.com/bsl. * Update copyright file headers to BUSL-1.1 --------- Co-authored-by: hashicorp-copywrite[bot] <110428419+hashicorp-copywrite[bot]@users.noreply.github.com>
112 lines
2.8 KiB
Go
112 lines
2.8 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package checksum
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
packersdk "github.com/hashicorp/packer-plugin-sdk/packer"
|
|
"github.com/hashicorp/packer-plugin-sdk/template"
|
|
"github.com/hashicorp/packer/builder/file"
|
|
)
|
|
|
|
const expectedFileContents = "Hello world!"
|
|
|
|
func TestChecksumSHA1(t *testing.T) {
|
|
const config = `
|
|
{
|
|
"post-processors": [
|
|
{
|
|
"type": "checksum",
|
|
"checksum_types": ["sha1"],
|
|
"output": "sha1sums"
|
|
}
|
|
]
|
|
}
|
|
`
|
|
artifact := testChecksum(t, config)
|
|
defer artifact.Destroy()
|
|
|
|
f, err := os.Open("sha1sums")
|
|
if err != nil {
|
|
t.Errorf("Unable to read checksum file: %s", err)
|
|
}
|
|
if buf, _ := ioutil.ReadAll(f); !bytes.Equal(buf, []byte("d3486ae9136e7856bc42212385ea797094475802\tpackage.txt\n")) {
|
|
t.Errorf("Failed to compute checksum: %s\n%s", buf, "d3486ae9136e7856bc42212385ea797094475802 package.txt")
|
|
}
|
|
|
|
defer f.Close()
|
|
}
|
|
|
|
// Test Helpers
|
|
|
|
func setup(t *testing.T) (packersdk.Ui, packersdk.Artifact, error) {
|
|
// Create fake UI and Cache
|
|
ui := packersdk.TestUi(t)
|
|
|
|
// Create config for file builder
|
|
const fileConfig = `{"builders":[{"type":"file","target":"package.txt","content":"Hello world!"}]}`
|
|
tpl, err := template.Parse(strings.NewReader(fileConfig))
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("Unable to parse setup configuration: %s", err)
|
|
}
|
|
|
|
// Prepare the file builder
|
|
builder := file.Builder{}
|
|
_, warnings, err := builder.Prepare(tpl.Builders["file"].Config)
|
|
if len(warnings) > 0 {
|
|
for _, warn := range warnings {
|
|
return nil, nil, fmt.Errorf("Configuration warning: %s", warn)
|
|
}
|
|
}
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("Invalid configuration: %s", err)
|
|
}
|
|
|
|
// Run the file builder
|
|
artifact, err := builder.Run(context.Background(), ui, nil)
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("Failed to build artifact: %s", err)
|
|
}
|
|
|
|
return ui, artifact, err
|
|
}
|
|
|
|
func testChecksum(t *testing.T, config string) packersdk.Artifact {
|
|
ui, artifact, err := setup(t)
|
|
if err != nil {
|
|
t.Fatalf("Error bootstrapping test: %s", err)
|
|
}
|
|
if artifact != nil {
|
|
defer artifact.Destroy()
|
|
}
|
|
|
|
tpl, err := template.Parse(strings.NewReader(config))
|
|
if err != nil {
|
|
t.Fatalf("Unable to parse test config: %s", err)
|
|
}
|
|
|
|
checksum := PostProcessor{}
|
|
checksum.Configure(tpl.PostProcessors[0][0].Config)
|
|
|
|
// I get the feeling these should be automatically available somewhere, but
|
|
// some of the post-processors construct this manually.
|
|
checksum.config.ctx.BuildName = "chocolate"
|
|
checksum.config.PackerBuildName = "vanilla"
|
|
checksum.config.PackerBuilderType = "file"
|
|
|
|
artifactOut, _, _, err := checksum.PostProcess(context.Background(), ui, artifact)
|
|
if err != nil {
|
|
t.Fatalf("Failed to checksum artifact: %s", err)
|
|
}
|
|
|
|
return artifactOut
|
|
}
|