From ae5ed9ee86e73203a8ea7e5f1cf191b144da1b9f Mon Sep 17 00:00:00 2001
From: Bill Keenan <bill@bigmojo.net>
Date: Wed, 27 Jun 2018 14:15:16 -0400
Subject: [PATCH] added bargraph unit test

---
 wtf/bargraph.go                     | 25 ++++++++++++++--------
 wtf_tests/bargraph/bargraph_test.go | 33 +++++++++++++++++++++++++++++
 2 files changed, 49 insertions(+), 9 deletions(-)
 create mode 100644 wtf_tests/bargraph/bargraph_test.go

diff --git a/wtf/bargraph.go b/wtf/bargraph.go
index 977f526f..7a1a5e9e 100644
--- a/wtf/bargraph.go
+++ b/wtf/bargraph.go
@@ -11,9 +11,10 @@ import (
 
 //BarGraph lets make graphs
 type BarGraph struct {
-	enabled   bool
-	focusable bool
-
+	enabled     bool
+	focusable   bool
+	starChar    string
+	maxStars    int
 	Name        string
 	RefreshedAt time.Time
 	RefreshInt  int
@@ -27,9 +28,10 @@ type BarGraph struct {
 // NewBarGraph initialize your fancy new graph
 func NewBarGraph(name string, configKey string, focusable bool) BarGraph {
 	widget := BarGraph{
-		enabled:   Config.UBool(fmt.Sprintf("wtf.mods.%s.enabled", configKey), false),
-		focusable: focusable,
-
+		enabled:    Config.UBool(fmt.Sprintf("wtf.mods.%s.enabled", configKey), false),
+		focusable:  focusable,
+		starChar:   Config.UString(fmt.Sprintf("wtf.mods.%s.graphIcon", configKey), name),
+		maxStars:   Config.UInt(fmt.Sprintf("wtf.mods.%s.graphStars", configKey), 20),
 		Name:       Config.UString(fmt.Sprintf("wtf.mods.%s.title", configKey), name),
 		RefreshInt: Config.UInt(fmt.Sprintf("wtf.mods.%s.refreshInterval", configKey)),
 	}
@@ -99,8 +101,14 @@ func (widget *BarGraph) addView() {
 
 // BuildBars will build a string of * to represent your data of [time][value]
 // time should be passed as a int64
-func (widget *BarGraph) BuildBars(maxStars int, starChar string, data [][2]int64) {
+func (widget *BarGraph) BuildBars(data [][2]int64) {
 
+	widget.View.SetText(BuildStars(data, widget.maxStars, widget.starChar))
+
+}
+
+//BuildStars build the string to display
+func BuildStars(data [][2]int64, maxStars int, starChar string) string {
 	var buffer bytes.Buffer
 
 	//counter to inintialize min value
@@ -158,8 +166,7 @@ func (widget *BarGraph) BuildBars(maxStars int, starChar string, data [][2]int64
 		buffer.WriteString(fmt.Sprintf("%s -\t [red]%s[white] - (%d)\n", t.Format("Jan 02, 2006"), stars, val))
 	}
 
-	widget.View.SetText(buffer.String())
-
+	return buffer.String()
 }
 
 /* -------------------- Exported Functions -------------------- */
diff --git a/wtf_tests/bargraph/bargraph_test.go b/wtf_tests/bargraph/bargraph_test.go
new file mode 100644
index 00000000..3135c925
--- /dev/null
+++ b/wtf_tests/bargraph/bargraph_test.go
@@ -0,0 +1,33 @@
+package bargraphtests
+
+import (
+	"testing"
+
+	. "github.com/senorprogrammer/wtf/wtf"
+	. "github.com/stretchr/testify/assert"
+)
+
+// MakeData - Create sample data
+func makeData() [][2]int64 {
+
+	//this could come from config
+	const lineCount = 2
+	var stats [lineCount][2]int64
+
+	stats[0][1] = 1530122942
+	stats[0][0] = 100
+
+	stats[1][1] = 1530132942
+	stats[1][0] = 210
+
+	return stats[:]
+
+}
+
+//TestOutput of the bargraph make string (BuildStars) function
+func TestOutput(t *testing.T) {
+
+	result := BuildStars(makeData(), 20, "*")
+
+	Equal(t, result, "Jan 18, 1970 -\t [red]*[white] - (100)\nJan 18, 1970 -\t [red]********************[white] - (210)\n")
+}