// SPDX-License-Identifier: BSD-3-Clause
//go:build freebsd || openbsd

package mem

import (
	"testing"

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

const validFreeBSD = `Device:       1kB-blocks      Used:
/dev/gpt/swapfs    1048576          1234
/dev/md0         1048576          666
`

const validOpenBSD = `Device       1K-blocks      Used	Avail	Capacity	Priority
/dev/wd0b    655025          1234	653791	1%	0
`

const invalid = `Device:       512-blocks      Used:
/dev/gpt/swapfs    1048576          1234
/dev/md0         1048576          666
`

func TestParseSwapctlOutput_FreeBSD(t *testing.T) {
	stats, err := parseSwapctlOutput(validFreeBSD)
	require.NoError(t, err)

	assert.Equal(t, SwapDevice{
		Name:      "/dev/gpt/swapfs",
		UsedBytes: 1263616,
		FreeBytes: 1072478208,
	}, *stats[0])

	assert.Equal(t, SwapDevice{
		Name:      "/dev/md0",
		UsedBytes: 681984,
		FreeBytes: 1073059840,
	}, *stats[1])
}

func TestParseSwapctlOutput_OpenBSD(t *testing.T) {
	stats, err := parseSwapctlOutput(validOpenBSD)
	require.NoError(t, err)

	assert.Equal(t, SwapDevice{
		Name:      "/dev/wd0b",
		UsedBytes: 1234 * 1024,
		FreeBytes: 653791 * 1024,
	}, *stats[0])
}

func TestParseSwapctlOutput_Invalid(t *testing.T) {
	_, err := parseSwapctlOutput(invalid)
	assert.Error(t, err)
}

func TestParseSwapctlOutput_Empty(t *testing.T) {
	_, err := parseSwapctlOutput("")
	assert.Error(t, err)
}