custom_test.go

package widgets

import (
	"testing"

	"github.com/yuys13/agystatusline/types"
)

func TestWidgetInterfaces(t *testing.T) {
	RegisterAll()
	ctx := types.RenderContext{
		Data: types.StatusJSON{},
	}
	item := types.WidgetItem{
		Type: "model",
	}

	widgetsList := []string{
		"model", "git-branch", "git-changes", "quota", "custom-text",
		"sandbox", "agent-state", "context-bar", "quota-bar", "artifacts",
		"subagents", "tasks",
	}

	for _, name := range widgetsList {
		t.Run(name, func(t *testing.T) {
			w := GetWidget(name)
			if w == nil {
				t.Fatalf("Widget %q not registered", name)
			}
			if nameStr := w.GetDisplayName(); nameStr == "" {
				t.Errorf("GetDisplayName() returned empty for %q", name)
			}
			if defaultColor := w.GetDefaultColor(); defaultColor == "" {
				t.Errorf("GetDefaultColor() returned empty for %q", name)
			}
			if bodyColor := w.GetBodyColor(item, ctx); bodyColor == "" {
				t.Errorf("GetBodyColor() returned empty for %q", name)
			}
		})
	}
}