added unit tests

Signed-off-by: Frederic BIDON <fredbi@yahoo.com>
bxcodec · May 10, 2021 · 321a335 · 321a335
1 parent 34c1bc0
commit 321a335
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/random_source_test.go b/random_source_test.go
@@ -0,0 +1,25 @@
+package faker
+
+import (
+	cryptorand "crypto/rand"
+	"io"
+	mathrand "math/rand"
+	"testing"
+	"time"
+)
+
+func TestSetRandomSource(t *testing.T) {
+	SetRandomSource(NewSafeSource(mathrand.NewSource(time.Now().UnixNano())))
+
+	_ = rand.Int31n(100)
+}
+
+func TestSetCryptoSource(t *testing.T) {
+	SetCryptoSource(cryptorand.Reader)
+
+	buf := make([]byte, 10)
+	_, err := io.ReadFull(crypto, buf)
+	if err != nil {
+		t.Error("Expected Not Error, But Got: ", err)
+	}
+}