termui/utils_test.go

// Copyright 2017 Zack Guo <zack.y.guo@gmail.com>. All rights reserved.
// Use of this source code is governed by a MIT license that can
// be found in the LICENSE file.

package termui

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestStr2Rune(t *testing.T) {
	s := "你好,世界."
	rs := str2runes(s)
	if len(rs) != 6 {
		t.Error(t)
	}
}

func TestWidth(t *testing.T) {
	s0 := "つのだ☆HIRO"
	s1 := "11111111111"
	// above not align for setting East Asian Ambiguous to wide!!

	if strWidth(s0) != strWidth(s1) {
		t.Error("str len failed")
	}

	len1 := []rune{'a', '2', '&', '｢', 'ｵ', '｡'} //will false: 'ᆵ', 'ᄚ', 'ᄒ'
	for _, v := range len1 {
		if charWidth(v) != 1 {
			t.Error("len1 failed")
		}
	}

	len2 := []rune{'漢', '字', '한', '자', '你', '好', 'だ', '。', '％', 'ｓ', 'Ｅ', 'ョ', '、', 'ヲ'}
	for _, v := range len2 {
		if charWidth(v) != 2 {
			t.Error("len2 failed")
		}
	}
}

func TestTrim(t *testing.T) {
	s := "つのだ☆HIRO"
	if string(trimStr2Runes(s, 10)) != "つのだ☆HI"+dot {
		t.Error("trim failed")
	}
	if string(trimStr2Runes(s, 11)) != "つのだ☆HIRO" {
		t.Error("avoid tail trim failed")
	}
	if string(trimStr2Runes(s, 15)) != "つのだ☆HIRO" {
		t.Error("avoid trim failed")
	}
}

func TestTrimStrIfAppropriate_NoTrim(t *testing.T) {
	assert.Equal(t, "hello", TrimStrIfAppropriate("hello", 5))
}

func TestTrimStrIfAppropriate(t *testing.T) {
	assert.Equal(t, "hel…", TrimStrIfAppropriate("hello", 4))
	assert.Equal(t, "h…", TrimStrIfAppropriate("hello", 2))
}

func TestStringToAttribute(t *testing.T) {
	assert.Equal(t, ColorRed, StringToAttribute("ReD"))
	assert.Equal(t, ColorRed|AttrBold, StringToAttribute("RED, bold"))
}