root_test.go

//go:build !docker

package main

import (
	"os"
	"testing"

	"github.com/jiro4989/textimg/v3/config"
	"github.com/stretchr/testify/assert"
)

func TestRunRootCommand(t *testing.T) {
	b, _ := os.ReadFile(inDir + "/red_grad.txt")
	grad := string(b)
	b, _ = os.ReadFile(inDir + "/255.txt")
	c255 := string(b)

	tests := []struct {
		desc       string
		c          config.Config
		args       []string
		envs       config.EnvVars
		wantErr    bool
		existsFile string
	}{
		{
			desc: "正常系: PrintEnvironmentsが設定されていると環境変数を出力して終了",
			c: config.Config{
				PrintEnvironments: true,
			},
			args:    []string{"hello"},
			envs:    config.EnvVars{},
			wantErr: false,
		},
		{
			desc: "正常系: 正常系がパスする。出力先はモックWriterなのでファイルは生成されない",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = "t.png"
				return c
			}(),
			args:    []string{"hello"},
			envs:    config.EnvVars{},
			wantErr: false,
		},
		{
			desc: "異常系: Writerがエラーを返す",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = "t.png"
				c.Writer = config.NewMockWriter(true, false)
				return c
			}(),
			args:    []string{"hello"},
			envs:    config.EnvVars{},
			wantErr: true,
		},
		// 旧 main_test.go を移行してきたもの
		{
			desc: "正常系: 画像ファイルに出力する",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_font_is_red_and_background_is_black.png"
				c.Writer = nil
				return c
			}(),
			args:       []string{"1234\x1b[31mred\x1b[m5678\nabcd\x1b[32mgreen\x1b[0mefgh\nあい\x1b[33mう\x1b[mえお"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_font_is_red_and_background_is_black.png",
		},
		{
			desc: "正常系: Foregroundのみもとにもどす、Backgroundのみもとに戻す",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_foreground_default_background_default.png"
				c.Writer = nil
				return c
			}(),
			args:       []string{"\x1b[31;42mRedGreen\x1b[39mRedGreen\x1b[49mRedGreen"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_foreground_default_background_default.png",
		},
		{
			desc: "正常系: 256色を使う",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_color_256.png"
				c.Writer = nil
				return c
			}(),
			args:       []string{c255},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_color_256.png",
		},
		{
			desc: "正常系: RGB色を使う",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_color_rgb.png"
				c.Writer = nil
				return c
			}(),
			args:       []string{grad},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_color_rgb.png",
		},
		{
			desc: "正常系: JPEGで出力する",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_jpeg.jpeg"
				c.Writer = nil
				return c
			}(),
			args:       []string{"jpeg"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_jpeg.jpeg",
		},
		{
			desc: "正常系: GIFで出力する",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_gif.gif"
				c.Writer = nil
				return c
			}(),
			args:       []string{"gif"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_gif.gif",
		},
		{
			desc: "正常系: 日本語と絵文字を描画する（ただし豆腐になる）。このテストはDockerの方で実施する",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_tofu.png"
				c.Writer = nil
				return c
			}(),
			args:       []string{"あいうえお👍"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_tofu.png",
		},
		{
			desc: "正常系: 前景色と背景色を反転する",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_reverse.png"
				c.Writer = nil
				return c
			}(),
			args:       []string{"\x1b[31;42mRED\x1b[7m\nGREEN\x1b[0m"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_reverse.png",
		},
		{
			desc: "正常系: 文字色と背景色を変更する",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_font_is_green_and_background_is_blue.png"
				c.Writer = nil
				c.Foreground = "green"
				c.Background = "blue"
				return c
			}(),
			args:       []string{"green"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_font_is_green_and_background_is_blue.png",
		},
		{
			desc: "正常系: カンマ区切りで指定",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_font_is_blue_and_background_is_red.png"
				c.Writer = nil
				c.Foreground = "0,0,255,255"
				c.Background = "255,0,0,255"
				return c
			}(),
			args:       []string{"blue"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_font_is_blue_and_background_is_red.png",
		},
		{
			desc: "正常系: Slackアイコンサイズで生成する",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_font_is_blue_and_background_is_red_slack_icon_size.png"
				c.Writer = nil
				c.Foreground = "0,0,255,255"
				c.Background = "255,0,0,255"
				c.ToSlackIcon = true
				return c
			}(),
			args:       []string{"slack"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_font_is_blue_and_background_is_red_slack_icon_size.png",
		},
		{
			desc: "正常系: 明示的に幅を指定できる",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_font_is_blue_and_background_is_red_100x200.png"
				c.Writer = nil
				c.Foreground = "0,0,255,255"
				c.Background = "255,0,0,255"
				c.ResizeWidth = 100
				c.ResizeHeight = 200
				return c
			}(),
			args:       []string{"100x200"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_font_is_blue_and_background_is_red_100x200.png",
		},
		{
			desc: "正常系: Widthのみを指定した場合はHeightが調整される",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_font_is_blue_and_background_is_red_100w.png"
				c.Writer = nil
				c.Foreground = "0,0,255,255"
				c.Background = "255,0,0,255"
				c.ResizeWidth = 100
				return c
			}(),
			args:       []string{"100w"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_font_is_blue_and_background_is_red_100w.png",
		},
		{
			desc: "正常系: Heightのみを指定した場合はWidthが調整される",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_font_is_blue_and_background_is_red_100h.png"
				c.Writer = nil
				c.Foreground = "0,0,255,255"
				c.Background = "255,0,0,255"
				c.ResizeHeight = 100
				return c
			}(),
			args:       []string{"100h"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_font_is_blue_and_background_is_red_100h.png",
		},
		{
			desc: "正常系: 1行のアニメを生成できる",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_animation_1_line.gif"
				c.Writer = nil
				c.UseAnimation = true
				c.LineCount = 1
				return c
			}(),
			args:       []string{"\x1b[31m1\n\x1b[32m2\n\x1b[33m3\n\x1b[34m4"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_animation_1_line.gif",
		},
		{
			desc: "正常系: 2行のアニメを生成できる",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_animation_2_line.gif"
				c.Writer = nil
				c.UseAnimation = true
				c.LineCount = 2
				return c
			}(),
			args:       []string{"\x1b[31m1\n\x1b[32m2\n\x1b[33m3\n\x1b[34m4"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_animation_2_line.gif",
		},
		{
			desc: "正常系: 4行のアニメを生成できる",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_animation_4_line.gif"
				c.Writer = nil
				c.UseAnimation = true
				c.LineCount = 4
				return c
			}(),
			args:       []string{"\x1b[31m1\n\x1b[32m2\n\x1b[33m3\n\x1b[34m4\n\x1b[31m5\n\x1b[32m6\n\x1b[33m7\n\x1b[34m8"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_animation_4_line.gif",
		},
		{
			desc: "正常系: 8行のアニメを生成できる",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_animation_8_line.gif"
				c.Writer = nil
				c.UseAnimation = true
				c.LineCount = 8
				return c
			}(),
			args:       []string{"\x1b[31m1\n\x1b[32m2\n\x1b[33m3\n\x1b[34m4\n\x1b[31m5\n\x1b[32m6\n\x1b[33m7\n\x1b[34m8"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_animation_8_line.gif",
		},
		{
			desc: "正常系: 4行のアニメを2行ずつスライドする",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_animation_4_line_slide_2_forever.gif"
				c.Writer = nil
				c.UseAnimation = true
				c.LineCount = 4
				c.UseSlideAnimation = true
				c.SlideWidth = 2
				c.SlideForever = true
				c.Delay = 100
				return c
			}(),
			args:       []string{"\x1b[31m1\n\x1b[32m2\n\x1b[33m3\n\x1b[34m4\n\x1b[31m5\n\x1b[32m6\n\x1b[33m7\n\x1b[34m8"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_animation_4_line_slide_2_forever.gif",
		},
		{
			desc: "正常系: 4行のアニメを3行ずつスライドする",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_animation_4_line_slide_3_forever.gif"
				c.Writer = nil
				c.UseAnimation = true
				c.LineCount = 4
				c.UseSlideAnimation = true
				c.SlideWidth = 3
				c.SlideForever = true
				c.Delay = 100
				return c
			}(),
			args:       []string{"\x1b[31m1\n\x1b[32m2\n\x1b[33m3\n\x1b[34m4\n\x1b[31m5\n\x1b[32m6\n\x1b[33m7\n\x1b[34m8"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_animation_4_line_slide_3_forever.gif",
		},
		{
			desc: "正常系: SlackアイコンサイズでアニメーションGIFを生成できる",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_slack_icon_size_animation.gif"
				c.Writer = nil
				c.ToSlackIcon = true
				c.UseAnimation = true
				return c
			}(),
			args:       []string{"1\n2\n3\n4"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_slack_icon_size_animation.gif",
		},
		{
			desc: "正常系: すでに同名のファイルが存在する時、別名で保存される",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_numbering.png"
				c.Writer = nil
				c.SaveNumberedFile = true
				return c
			}(),
			args:       []string{"number"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_numbering.png",
		},
		{
			desc: "正常系: すでに同名のファイルが存在する時、別名で保存される_2",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_numbering.png"
				c.Writer = nil
				c.SaveNumberedFile = true
				return c
			}(),
			args:       []string{"number"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_numbering_2.png",
		},
		{
			desc: "正常系: すでに同名のファイルが存在する時、別名で保存される_3",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_numbering.png"
				c.Writer = nil
				c.SaveNumberedFile = true
				return c
			}(),
			args:       []string{"number"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_numbering_3.png",
		},
		{
			desc: "正常系: フォントインデックスを指定できる",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_index.png"
				c.Writer = nil
				c.FontIndex = 0
				c.EmojiFontIndex = 0
				return c
			}(),
			args:       []string{"index"},
			envs:       config.EnvVars{},
			wantErr:    false,
			existsFile: outDir + "/root_test_index.png",
		},
		{
			desc: "異常系: 空文字列は不正",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_empty_string.png"
				c.Writer = nil
				return c
			}(),
			args:    []string{""},
			envs:    config.EnvVars{},
			wantErr: true,
		},
		{
			desc: "異常系: 改行文字のみは不正",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_only_line.png"
				c.Writer = nil
				return c
			}(),
			args:    []string{"\n\n\n"},
			envs:    config.EnvVars{},
			wantErr: true,
		},
		{
			desc: "異常系: 色文字列が不正",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_numbering.png"
				c.Writer = nil
				c.Foreground = "ggg"
				return c
			}(),
			args:    []string{"ggg"},
			envs:    config.EnvVars{},
			wantErr: true,
		},
		{
			desc: "異常系: 背景色が不正",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_numbering.png"
				c.Writer = nil
				c.Background = "ggg"
				return c
			}(),
			args:    []string{"ggg"},
			envs:    config.EnvVars{},
			wantErr: true,
		},
		{
			desc: "異常系: 不正なフォント指定",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_numbering.png"
				c.Writer = nil
				c.FontFile = inDir + "/illegal_font.ttc"
				return c
			}(),
			args:    []string{"ggg"},
			envs:    config.EnvVars{},
			wantErr: true,
		},
		{
			desc: "異常系: 不正な絵文字フォント指定",
			c: func() config.Config {
				c := newDefaultConfig()
				c.Outpath = outDir + "/root_test_numbering.png"
				c.Writer = nil
				c.EmojiFontFile = inDir + "/illegal_font.ttc"
				return c
			}(),
			args:    []string{"ggg"},
			envs:    config.EnvVars{},
			wantErr: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.desc, func(t *testing.T) {
			assert := assert.New(t)

			err := RunRootCommand(tt.c, tt.args, tt.envs)
			if tt.wantErr {
				assert.Error(err)
				return
			}
			assert.NoError(err)
			if tt.existsFile != "" {
				_, err := os.Stat(tt.existsFile)
				got := os.IsNotExist(err)
				assert.False(got)
			}
		})
	}
}

func TestComplementWidthHeight(t *testing.T) {
	type TestData struct {
		desc       string
		x, y, w, h int
		wantWidth  int
		wantHeight int
	}
	tds := []TestData{
		{
			desc:       "正常系: wが0のときはwidthが調整される",
			x:          200,
			y:          100,
			w:          0,
			h:          200,
			wantWidth:  400,
			wantHeight: 200,
		},
		{
			desc:       "正常系: hが0のときはheightが調整される",
			x:          200,
			y:          100,
			w:          100,
			h:          0,
			wantWidth:  100,
			wantHeight: 50,
		},
		{
			desc:       "正常系: hが0のときはheightが調整される",
			x:          200,
			y:          100,
			w:          100,
			h:          0,
			wantWidth:  100,
			wantHeight: 50,
		},
		{
			desc:       "正常系: wとhが0出ないときはwとhが返る",
			x:          200,
			y:          100,
			w:          400,
			h:          300,
			wantWidth:  400,
			wantHeight: 300,
		},
	}
	for _, v := range tds {
		t.Run(v.desc, func(t *testing.T) {
			a := assert.New(t)
			w, h := complementWidthHeight(v.x, v.y, v.w, v.h)
			a.Equal(v.wantWidth, w)
			a.Equal(v.wantHeight, h)
		})
	}
}