tests(testutil): add tests for StripProgress and AssertBody (#231)

Author: ultimateboy

pkg/testutil/testutil.go

@@ -87,13 +87,13 @@ func AssertBody(t *testing.T, expected interface{}, req *http.Request) {
 // StripProgress strips the output from the progress method
 func StripProgress(input string) string {
 	first := strings.Index(input, "\b")
-	// If /b charecter not part of string
+	// If \b charecter not part of string
 	if first == -1 {
 		return input
 	}
 	last := strings.LastIndex(input, "\b")
 
-	// return string without /b or the charecters it deletes.
+	// return string without \b or the characters it deletes.
 	return input[:first-(last-first+1)] + input[last+1:]
 }
 

pkg/testutil/testutil_test.go

@@ -0,0 +1,49 @@
+package testutil
+
+import (
+	"bytes"
+	"io"
+	"net/http"
+	"testing"
+
+	"github.com/arschles/assert"
+)
+
+type nopCloser struct {
+	io.Reader
+}
+
+func (nopCloser) Close() error { return nil }
+
+// TestStripProgress ensures StripProgress strips what is expected.
+func TestStripProgress(t *testing.T) {
+	testInput := "Lorem ipsum dolar sit amet"
+	expectedOutput := "Lorem ipsum dolar sit amet"
+
+	assert.Equal(t, StripProgress(testInput), expectedOutput, "output")
+
+	testInput = "Lorem ipsum dolar sit amet...\b\b\b"
+	assert.Equal(t, StripProgress(testInput), expectedOutput, "output")
+}
+
+// TestAssertBody ensures AssertBody correctly marshals into the interface.
+func TestAssertBody(t *testing.T) {
+
+	b := nopCloser{bytes.NewBufferString(`{"data":{"lorem":"ipsum"},"dolar":["sit","amet"]}`)}
+
+	sampleRequest := http.Request{
+		Body: b,
+	}
+
+	expected := map[string]interface{}{
+		"data": map[string]interface{}{
+			"lorem": "ipsum",
+		},
+		"dolar": []string{
+			"sit",
+			"amet",
+		},
+	}
+
+	AssertBody(t, expected, &sampleRequest)
+}