package data

import (
	"testing"

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

func TestFrequencyName(t *testing.T) {
	assert := assert.New(t)

	assert.Equal("11a", WirelessAirtime{Frequency: 5000}.FrequencyName())
	assert.Equal("11g", WirelessAirtime{Frequency: 4999}.FrequencyName())
	assert.Equal("11g", WirelessAirtime{Frequency: 2412}.FrequencyName())
}

func TestUtilization(t *testing.T) {
	assert := assert.New(t)

	t1 := &WirelessAirtime{
		ActiveTime: 20,
		BusyTime:   0,
		TxTime:     5,
		RxTime:     0,
	}
	t2 := &WirelessAirtime{
		ActiveTime: 120,
		BusyTime:   10,
		TxTime:     25,
		RxTime:     15,
	}
	t3 := &WirelessAirtime{
		ActiveTime: 200,
		BusyTime:   40,
		TxTime:     35,
		RxTime:     15,
	}

	t1.SetUtilization(t2)
	assert.Zero(t1.ChanUtil)
	assert.Zero(t1.TxUtil)
	assert.Zero(t1.RxUtil)

	t2.SetUtilization(t1)
	assert.NotZero(t2.ChanUtil)
	assert.EqualValues(45, t2.ChanUtil)
	assert.EqualValues(20, t2.RxUtil)
	assert.EqualValues(15, t2.TxUtil)

	t3.SetUtilization(t2)
	assert.EqualValues(50, t3.ChanUtil)
	assert.EqualValues(12.5, t3.RxUtil)
	assert.EqualValues(0, t3.TxUtil)
}

func TestWirelessStatistics(t *testing.T) {
	assert := assert.New(t)

	stats := WirelessStatistics([]*WirelessAirtime{{
		Frequency:   2400,
		ActiveTime: 20,
		TxTime:     10,
	}})

	// Different Frequency, should not change anything
	stats.SetUtilization([]*WirelessAirtime{{
		Frequency:   5000,
		ActiveTime: 15,
		TxTime:     1,
	}})
	assert.EqualValues(0, stats[0].ChanUtil)

	// Same Frequency, should set the utilization
	stats.SetUtilization([]*WirelessAirtime{{
		Frequency:   2400,
		ActiveTime: 10,
		TxTime:     5,
	}})
	assert.EqualValues(50, stats[0].ChanUtil)
}