oh-my-posh/src/engine/image_test.go

package engine

import (
	"io/ioutil"
	"oh-my-posh/color"
	"oh-my-posh/shell"
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
)

func runImageTest(content string) error {
	poshImagePath := "jandedobbeleer.png"
	file, err := ioutil.TempFile("", poshImagePath)
	if err != nil {
		return err
	}
	defer os.Remove(file.Name())
	ansi := &color.Ansi{}
	ansi.InitPlain(shell.PLAIN)
	image := &ImageRenderer{
		AnsiString: content,
		Ansi:       ansi,
	}
	image.Init("~/jandedobbeleer.omp.json")
	err = image.SavePNG()
	return err
}

func TestStringImageFileWithText(t *testing.T) {
	err := runImageTest("foobar")
	assert.NoError(t, err)
}

func TestStringImageFileWithANSI(t *testing.T) {
	prompt := ` jan  `
	err := runImageTest(prompt)
	assert.NoError(t, err)
}