new Unicode code point search tests

Author: chrissimpkins

uni_test.go

@@ -10,7 +10,29 @@ import (
 	"testing"
 )
 
-func TestMainFunction(t *testing.T) {
+// Constant tests
+
+func TestVersionString(t *testing.T) {
+	r, _ := regexp.Compile(`\d{1,2}.\d{1,2}.\d{1,2}`)
+	// match expected format of version string
+	if r.MatchString(version) != true {
+		t.Errorf("[FAIL] Unexpected version string format identified.")
+	}
+}
+
+func TestUsageString(t *testing.T) {
+	if strings.HasPrefix(usage, "Usage:") == false {
+		t.Errorf("[FAIL] Unexpected usage string format.")
+	}
+}
+
+func TestHelpString(t *testing.T) {
+	if strings.HasPrefix(help, "=====") == false {
+		t.Errorf("[FAIL] Unexpected help string format.")
+	}
+}
+
+func TestMainFunctionSingleGlyph(t *testing.T) {
 	os.Args = []string{"uni", "j"}
 	old := os.Stdout // keep backup of stdout
 	r, w, err := os.Pipe()
@@ -37,28 +59,115 @@ func TestMainFunction(t *testing.T) {
 
 	if len(out) == 0 {
 		t.Errorf("[FAIL] Test of main() function did not return standard output response")
-	} else if !strings.HasPrefix(out, "U+006A") {
-		t.Errorf("[FAIL] Expected execution of 'uni f' to return string that begins with 'U+006A', but instead it returned %s", out)
+	}
+
+	if !(out == "U+006A 'j'\n") {
+		t.Errorf("[FAIL] Expected execution of 'uni f' to return string U+006A 'j', but instead it returned %s", out)
 	}
 
 }
 
-func TestVersionString(t *testing.T) {
-	r, _ := regexp.Compile(`\d{1,2}.\d{1,2}.\d{1,2}`)
-	// match expected format of version string
-	if r.MatchString(version) != true {
-		t.Errorf("[FAIL] Unexpected version string format identified.")
+func TestMainFunctionMultiGlyph(t *testing.T) {
+	os.Args = []string{"uni", "jj"}
+	old := os.Stdout // keep backup of stdout
+	r, w, err := os.Pipe()
+	if err != nil {
+		log.Fatal(err)
 	}
+	os.Stdout = w
+
+	outC := make(chan string)
+
+	// copy the output in a separate goroutine
+	go func() {
+		var buf bytes.Buffer
+		io.Copy(&buf, r)
+		outC <- buf.String()
+	}()
+
+	main() // call main function with mock os.Args defined above
+
+	// back to normal state
+	w.Close()
+	os.Stdout = old // restoring the real stdout
+	out := <-outC
+
+	if len(out) == 0 {
+		t.Errorf("[FAIL] Test of main() function did not return standard output response")
+	}
+
+	if !(out == "U+006A 'j'\nU+006A 'j'\n") {
+		t.Errorf("[FAIL] Expected execution of 'uni f' to return string U+006A 'j' * 2, but instead it returned %s", out)
+	}
+
 }
 
-func TestUsageString(t *testing.T) {
-	if strings.HasPrefix(usage, "Usage:") == false {
-		t.Errorf("[FAIL] Unexpected usage string format.")
+func TestMainFunctionSingleUnicodeCodePoint(t *testing.T) {
+	os.Args = []string{"uni", "-g", "006A"}
+	old := os.Stdout // keep backup of stdout
+	r, w, err := os.Pipe()
+	if err != nil {
+		log.Fatal(err)
+	}
+	os.Stdout = w
+
+	outC := make(chan string)
+
+	// copy the output in a separate goroutine
+	go func() {
+		var buf bytes.Buffer
+		io.Copy(&buf, r)
+		outC <- buf.String()
+	}()
+
+	main() // call main function with mock os.Args defined above
+
+	// back to normal state
+	w.Close()
+	os.Stdout = old // restoring the real stdout
+	out := <-outC
+
+	if len(out) == 0 {
+		t.Errorf("[FAIL] Test of main() function did not return standard output response")
+	}
+
+	if !(out == "U+006A 'j'\n") {
+		t.Errorf("[FAIL] Expected execution of 'uni f' to return string U+006A 'j', but instead it returned %s", out)
 	}
+
 }
 
-func TestHelpString(t *testing.T) {
-	if strings.HasPrefix(help, "=====") == false {
-		t.Errorf("[FAIL] Unexpected help string format.")
+func TestMainFunctionMultiUnicodeCodePoint(t *testing.T) {
+	os.Args = []string{"uni", "-g", "006A", "006A"}
+	old := os.Stdout // keep backup of stdout
+	r, w, err := os.Pipe()
+	if err != nil {
+		log.Fatal(err)
 	}
+	os.Stdout = w
+
+	outC := make(chan string)
+
+	// copy the output in a separate goroutine
+	go func() {
+		var buf bytes.Buffer
+		io.Copy(&buf, r)
+		outC <- buf.String()
+	}()
+
+	main() // call main function with mock os.Args defined above
+
+	// back to normal state
+	w.Close()
+	os.Stdout = old // restoring the real stdout
+	out := <-outC
+
+	if len(out) == 0 {
+		t.Errorf("[FAIL] Test of main() function did not return standard output response")
+	}
+
+	if !(out == "U+006A 'j'\nU+006A 'j'\n") {
+		t.Errorf("[FAIL] Expected execution of 'uni f' to return string U+006A 'j' * 2, but instead it returned %s", out)
+	}
+
 }