vault/physical/raft/bolt_64bit_test.go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: BUSL-1.1

//go:build !386 && !arm

package raft

import (
	"os"
	"strconv"
	"testing"
)

func Test_BoltOptions(t *testing.T) {
	t.Parallel()
	key := "VAULT_RAFT_INITIAL_MMAP_SIZE"

	testCases := []struct {
		name         string
		env          string
		expectedSize int
	}{
		{"none", "", 100 * 1024 * 1024 * 1024},
		{"5MB", strconv.Itoa(5 * 1024 * 1024), 5 * 1024 * 1024},
		{"negative", "-1", 0},
	}

	for _, tc := range testCases {
		tc := tc

		t.Run(tc.name, func(t *testing.T) {
			if tc.env != "" {
				current := os.Getenv(key)
				defer os.Setenv(key, current)
				os.Setenv(key, tc.env)
			}

			o := boltOptions("")

			if o.InitialMmapSize != tc.expectedSize {
				t.Errorf("expected InitialMmapSize to be %d but it was %d", tc.expectedSize, o.InitialMmapSize)
			}
		})
	}
}

// TestMmapFlags tests the getMmapFlags function, ensuring it returns the appropriate integer representing the desired mmap flag.
func TestMmapFlags(t *testing.T) {
	testCases := []struct {
		name               string
		disableMapPopulate bool
	}{
		{"MAP_POPULATE is enabled", false},
		{"MAP_POPULATE disabled by env var", true},
	}

	for _, tc := range testCases {
		tc := tc

		t.Run(tc.name, func(t *testing.T) {
			if tc.disableMapPopulate {
				t.Setenv("VAULT_RAFT_DISABLE_MAP_POPULATE", "true")
			}

			isEnabled := usingMapPopulate(getMmapFlags(""))
			if tc.disableMapPopulate && isEnabled {
				t.Error("expected MAP_POPULATE to be disabled but it was enabled")
			}
		})
	}
}