package app

import (
	"os"
	"testing"

	"github.com/helmfile/helmfile/pkg/testutil"
)

// TestFormatAsTable tests the FormatAsTable function.
func TestFormatAsTable(t *testing.T) {
	h := []*HelmRelease{
		{
			Name:      "test",
			Namespace: "test",
			Enabled:   true,
			Installed: true,
			Labels:    "test",
			Chart:     "test",
			Version:   "test",
		},
		{
			Name:      "test1",
			Namespace: "test2",
			Enabled:   false,
			Installed: false,
			Labels:    "test1",
			Chart:     "test1",
			Version:   "test1",
		},
	}

	tableoutput := "testdata/formatters/tableoutput"
	expectd, err := os.ReadFile(tableoutput)
	if err != nil {
		t.Errorf("error reading %s: %v", tableoutput, err)
	}

	result := testutil.CaptureStdout(func() {
		FormatAsTable(h)
	})
	if result != string(expectd) {
		t.Errorf("FormatAsTable() = %v, want %v", result, string(expectd))
	}
}

func TestFormatAsJson(t *testing.T) {
	h := []*HelmRelease{
		{
			Name:      "test",
			Namespace: "test",
			Enabled:   true,
			Installed: true,
			Labels:    "test",
			Chart:     "test",
			Version:   "test",
		},
		{
			Name:      "test1",
			Namespace: "test2",
			Enabled:   false,
			Installed: false,
			Labels:    "test1",
			Chart:     "test1",
			Version:   "test1",
		},
	}
	output := "testdata/formatters/jsonoutput"
	expectd, err := os.ReadFile(output)
	if err != nil {
		t.Errorf("error reading %s: %v", output, err)
	}
	result := testutil.CaptureStdout(func() {
		FormatAsJson(h)
	})

	if result != string(expectd) {
		t.Errorf("FormatAsJson() = %v, want %v", result, string(expectd))
	}
}