Fixed file extension test to handle extensions with one letter

Signed-off-by: Jonathan Schweder <jonathanschweder@gmail.com>
jaswdr · Oct 10, 2020 · 1fa8ba6 · 1fa8ba6
1 parent c9bbf0f
commit 1fa8ba6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/faker_test.go b/faker_test.go
@@ -3,9 +3,9 @@ package faker
 import (
 	"fmt"
 	"math/rand"
+	"reflect"
 	"strings"
 	"testing"
-	"reflect"
 )
 
 func Expect(t *testing.T, expected, got interface{}, values ...interface{}) {
@@ -191,4 +191,4 @@ func TestBoolWithChance(t *testing.T) {
 	Expect(t, false, f.BoolWithChance(0))
 	Expect(t, true, f.BoolWithChance(101))
 	Expect(t, false, f.BoolWithChance(-1))
-}
+}
diff --git a/file_test.go b/file_test.go
@@ -7,7 +7,7 @@ import (
 
 func TestExtension(t *testing.T) {
 	p := New().File()
-	Expect(t, true, len(p.Extension()) > 2)
+	Expect(t, true, p.Extension() != "")
 }
 
 func TestFileWithExtension(t *testing.T) {