61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
package nomino
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestWithUUID(t *testing.T) {
|
|
var c config
|
|
WithUUID()(&c)
|
|
st, err := c.generator()
|
|
assert.NoError(t, err)
|
|
_, parseErr := uuid.Parse(st)
|
|
assert.NoError(t, parseErr)
|
|
}
|
|
|
|
type badRead struct{}
|
|
|
|
func (badRead) Read([]byte) (int, error) {
|
|
return 0, errors.New("sorry")
|
|
}
|
|
|
|
func TestWithUUIDFail(t *testing.T) {
|
|
var c config
|
|
uuid.SetRand(badRead{})
|
|
defer uuid.SetRand(nil)
|
|
WithUUID()(&c)
|
|
_, err := c.generator()
|
|
assert.Equal(t, errors.New("sorry"), err)
|
|
}
|
|
|
|
func TestWithTimestamp(t *testing.T) {
|
|
var c config
|
|
WithTimestamp()(&c)
|
|
n := time.Now()
|
|
st, err := c.generator()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, n.Format(FileTimestamp), st)
|
|
}
|
|
|
|
func TestWithTime(t *testing.T) {
|
|
var c config
|
|
d := time.Date(1986, time.March, 28, 12, 0, 0, 0, time.UTC)
|
|
WithTime(d)(&c)
|
|
st, err := c.generator()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, d.Format(FileTimestamp), st)
|
|
}
|
|
|
|
func TestWithTimestampUTC(t *testing.T) {
|
|
var c config
|
|
WithTimestampUTC()(&c)
|
|
n := time.Now()
|
|
st, err := c.generator()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, n.UTC().Format(FileTimestampNoTZ), st)
|
|
}
|