Merge pull request #83 from AleFossa/main

Mzack9999 · web-flow · commit 1d3007facb2c · 2023-02-12T23:02:46.000+01:00
Adding Truncate methods to stringsutil
diff --git a/strings/stringsutil.go b/strings/stringsutil.go
@@ -245,3 +245,11 @@ func LongestRepeatingSequence(s string) LongestSequence {
 	}
 	return LongestSequence{Sequence: res, Count: resCount}
 }
+
+// Truncate a string to max length
+func Truncate(data string, maxSize int) string {
+	if maxSize >= 0 && len(data) > maxSize {
+		return data[:maxSize]
+	}
+	return data
+}
diff --git a/strings/stringsutil_test.go b/strings/stringsutil_test.go
@@ -288,3 +288,26 @@ func TestLongestRepeatingSequence(t *testing.T) {
 		require.Equalf(t, test.expected, result.Sequence, "test: %s, expected %q, got: %s", test.s, test.expected, result.Sequence)
 	}
 }
+
+type truncateTest struct {
+	test    string
+	maxSize int
+	result  string
+}
+
+func TestTruncate(t *testing.T) {
+	tests := []truncateTest{
+		{test: "abcd", maxSize: -1, result: "abcd"},
+		{test: "abcd", maxSize: 0, result: ""},
+		{test: "abcde", maxSize: 3, result: "abc"},
+		{test: "abcdef", maxSize: 8, result: "abcdef"},
+		{test: "abcdefg", maxSize: 6, result: "abcdef"},
+		{test: "aaaa", maxSize: 20, result: "aaaa"},
+		{test: "aaaa", maxSize: 4, result: "aaaa"},
+	}
+
+	for _, test := range tests {
+		res := Truncate(test.test, test.maxSize)
+		require.Equalf(t, test.result, res, "test:%s maxsize: %d result: %s", test.test, test.maxSize, res)
+	}
+}

Original file line number	Diff line number	Diff line change
`@@ -245,3 +245,11 @@ func LongestRepeatingSequence(s string) LongestSequence {`
`245`	`245`	`}`
`246`	`246`	`return LongestSequence{Sequence: res, Count: resCount}`
`247`	`247`	`}`
	`248`	`+`
	`249`	`+// Truncate a string to max length`
	`250`	`+func Truncate(data string, maxSize int) string {`
	`251`	`+ if maxSize >= 0 && len(data) > maxSize {`
	`252`	`+ return data[:maxSize]`
	`253`	`+ }`
	`254`	`+ return data`
	`255`	`+}`