portainer/pkg/librand/rand_test.go

package librand

import (
	"strings"
	"testing"

	"github.com/portainer/portainer/pkg/fips"

	"github.com/stretchr/testify/require"
)

func init() {
	fips.InitFIPS(false)
}

func TestIntn(t *testing.T) {
	t.Parallel()
	i := Intn(10)

	if i >= 10 || i < 0 {
		t.Fatalf("random number %d wasn't within interval", i)
	}
}

func TestInternalIntn(t *testing.T) {
	t.Parallel()
	testCases := []struct {
		name string
		max  int
		fips bool
	}{
		{
			name: "non-fips mode",
			max:  10,
			fips: false,
		},
		{
			name: "fips mode",
			max:  10,
			fips: true,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			i := intn(tc.max, tc.fips)

			if i >= tc.max || i < 0 {
				t.Fatalf("random number %d wasn't within interval", i)
			}
		})
	}
}

func TestString(t *testing.T) {
	t.Parallel()

	const iterations = 1000

	seen := make(map[string]bool, iterations)
	for range iterations {
		s := String(16)

		require.Len(t, s, 16)
		require.Empty(t, strings.Trim(s, alphanumericChars))
		require.False(t, seen[s])

		seen[s] = true
	}
}

func TestFloat64(t *testing.T) {
	t.Parallel()
	f := Float64()

	if f >= 1 || f < 0 {
		t.Fatalf("random float %v wasn't within interval", f)
	}
}

func TestInternalFloat64(t *testing.T) {
	t.Parallel()
	testCases := []struct {
		name string
		fips bool
	}{
		{
			name: "non-fips mode",
			fips: false,
		},
		{
			name: "fips mode",
			fips: true,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			f := randomFloat64(tc.fips)
			if f >= 1 || f < 0 {
				t.Fatalf("random float %v wasn't within interval", f)
			}
		})
	}
}