package file

import (
	"github.com/jesseduffield/lazygit/pkg/config"
	. "github.com/jesseduffield/lazygit/pkg/integration/components"
)

var CollapseExpand = NewIntegrationTest(NewIntegrationTestArgs{
	Description:  "Collapsing and expanding all files in the file tree",
	ExtraCmdArgs: []string{},
	Skip:         false,
	SetupConfig: func(config *config.AppConfig) {
	},
	SetupRepo: func(shell *Shell) {
		shell.CreateDir("dir")
		shell.CreateFile("dir/file-one", "original content\n")
		shell.CreateDir("dir2")
		shell.CreateFile("dir2/file-two", "original content\n")
	},
	Run: func(t *TestDriver, keys config.KeybindingConfig) {
		t.Views().Files().
			IsFocused().
			Lines(
				Contains("dir").IsSelected(),
				Contains("??").Contains("file-one"),
				Contains("dir2"),
				Contains("??").Contains("file-two"),
			)

		t.Views().Files().
			Press(keys.Files.CollapseAll).
			Lines(
				Contains("dir"),
				Contains("dir2"),
			)

		t.Views().Files().
			Press(keys.Files.ExpandAll).
			Lines(
				Contains("dir").IsSelected(),
				Contains("??").Contains("file-one"),
				Contains("dir2"),
				Contains("??").Contains("file-two"),
			)
	},
})