Fix: Update

Signed-off-by: vr-varad <varadgupta21@gmail.com>

Author: vr-varad

random/random.go

@@ -184,9 +184,9 @@ func RandIPv4Private() net.IP {
 func RandIPv4Loopback() net.IP {
 	return net.IPv4(
 		127,
-		0,
-		0,
-		1,
+		byte(RandIntRange(1, 256)),
+		byte(RandIntRange(1, 256)),
+		byte(RandIntRange(1, 256)),
 	)
 }
 

random/random_test.go

@@ -204,23 +204,8 @@ func TestRandIPv4Private(t *testing.T) {
 			t.Error("Generated IP is not a valid IPv4 address: " + r.String())
 		}
 
-		// Check if the IP is in one of the private ranges
-		isPrivate := false
-
-		// 10.0.0.0/8 (10.0.0.0 - 10.255.255.255)
-		if r[0] == 10 {
-			isPrivate = true
-		}
-
-		// 172.16.0.0/12 (172.16.0.0 - 172.31.255.255)
-		if r[0] == 172 && r[1] >= 16 && r[1] <= 31 {
-			isPrivate = true
-		}
-
-		// 192.168.0.0/16 (192.168.0.0 - 192.168.255.255)
-		if r[0] == 192 && r[1] == 168 {
-			isPrivate = true
-		}
+		// Check if the IP is private
+		isPrivate := r.IsPrivate()
 
 		if !isPrivate {
 			t.Error("Generated IP is not in a private range: " + r.String())
@@ -229,7 +214,7 @@ func TestRandIPv4Private(t *testing.T) {
 }
 
 func TestRandIPv4Loopback(t *testing.T) {
-	for range 100 {
+	for range 5 {
 		r := RandIPv4Loopback()
 		parsed := net.ParseIP(r.String())
 		if parsed == nil {
@@ -239,10 +224,11 @@ func TestRandIPv4Loopback(t *testing.T) {
 			t.Error("Generated IP is not a valid IPv4 address: " + r.String())
 		}
 
-		// Check if it's exactly 127.0.0.1
-		expected := "127.0.0.1"
-		if r.String() != expected {
-			t.Error("Generated IP is not the expected loopback address. Expected: " + expected + ", Got: " + r.String())
+		// Check if the IP is loopback
+		isLoopback := r.IsLoopback()
+
+		if !isLoopback {
+			t.Error("Generated IP is not the expected loopback address:" + r.String())
 		}
 	}
 }