standard/qushi/gochart/util/utils_test.go

package charter


import (
	"testing"
	"time"

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

func TestCreateLineChart(t *testing.T) {
	testAssert := assert.New(t)

	//tests := []struct {
	//	title     string
	//	endTime   time.Time
	//	barValues []LineYValue
	//}{
	//	{"line chart", time.Now(), []LineYValue{
	//		{"asd", []float64{1, 2, 300, 100, 200, 6, 700}},
	//		{"hgj", []float64{400, 500000, 200, 50, 5, 800, 7}},
	//		{"dfg45r", []float64{1, 2, 700, 100, 200, 6, 700}},
	//		{"2342sr", []float64{400, 500000, 200, 50, 5, 800, 7}},
	//		{"das21-asd", []float64{300000, 200000, 400000, 100000, 400000, 450000, 400000}},
	//		{"csc", []float64{400, 500000, 200, 50, 5, 800, 7}},
	//		{"mhj", []float64{1, 2, 300, 100, 200, 6, 700}},
	//		{"876ijgh", []float64{400, 500000, 200, 50, 5, 800, 7}},
	//		{"fbfdv", []float64{1, 2, 300, 100, 200, 6, 700}},
	//		{"67ds", []float64{400, 10000, 200, 50, 5, 800, 7}},
	//		{"67bdfv", []float64{1, 2, 300, 100, 200, 6, 700}},
	//		{"sdf324", []float64{400, 500000, 200, 50, 5, 800, 7}},
	//		{"vdf67", []float64{1, 2, 300, 100, 200, 6, 700}},
	//		{"vdfs234", []float64{400, 500000, 200, 50, 5, 800, 7}},
	//		{"123sdf", []float64{1, 2, 700, 100, 200, 6, 700}},
	//		{"aasdasd", []float64{400, 500000, 200, 50, 5, 800, 7}},
	//		{"aasd", []float64{1, 2, 300, 100, 200, 6, 700}},
	//		{"basd", []float64{400, 500000, 200, 50, 5, 800, 7}},
	//		{"cczx", []float64{1, 2, 300, 100, 200, 6, 700}},
	//		{"qweqw", []float64{400, 500000, 200, 50, 5, 800, 7}},
	//		{"asdadf", []float64{1, 2, 300, 100, 200, 6, 700}},
	//		{"fghfh", []float64{400, 500000, 200, 50, 5, 800, 7}},
	//		{"erttyrt", []float64{1, 2, 300, 100, 200, 6, 700}}}},
	//}
	tests := []struct {
			title     string
			endTime   time.Time
			barValues []LineYValue}{
			{"业务费/能量值消费趋势", time.Now(), []LineYValue{
				{"业务费", []float64{1, 2, 300, 100, 200, 6, 700}},
				{"能量值", []float64{1, 2, 700, 100, 200, 6, 700}}}},
		}

	for _, test := range tests {
		img,err := createLineChart(test.title, test.endTime, test.barValues)
		testAssert.Equal(img, "")
		testAssert.Equal(err, nil)
	}
}

func TestCreatePieChart(t *testing.T) {
	testAssert := assert.New(t)

	tests := []struct {
		title     string
		pieValues []ChartValue
	}{
		{"pie chart", []ChartValue{{"asdas", 20000}, {"q12asd", 300000}, {"ascasd", 3000}}},
	}

	for _, test := range tests {
		img, err := createPieChart(test.title, test.pieValues)
		testAssert.Equal(img, "")
		testAssert.Equal(err, nil)
	}
}

func TestCreateBarChart(t *testing.T) {
	testAssert := assert.New(t)

	tests := []struct {
		title     string
		pieValues []ChartValue
	}{
		{"bar chart", []ChartValue{{"asdascasd\nasd-asd", 20}, {"asdascascasdasdasd.go\nasdasd-asdasd", 30}, {"asasdasd.asdasd]\nasdasd-asda", 100},
			{"asdasdasda.go\nasdasd-asdasd", 20}, {"asdasd.asdasd\ngeass", 30}, {"asdasdasd\nasdasd-asdasd", 100},
			{"asdasd_adsdasd_dasd\asd-asd", 20}, {"asdascasdcad\nasdasdasda", 30}, {"asdasdasdasd", 100},
			{"asdasclkhy9p867p9", 20}}},
	}

	for _, test := range tests {
		img, err := createBarChart(test.title, test.pieValues)
		testAssert.Equal(img, "")
		testAssert.Equal(err, nil)
	}
}