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
54 changes: 54 additions & 0 deletions ip_addr.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,54 @@
// 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"
)

const (
ipv4Len = 4
ipv6Len = 16
)

var (
ipv4Gen = SliceOfN(Byte(), ipv4Len, ipv4Len)
ipv6Gen = SliceOfN(Byte(), ipv6Len, ipv6Len)
)

type ipGen struct {
ipv6 bool
}

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

func (g *ipGen) value(t *T) net.IP {
Copy link
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Maybe it is good idea to use netip? With generics we already require Go 1.18.

Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Makes sense. Done.

var gen *Generator[[]byte]
if g.ipv6 {
gen = ipv6Gen
} else {
gen = ipv4Gen
}

b := gen.Draw(t, g.String())
return net.IP(b)
}

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

func IPv6() *Generator[net.IP] {
return newGenerator[net.IP](&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
}