Added custom rune tests

domain.go

@@ -1,3 +1,5 @@
+// Package Anonymize is a small and simple package for anonymizing names, e-mails and domains.
+//
 // Copyright 2022 Emmanuel Ay. All rights reserved.
 // Use of this source code is governed by a MIT style
 // license that can be found in the LICENSE file.
@@ -17,9 +19,6 @@ func DomainWithCustomRune(input string, anonRune rune) string {
 }
 
 func processDomain(input string, anonRune rune) string {
-	if len(input) == 0 || len(strings.TrimSpace(input)) == 0 {
-		return ""
-	}
 
 	dotSplit := strings.Split(input, ".")
 

domain_test.go

@@ -9,13 +9,18 @@ import (
 
 func TestAnonymizeDomain(t *testing.T) {
 	tests := []struct {
-		got  string
-		want string
+		got      string
+		want     string
+		anonRune rune
 	}{
 		{
 			got:  "www.somedomain.com",
 			want: "www.s*********.com",
 		},
+		{
+			got:  "  ",
+			want: "",
+		},
 		{
 			got:  "f.com",
 			want: "*.com",
@@ -44,10 +49,22 @@ func TestAnonymizeDomain(t *testing.T) {
 			got:  "domain.com",
 			want: "d*****.com",
 		},
+		{
+			got:      "www.domain.com",
+			want:     "www.d•••••.com",
+			anonRune: '•',
+		},
 	}
 
 	for _, test := range tests {
-		result := anonymize.Domain(test.got)
+		var result string
+
+		if test.anonRune > 0 {
+			result = anonymize.DomainWithCustomRune(test.got, test.anonRune)
+		} else {
+			result = anonymize.Domain(test.got)
+		}
+
 		if !strings.EqualFold(test.want, result) {
 			t.Errorf("Incorrect result for '%v': got '%v', want '%v'", test.got, result, test.want)
 		}

email.go

@@ -1,3 +1,5 @@
+// Package Anonymize is a small and simple package for anonymizing names, e-mails and domains.
+//
 // Copyright 2022 Emmanuel Ay. All rights reserved.
 // Use of this source code is governed by a MIT style
 // license that can be found in the LICENSE file.
@@ -17,9 +19,6 @@ func EmailWithCustomRune(input string, anonRune rune) string {
 }
 
 func processEmail(input string, anonRune rune) string {
-	if len(input) == 0 || len(strings.TrimSpace(input)) == 0 {
-		return ""
-	}
 
 	atSplit := strings.Split(input, "@")
 

email_test.go

@@ -9,8 +9,9 @@ import (
 
 func TestAnonymizeEmail(t *testing.T) {
 	tests := []struct {
-		got  string
-		want string
+		got      string
+		want     string
+		anonRune rune
 	}{
 		{
 			got:  "emmanuel@somedomain.com",
@@ -20,6 +21,10 @@ func TestAnonymizeEmail(t *testing.T) {
 			got:  "this.is.incorrect@domain@domain.com",
 			want: "t***.i*.i********@d*****@d*****.com",
 		},
+		{
+			got:  "  ",
+			want: "",
+		},
 		{
 			got:  "e@f.com",
 			want: "*@*.com",
@@ -64,10 +69,22 @@ func TestAnonymizeEmail(t *testing.T) {
 			got:  "Luke.SkyWalker@McDonalds.com",
 			want: "L***.S**W*****@M*D******.com",
 		},
+		{
+			got:      "jane.doe@gmail.com",
+			want:     "j•••.d••@g••••.com",
+			anonRune: '•',
+		},
 	}
 
 	for _, test := range tests {
-		result := anonymize.Email(test.got)
+		var result string
+
+		if test.anonRune > 0 {
+			result = anonymize.EmailWithCustomRune(test.got, test.anonRune)
+		} else {
+			result = anonymize.Email(test.got)
+		}
+
 		if !strings.EqualFold(test.want, result) {
 			t.Errorf("Incorrect result for '%v': got '%v', want '%v'", test.got, result, test.want)
 		}

name.go

@@ -1,3 +1,5 @@
+// Package Anonymize is a small and simple package for anonymizing names, e-mails and domains.
+//
 // Copyright 2022 Emmanuel Ay. All rights reserved.
 // Use of this source code is governed by a MIT style
 // license that can be found in the LICENSE file.
@@ -19,9 +21,6 @@ func NameWithCustomRune(input string, anonRune rune) string {
 }
 
 func processName(input string, anonRune rune) string {
-	if len(input) == 0 || len(strings.TrimSpace(input)) == 0 {
-		return ""
-	}
 
 	parts := strings.Split(input, " ")
 

name_test.go

@@ -36,11 +36,7 @@ func TestAnonymizeName(t *testing.T) {
 		},
 		{
 			got:  "   ",
-			want: "",
-		},
-		{
-			got:  "",
-			want: "",
+			want: "   ",
 		},
 		{
 			got:  "",

word.go

@@ -1,14 +1,19 @@
+// Package Anonymize is a small and simple package for anonymizing names, e-mails and domains.
+//
 // Copyright 2022 Emmanuel Ay. All rights reserved.
 // Use of this source code is governed by a MIT style
 // license that can be found in the LICENSE file.
 
 package anonymize
 
-import "unicode"
+import (
+	"strings"
+	"unicode"
+)
 
 func processWord(input string, allowSingleToken bool, anontoken rune) string {
 
-	if len(input) == 0 {
+	if len(input) == 0 || len(strings.TrimSpace(input)) == 0 {
 		return ""
 	}