dan/nomino
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

♻️ Export Generator, and switch to WithGenerator Option

Browse source
This commit is contained in:
Dan Jones 2025-03-10 13:48:11 -05:00
commit ce5f823d64
3 changed files with 53 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

37
generators_test.go
View file

@ -9,11 +9,18 @@ import (
"github.com/stretchr/testify/assert"
)
func TestWithUUID(t *testing.T) {
func TestWithGenerator(t *testing.T) {
g := func() (string, error) { return "abc", nil }
var c config
WithUUID()(&c)
WithGenerator(g)(&c)
st, err := c.generator()
assert.NoError(t, err)
assert.Equal(t, "abc", st)
}
func TestUUID(t *testing.T) {
st, err := UUID()()
assert.NoError(t, err)
_, parseErr := uuid.Parse(st)
assert.NoError(t, parseErr)
}
@ -24,38 +31,32 @@ func (badRead) Read([]byte) (int, error) {
return 0, errors.New("sorry")
}
func TestWithUUIDFail(t *testing.T) {
var c config
func TestUUIDFail(t *testing.T) {
uuid.SetRand(badRead{})
defer uuid.SetRand(nil)
WithUUID()(&c)
_, err := c.generator()
_, err := UUID()()
assert.Equal(t, errors.New("sorry"), err)
}
func TestWithTimestamp(t *testing.T) {
var c config
WithTimestamp()(&c)
func TestTimestamp(t *testing.T) {
n := time.Now()
st, err := c.generator()
st, err := Timestamp()()
assert.NoError(t, err)
assert.Equal(t, n.Format(FileTimestamp), st)
}
func TestWithTime(t *testing.T) {
var c config
func TestTime(t *testing.T) {
d := time.Date(1986, time.March, 28, 12, 0, 0, 0, time.UTC)
WithTime(d)(&c)
st, err := c.generator()
st, err := Time(d)()
assert.NoError(t, err)
assert.Equal(t, d.Format(FileTimestamp), st)
}
func TestWithTimestampUTC(t *testing.T) {
var c config
WithTimestampUTC()(&c)
func TestTimestampUTC(t *testing.T) {
n := time.Now()
st, err := c.generator()
st, err := TimestampUTC()()
assert.NoError(t, err)
assert.Equal(t, n.UTC().Format(FileTimestampNoTZ), st)
}