package net_test

import (
	"testing"

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

func TestGetNetworkRange(t *testing.T) {
	first, last, err := net.GetNetworkRange("10.0.0.0/23")
	assert.NoError(t, err)
	assert.Equal(t, "10.0.0.2", first.String())
	assert.Equal(t, "10.0.1.254", last.String())

	first, last, err = net.GetNetworkRange("172.16.0.0/12")
	assert.NoError(t, err)
	assert.Equal(t, "172.16.0.2", first.String())
	assert.Equal(t, "172.31.255.254", last.String())
}