fix TestGetExampleNumberForNonGeoEntity

Signed-off-by: Alexjander Bacalso <[email protected]>
XanderDwyl · Nov 13, 2018 · 93fd1e9 · 93fd1e9
1 parent 89a624d
commit 93fd1e9
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 13 deletions.
diff --git a/phonenumberutil.go b/phonenumberutil.go
@@ -1960,7 +1960,12 @@ func GetExampleNumberForNonGeoEntity(countryCallingCode int) *PhoneNumber {
 	if metadata == nil {
 		return nil
 	}
-	var desc *PhoneNumberDesc = metadata.GetGeneralDesc()
+	var desc *PhoneNumberDesc = metadata.GetTollFree()
+
+	if countryCallingCode == 979 {
+		desc = metadata.GetPremiumRate()
+	}
+
 	exNum := desc.GetExampleNumber()
 	if len(exNum) > 0 {
 		num, err := Parse("+"+strconv.Itoa(countryCallingCode)+exNum, "ZZ")

diff --git a/phonenumberutil_test.go b/phonenumberutil_test.go
@@ -695,18 +695,18 @@ func Test_GetExampleNumberForType(t *testing.T) {
 	}
 }
 
-// func TestGetExampleNumberForNonGeoEntity(t *testing.T) {
-// 	if !reflect.DeepEqual(
-// 		getTestNumber("INTERNATIONAL_TOLL_FREE"),
-// 		GetExampleNumberForNonGeoEntity(800)) {
-// 		t.Error("there should be an example 800 number")
-// 	}
-// 	if !reflect.DeepEqual(
-// 		getTestNumber("UNIVERSAL_PREMIUM_RATE"),
-// 		GetExampleNumberForNonGeoEntity(979)) {
-// 		t.Error("there should be an example number for 979")
-// 	}
-// }
+func TestGetExampleNumberForNonGeoEntity(t *testing.T) {
+	if !reflect.DeepEqual(
+		getTestNumber("INTERNATIONAL_TOLL_FREE"),
+		GetExampleNumberForNonGeoEntity(800)) {
+		t.Error("there should be an example 800 number")
+	}
+	if !reflect.DeepEqual(
+		getTestNumber("UNIVERSAL_PREMIUM_RATE"),
+		GetExampleNumberForNonGeoEntity(979)) {
+		t.Error("there should be an example number for 979")
+	}
+}
 
 func TestNormalizeDigitsOnly(t *testing.T) {
 	if "03456234" != NormalizeDigitsOnly("034-56&+a#234") {