TestCapitalize

qiniu · Jan 12, 2025 · 33dcf91 · 33dcf91
1 parent 5e9ce17
commit 33dcf91
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/stringutil/string_test.go b/stringutil/string_test.go
@@ -22,6 +22,18 @@ import (
 	"testing"
 )
 
+func TestCapitalize(t *testing.T) {
+	if ret := Capitalize(""); ret != "" {
+		t.Fatal("Capitalize:", ret)
+	}
+	if ret := Capitalize("hello"); ret != "Hello" {
+		t.Fatal("Capitalize:", ret)
+	}
+	if ret := Capitalize("Hello"); ret != "Hello" {
+		t.Fatal("Capitalize:", ret)
+	}
+}
+
 func TestConcat(t *testing.T) {
 	if ret := Concat("1"); ret != "1" {
 		t.Fatal("Concat(1):", ret)

diff --git a/stringutil/transform.go b/stringutil/transform.go
@@ -25,7 +25,7 @@ import (
 // its upper case.
 func Capitalize(str string) string {
 	c, nc := utf8.DecodeRuneInString(str)
-	if c == utf8.RuneError || unicode.IsUpper(c) {
+	if unicode.IsUpper(c) || c == utf8.RuneError {
 		return str
 	}
 	ret := make([]byte, len(str))