added BoolWithChance() function (#18)

jaswdr · Oct 7, 2020 · 9fc136a · 9fc136a
1 parent 25517ca
commit 9fc136a
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/faker.go b/faker.go
@@ -165,6 +165,17 @@ func (f Faker) Bool() bool {
 	return f.IntBetween(0, 100) > 50
 }
 
+// BoolWithChance returns true with chance chanceTrue in percents otherwise returns false
+func (f Faker) BoolWithChance(chanceTrue int) bool {
+	if chanceTrue <= 0 {
+		return false
+	} else if chanceTrue >= 100 {
+		return true
+	}
+
+	return f.IntBetween(0, 100) < chanceTrue
+}
+
 func (f Faker) Lorem() Lorem {
 	return Lorem{&f}
 }

diff --git a/faker_test.go b/faker_test.go
@@ -181,3 +181,14 @@ func TestBool(t *testing.T) {
 	tp := reflect.TypeOf(f.Bool())
 	Expect(t, "bool", tp.String())
 }
+
+func TestBoolWithChance(t *testing.T) {
+	f := New()
+	tp := reflect.TypeOf(f.BoolWithChance(30))
+	Expect(t, "bool", tp.String())
+
+	Expect(t, true, f.BoolWithChance(100))
+	Expect(t, false, f.BoolWithChance(0))
+	Expect(t, true, f.BoolWithChance(101))
+	Expect(t, false, f.BoolWithChance(-1))
+}