driftctl/pkg/cmd/scan/output/console_test.go

115 lines
2.8 KiB
Go
Raw Normal View History

package output
import (
"bytes"
"io"
"io/ioutil"
"os"
"path"
"testing"
"github.com/stretchr/testify/assert"
"github.com/cloudskiff/driftctl/test/goldenfile"
"github.com/cloudskiff/driftctl/pkg/analyser"
)
func TestConsole_Write(t *testing.T) {
type args struct {
analysis *analyser.Analysis
}
tests := []struct {
name string
goldenfile string
args args
wantErr bool
}{
{
name: "test console output",
goldenfile: "output.txt",
args: args{analysis: fakeAnalysis()},
wantErr: false,
},
{
name: "test console output no drift",
goldenfile: "output_no_drift.txt",
args: args{analysis: fakeAnalysisNoDrift()},
wantErr: false,
},
{
name: "test console output with json fields",
goldenfile: "output_json_fields.txt",
args: args{analysis: fakeAnalysisWithJsonFields()},
wantErr: false,
},
{
name: "test console output with resources which implement stringer",
goldenfile: "output_stringer_resources.txt",
args: args{analysis: fakeAnalysisWithStringerResources()},
wantErr: false,
2020-12-16 12:02:02 +00:00
},
{
name: "test console output with drift on computed fields",
goldenfile: "output_computed_fields.txt",
args: args{analysis: fakeAnalysisWithComputedFields()},
wantErr: false,
},
{
name: "test console output with AWS enumeration alerts",
goldenfile: "output_access_denied_alert_aws.txt",
args: args{analysis: fakeAnalysisWithAWSEnumerationError()},
wantErr: false,
},
{
name: "test console output with Github enumeration alerts",
goldenfile: "output_access_denied_alert_github.txt",
args: args{analysis: fakeAnalysisWithGithubEnumerationError()},
wantErr: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
c := NewConsole()
stdout := os.Stdout // keep backup of the real stdout
stderr := os.Stderr // keep backup of the real stderr
r, w, _ := os.Pipe()
os.Stdout = w
os.Stderr = w
if err := c.Write(tt.args.analysis); (err != nil) != tt.wantErr {
t.Errorf("Write() error = %v, wantErr %v", err, tt.wantErr)
}
outC := make(chan []byte)
// copy the output in a separate goroutine so printing can't block indefinitely
go func() {
var buf bytes.Buffer
_, _ = io.Copy(&buf, r)
outC <- buf.Bytes()
}()
// back to normal state
w.Close()
os.Stdout = stdout // restoring the real stdout
os.Stderr = stderr
out := <-outC
expectedFilePath := path.Join("./testdata", tt.goldenfile)
if *goldenfile.Update == tt.goldenfile {
if err := ioutil.WriteFile(expectedFilePath, out, 0600); err != nil {
t.Fatal(err)
}
}
expected, err := ioutil.ReadFile(expectedFilePath)
if err != nil {
t.Fatal(err)
}
assert.Equal(t, string(expected), string(out))
})
}
}