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{
		Active_time: 20,
		Busy_time:   0,
		Tx_time:     5,
		Rx_time:     0,
	}
	t2 := &WirelessAirtime{
		Active_time: 120,
		Busy_time:   10,
		Tx_time:     25,
		Rx_time:     15,
	}
	t3 := &WirelessAirtime{
		Active_time: 200,
		Busy_time:   40,
		Tx_time:     35,
		Rx_time:     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,
		Active_time: 20,
		Tx_time:     10,
	}})

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

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