mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-10-01 01:36:09 -04:00
9477999be2
* Clean up Terraform pkg * Add note to Terraform migration functions expecting to be run on initialized workspace --------- Signed-off-by: Daniel Weiße <dw@edgeless.systems>
120 lines
3.8 KiB
Go
120 lines
3.8 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
package upgrade
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/edgelesssys/constellation/v2/cli/internal/terraform"
|
|
"github.com/edgelesssys/constellation/v2/internal/cloud/cloudprovider"
|
|
"github.com/edgelesssys/constellation/v2/internal/constants"
|
|
"github.com/edgelesssys/constellation/v2/internal/file"
|
|
"github.com/spf13/afero"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestIAMMigrate(t *testing.T) {
|
|
upgradeID := "test-upgrade"
|
|
upgradeDir := filepath.Join(constants.UpgradeDir, upgradeID, constants.TerraformIAMUpgradeWorkingDir)
|
|
fs, file := setupMemFSAndFileHandler(t, []string{"terraform.tfvars", "terraform.tfstate"}, []byte("OLD"))
|
|
// act
|
|
fakeTfClient := &tfClientStub{upgradeID, file}
|
|
sut := &IAMMigrateCmd{
|
|
tf: fakeTfClient,
|
|
upgradeID: upgradeID,
|
|
csp: cloudprovider.AWS,
|
|
logLevel: terraform.LogLevelDebug,
|
|
iamWorkspace: constants.TerraformIAMWorkingDir,
|
|
upgradeWorkspace: constants.UpgradeDir,
|
|
}
|
|
hasDiff, err := sut.Plan(context.Background(), file, bytes.NewBuffer(nil))
|
|
// assert
|
|
assert.NoError(t, err)
|
|
assert.False(t, hasDiff)
|
|
assertFileExists(fs, filepath.Join(upgradeDir, "terraform.tfvars"), t)
|
|
assertFileExists(fs, filepath.Join(upgradeDir, "terraform.tfstate"), t)
|
|
|
|
// act
|
|
err = sut.Apply(context.Background(), file)
|
|
assert.NoError(t, err)
|
|
// assert
|
|
assertFileReadsContent(file, filepath.Join(constants.TerraformIAMWorkingDir, "terraform.tfvars"), "NEW", t)
|
|
assertFileReadsContent(file, filepath.Join(constants.TerraformIAMWorkingDir, "terraform.tfstate"), "NEW", t)
|
|
assertFileDoesntExist(fs, filepath.Join(upgradeDir), t)
|
|
}
|
|
|
|
func assertFileReadsContent(file file.Handler, path string, expectedContent string, t *testing.T) {
|
|
bt, err := file.Read(path)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, expectedContent, string(bt))
|
|
}
|
|
|
|
func assertFileExists(fs afero.Fs, path string, t *testing.T) {
|
|
res, err := fs.Stat(path)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, res)
|
|
}
|
|
|
|
func assertFileDoesntExist(fs afero.Fs, path string, t *testing.T) {
|
|
res, err := fs.Stat(path)
|
|
assert.Error(t, err)
|
|
assert.Nil(t, res)
|
|
}
|
|
|
|
// setupMemFSAndFileHandler sets up a file handler with a memory file system and writes the given files with the given content.
|
|
func setupMemFSAndFileHandler(t *testing.T, files []string, content []byte) (afero.Fs, file.Handler) {
|
|
fs := afero.NewMemMapFs()
|
|
file := file.NewHandler(fs)
|
|
err := file.MkdirAll(constants.TerraformIAMWorkingDir)
|
|
require.NoError(t, err)
|
|
|
|
for _, f := range files {
|
|
err := file.Write(filepath.Join(constants.TerraformIAMWorkingDir, f), content)
|
|
require.NoError(t, err)
|
|
}
|
|
return fs, file
|
|
}
|
|
|
|
type tfClientStub struct {
|
|
upgradeID string
|
|
file file.Handler
|
|
}
|
|
|
|
func (t *tfClientStub) Plan(_ context.Context, _ terraform.LogLevel) (bool, error) {
|
|
return false, nil
|
|
}
|
|
|
|
func (t *tfClientStub) ShowPlan(_ context.Context, _ terraform.LogLevel, _ io.Writer) error {
|
|
return nil
|
|
}
|
|
|
|
func (t *tfClientStub) ApplyIAM(_ context.Context, _ cloudprovider.Provider, _ terraform.LogLevel) (terraform.IAMOutput, error) {
|
|
upgradeDir := filepath.Join(constants.UpgradeDir, t.upgradeID, constants.TerraformIAMUpgradeWorkingDir)
|
|
err := t.file.Remove(filepath.Join(upgradeDir, "terraform.tfvars"))
|
|
if err != nil {
|
|
return terraform.IAMOutput{}, err
|
|
}
|
|
err = t.file.Write(filepath.Join(upgradeDir, "terraform.tfvars"), []byte("NEW"))
|
|
if err != nil {
|
|
return terraform.IAMOutput{}, err
|
|
}
|
|
err = t.file.Remove(filepath.Join(upgradeDir, "terraform.tfstate"))
|
|
if err != nil {
|
|
return terraform.IAMOutput{}, err
|
|
}
|
|
err = t.file.Write(filepath.Join(upgradeDir, "terraform.tfstate"), []byte("NEW"))
|
|
if err != nil {
|
|
return terraform.IAMOutput{}, err
|
|
}
|
|
return terraform.IAMOutput{}, nil
|
|
}
|