spinner_test.go

package tvxwidgets_test

import (
	"github.com/gdamore/tcell/v2"
	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
	"github.com/rivo/tview"

	"github.com/navidys/tvxwidgets"
)

var _ = Describe("Spinner", Ordered, func() {
	var (
		app       *tview.Application
		headerBox *tview.Box
		spinner   *tvxwidgets.Spinner
		screen    tcell.SimulationScreen
	)

	BeforeAll(func() {
		app = tview.NewApplication()
		headerBox = tview.NewBox().SetBorder(true)
		spinner = tvxwidgets.NewSpinner()
		screen = tcell.NewSimulationScreen("UTF-8")

		if err := screen.Init(); err != nil {
			panic(err)
		}

		go func() {
			appLayout := tview.NewFlex().SetDirection(tview.FlexRow)
			appLayout.AddItem(headerBox, 1, 0, true)
			appLayout.AddItem(spinner, 50, 0, true)
			err := app.SetScreen(screen).SetRoot(appLayout, true).Run()
			if err != nil {
				panic(err)
			}
		}()
	})

	AfterAll(func() {
		app.Stop()
	})

	Describe("Focus", func() {
		It("checks primitivie focus", func() {
			app.SetFocus(headerBox)
			app.Draw()
			Expect(spinner.HasFocus()).To(Equal(false))

			app.SetFocus(spinner)
			app.Draw()
			Expect(spinner.HasFocus()).To(Equal(true))
		})
	})

	Describe("GetRect", func() {
		It("primitivie size", func() {
			x, y, width, heigth := spinner.GetRect()
			Expect(x).To(Equal(0))
			Expect(y).To(Equal(1))
			Expect(width).To(Equal(80))
			Expect(heigth).To(Equal(50))
		})
	})

	Describe("Style", func() {
		It("checks style", func() {
			spinner.SetStyle(tvxwidgets.SpinnerGrowHorizontal)
			spinner.Reset()
			app.Draw()

			prune, _, _, _ := screen.GetContent(0, 1)
			Expect(prune).To(Equal('▉'))

			spinner.Pulse()
			app.Draw()
			prune, _, _, _ = screen.GetContent(0, 1)
			Expect(prune).To(Equal('▊'))
		})
	})

	Describe("CustomStyle", func() {
		It("checks custom style", func() {
			customStyle := []rune{'\u2705', '\u274C'}
			spinner.SetCustomStyle(customStyle)
			spinner.Reset()

			app.Draw()
			prune, _, _, _ := screen.GetContent(0, 1)
			Expect(prune).To(Equal(customStyle[0]))

			spinner.Pulse()
			app.Draw()
			prune, _, _, _ = screen.GetContent(0, 1)
			Expect(prune).To(Equal(customStyle[1]))

			spinner.Pulse()
			app.Draw()
			prune, _, _, _ = screen.GetContent(0, 1)
			Expect(prune).To(Equal(customStyle[0]))
		})
	})
})