coderd/coderdtest/uuids.go

Rama main · réplica de lectura del repositorio original.

package coderdtest

import "github.com/google/uuid"

// DeterministicUUIDGenerator allows "naming" uuids for unit tests.
// An example of where this is useful, is when a tabled test references
// a UUID that is not yet known. An alternative to this would be to
// hard code some UUID strings, but these strings are not human friendly.
type DeterministicUUIDGenerator struct {
	Named map[string]uuid.UUID
}

func NewDeterministicUUIDGenerator() *DeterministicUUIDGenerator {
	return &DeterministicUUIDGenerator{
		Named: make(map[string]uuid.UUID),
	}
}

func (d *DeterministicUUIDGenerator) ID(name string) uuid.UUID {
	if v, ok := d.Named[name]; ok {
		return v
	}
	d.Named[name] = uuid.New()
	return d.Named[name]
}