nomino/generators_test.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)
}