Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add generators for URLs and domain names #18

Open
wants to merge 12 commits into
base: master
Choose a base branch
from
49 changes: 49 additions & 0 deletions ip_addr.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
// Copyright 2022 Walter Scheper <walter.scheper@gmail.com>
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at https://mozilla.org/MPL/2.0/.

package rapid

import (
"fmt"
"net/netip"
)

const (
ipv4Len = 4
ipv6Len = 16
)

type ipGen struct {
ipv6 bool
}

func (g *ipGen) String() string {
return fmt.Sprintf("IP(ipv6=%v)", g.ipv6)
}

func (g *ipGen) value(t *T) netip.Addr {
var b []byte
if g.ipv6 {
b = SliceOfN(Byte(), ipv6Len, ipv6Len).Draw(t, g.String())
} else {
b = SliceOfN(Byte(), ipv4Len, ipv4Len).Draw(t, g.String())
}

addr, _ := netip.AddrFromSlice(b)
return addr
}

func IPv4() *Generator[netip.Addr] {
return newGenerator[netip.Addr](&ipGen{
ipv6: false,
})
}

func IPv6() *Generator[netip.Addr] {
return newGenerator[netip.Addr](&ipGen{
ipv6: true,
})
}
45 changes: 45 additions & 0 deletions ip_addr_example_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
// Copyright 2020 Walter Scheper <walter.scheper@gmail.com>
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at https://mozilla.org/MPL/2.0/.

package rapid_test

import (
"fmt"

"pgregory.net/rapid"
)

func ExampleIPv4() {
gen := rapid.IPv4()

for i := 0; i < 5; i++ {
addr := gen.Example(i)
fmt.Println(addr.String())
}

// Output:
// 0.23.24.7
// 100.146.0.0
// 0.222.65.1
// 1.49.104.14
// 11.56.0.83
}

func ExampleIPv6() {
gen := rapid.IPv6()

for i := 0; i < 5; i++ {
addr := gen.Example(i)
fmt.Println(addr.String())
}

// Output:
// 17:1807:e2c4:8210:7202:f4b2:a0e2:8dc
// 6492:0:fa37:b00:b5c3:4e6:6a01:c802
// de:4101:9f5:3:104:5dc:b600:905
// 131:680e:97ff:d200:ae1:4d00:2300:103
// b38:53:ff07:200:8c28:ee:ad00:1b
}
Loading