github.com/Cloud-Foundations/Dominator@v0.3.4/lib/filegen/register_test.go (about)

     1  package filegen
     2  
     3  import (
     4  	"fmt"
     5  	"testing"
     6  	"time"
     7  
     8  	"github.com/Cloud-Foundations/Dominator/lib/log"
     9  	"github.com/Cloud-Foundations/Dominator/lib/log/testlogger"
    10  	"github.com/Cloud-Foundations/Dominator/lib/mdb"
    11  )
    12  
    13  type testGenerator struct{}
    14  
    15  var testData = []byte("data")
    16  
    17  func (g *testGenerator) Generate(machine mdb.Machine, logger log.Logger) (
    18  	data []byte, validUntil time.Time, err error) {
    19  	return testData, time.Now().Add(time.Minute), nil
    20  }
    21  
    22  func TestManyRegisters(t *testing.T) {
    23  	m := New(testlogger.New(t))
    24  	dataGenerator := &testGenerator{}
    25  	var pathnames []string
    26  	for count := 0; count < 100; count++ {
    27  		pathnames = append(pathnames, fmt.Sprintf("dir/file%d", count))
    28  	}
    29  	for _, pathname := range pathnames {
    30  		m.RegisterGeneratorForPath(pathname, dataGenerator)
    31  	}
    32  }