git-spice/log_graph_test.go at main · abhinav/git-spice · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package main

import (
	"bytes"
	"testing"

	"github.com/charmbracelet/colorprofile"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"go.abhg.dev/gs/internal/handler/list"
	"go.abhg.dev/gs/internal/ui"
)

func TestGraphLogPresenter_Present(t *testing.T) {
	var buf bytes.Buffer
	stderrView := ui.NewFileView(&buf)
	presenter := &graphLogPresenter{
		Stderr:            stderrView,
		Theme:             stderrView.Theme(),
		ChangeFormat:      changeFormatID,
		ShowCRStatus:      false,
		ShowCommentCounts: false,
		PushStatusFormat:  pushStatusDisabled,
		CurrentWorktree:   "/repo",
	}

	res := &list.BranchesResponse{
		Branches: []*list.BranchItem{
			{Name: "main", Aboves: []int{1}},
			{
				Name: "feature",
				Base: "main",
			},
		},
		TrunkIdx: 0,
	}

	err := presenter.Present(res, "feature")
	require.NoError(t, err)

	assert.Equal(t, "┏━■ feature ◀\nmain\n", buf.String())
}

func TestGraphLogPresenter_Present_preservesColor(t *testing.T) {
	var buf bytes.Buffer
	stderrView := ui.NewFileView(&colorprofile.Writer{
		Forward: &buf,
		Profile: colorprofile.TrueColor,
	})
	presenter := &graphLogPresenter{
		Stderr:            stderrView,
		Theme:             stderrView.Theme(),
		ChangeFormat:      changeFormatID,
		ShowCRStatus:      false,
		ShowCommentCounts: false,
		PushStatusFormat:  pushStatusDisabled,
		CurrentWorktree:   "/repo",
	}

	res := &list.BranchesResponse{
		Branches: []*list.BranchItem{
			{Name: "main", Aboves: []int{1}},
			{
				Name: "feature",
				Base: "main",
			},
		},
		TrunkIdx: 0,
	}

	err := presenter.Present(res, "feature")
	require.NoError(t, err)

	assert.Contains(t, buf.String(), "\x1b[")
}