Fix Truncate

TruncateWordsToWholeSentence knows if the summary is truncated, so let "him" decide.

Fixes #880

1 files changed, 28 insertions, 0 deletions

diff --git a/helpers/content_test.go b/helpers/content_test.go
index 6191eb158..44cee2f50 100644
--- a/helpers/content_test.go
+++ b/helpers/content_test.go
@@ -3,6 +3,7 @@ package helpers
 import (
 	"github.com/stretchr/testify/assert"
 	"html/template"
+	"strings"
 	"testing"
 )
 
@@ -33,3 +34,30 @@ func TestStripEmptyNav(t *testing.T) {
 func TestBytesToHTML(t *testing.T) {
 	assert.Equal(t, template.HTML("dobedobedo"), BytesToHTML([]byte("dobedobedo")))
 }
+
+func TestTruncateWordsToWholeSentence(t *testing.T) {
+	type test struct {
+		input, expected string
+		max             int
+		truncated       bool
+	}
+	data := []test{
+		{"a b c", "a b c", 12, false},
+		{"a b c", "a b c", 3, false},
+		{"a", "a", 1, false},
+		{"This is a sentence.", "This is a sentence.", 5, false},
+		{"This is also a sentence!", "This is also a sentence!", 1, false},
+		{"To be. Or not to be. That's the question.", "To be.", 1, true},
+		{" \nThis is not a sentence\n ", "This is not a", 4, true},
+	}
+	for i, d := range data {
+		output, truncated := TruncateWordsToWholeSentence(strings.Fields(d.input), d.max)
+		if d.expected != output {
+			t.Errorf("Test %d failed. Expected %q got %q", i, d.expected, output)
+		}
+
+		if d.truncated != truncated {
+			t.Errorf("Test %d failed. Expected truncated=%t got %t", i, d.truncated, truncated)
+		}
+	}
+}