github.com/square/finch@v0.0.0-20240412205204-6530c03e2b96/data/string_test.go (about)

     1  // Copyright 2024 Block, Inc.
     2  
     3  package data_test
     4  
     5  import (
     6  	"strconv"
     7  	"testing"
     8  
     9  	"github.com/square/finch/data"
    10  )
    11  
    12  func TestString_StrFillAz(t *testing.T) {
    13  	lens := []string{"1", "10", "100", "1000", "10000", "100000", "1000000"}
    14  	for _, strlen := range lens {
    15  		g, _ := data.NewStrFillAz(
    16  			map[string]string{
    17  				"len": strlen,
    18  			},
    19  		)
    20  
    21  		r := data.RunCount{}
    22  
    23  		v1 := g.Values(r)
    24  		if len(v1) != 1 {
    25  			t.Fatalf("len=%s: got %d values, expected 1: %v", strlen, len(v1), v1)
    26  		}
    27  		s1 := v1[0].(string)
    28  		n, _ := strconv.Atoi(strlen)
    29  		if len(s1) != n {
    30  			t.Errorf("len=%s: got len %d, expected %s: %s", strlen, len(s1), strlen, s1)
    31  		}
    32  	}
    33  }