package system

import (
	"testing"

	"github.com/influxdb/tivan/plugins/system/ps/cpu"
	"github.com/influxdb/tivan/plugins/system/ps/disk"
	"github.com/influxdb/tivan/plugins/system/ps/docker"
	"github.com/influxdb/tivan/plugins/system/ps/load"
	"github.com/influxdb/tivan/plugins/system/ps/mem"
	"github.com/influxdb/tivan/plugins/system/ps/net"
	"github.com/influxdb/tivan/testutil"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestSystemStats_GenerateStats(t *testing.T) {
	var mps MockPS

	defer mps.AssertExpectations(t)

	var acc testutil.Accumulator

	lv := &load.LoadAvgStat{
		Load1:  0.3,
		Load5:  1.5,
		Load15: 0.8,
	}

	mps.On("LoadAvg").Return(lv, nil)

	cts := cpu.CPUTimesStat{
		CPU:       "cpu0",
		User:      3.1,
		System:    8.2,
		Idle:      80.1,
		Nice:      1.3,
		Iowait:    0.2,
		Irq:       0.1,
		Softirq:   0.11,
		Steal:     0.0001,
		Guest:     8.1,
		GuestNice: 0.324,
		Stolen:    0.051,
	}

	mps.On("CPUTimes").Return([]cpu.CPUTimesStat{cts}, nil)

	du := &disk.DiskUsageStat{
		Path:        "/",
		Total:       128,
		Free:        23,
		InodesTotal: 1234,
		InodesFree:  234,
	}

	mps.On("DiskUsage").Return([]*disk.DiskUsageStat{du}, nil)

	diskio := disk.DiskIOCountersStat{
		ReadCount:    888,
		WriteCount:   5341,
		ReadBytes:    100000,
		WriteBytes:   200000,
		ReadTime:     7123,
		WriteTime:    9087,
		Name:         "sda1",
		IoTime:       123552,
		SerialNumber: "ab-123-ad",
	}

	mps.On("DiskIO").Return(map[string]disk.DiskIOCountersStat{"sda1": diskio}, nil)

	netio := net.NetIOCountersStat{
		Name:        "eth0",
		BytesSent:   1123,
		BytesRecv:   8734422,
		PacketsSent: 781,
		PacketsRecv: 23456,
		Errin:       832,
		Errout:      8,
		Dropin:      7,
		Dropout:     1,
	}

	mps.On("NetIO").Return([]net.NetIOCountersStat{netio}, nil)

	vms := &mem.VirtualMemoryStat{
		Total:       12400,
		Available:   7600,
		Used:        5000,
		UsedPercent: 47.1,
		Free:        1235,
		Active:      8134,
		Inactive:    1124,
		Buffers:     771,
		Cached:      4312,
		Wired:       134,
		Shared:      2142,
	}

	mps.On("VMStat").Return(vms, nil)

	sms := &mem.SwapMemoryStat{
		Total:       8123,
		Used:        1232,
		Free:        6412,
		UsedPercent: 12.2,
		Sin:         7,
		Sout:        830,
	}

	mps.On("SwapStat").Return(sms, nil)

	ds := &DockerContainerStat{
		Name: "blah",
		CPU: &cpu.CPUTimesStat{
			CPU:       "all",
			User:      3.1,
			System:    8.2,
			Idle:      80.1,
			Nice:      1.3,
			Iowait:    0.2,
			Irq:       0.1,
			Softirq:   0.11,
			Steal:     0.0001,
			Guest:     8.1,
			GuestNice: 0.324,
			Stolen:    0.051,
		},
		Mem: &docker.CgroupMemStat{
			ContainerID:             "blah",
			Cache:                   1,
			RSS:                     2,
			RSSHuge:                 3,
			MappedFile:              4,
			Pgpgin:                  5,
			Pgpgout:                 6,
			Pgfault:                 7,
			Pgmajfault:              8,
			InactiveAnon:            9,
			ActiveAnon:              10,
			InactiveFile:            11,
			ActiveFile:              12,
			Unevictable:             13,
			HierarchicalMemoryLimit: 14,
			TotalCache:              15,
			TotalRSS:                16,
			TotalRSSHuge:            17,
			TotalMappedFile:         18,
			TotalPgpgIn:             19,
			TotalPgpgOut:            20,
			TotalPgFault:            21,
			TotalPgMajFault:         22,
			TotalInactiveAnon:       23,
			TotalActiveAnon:         24,
			TotalInactiveFile:       25,
			TotalActiveFile:         26,
			TotalUnevictable:        27,
		},
	}

	mps.On("DockerStat").Return([]*DockerContainerStat{ds}, nil)

	ss := &SystemStats{ps: &mps}

	err := ss.Gather(&acc)
	require.NoError(t, err)

	assert.True(t, acc.CheckValue("system_load1", 0.3))
	assert.True(t, acc.CheckValue("system_load5", 1.5))
	assert.True(t, acc.CheckValue("system_load15", 0.8))

	cs := &CPUStats{ps: &mps}

	cputags := map[string]string{
		"cpu": "cpu0",
	}

	err = cs.Gather(&acc)
	require.NoError(t, err)

	assert.True(t, acc.CheckTaggedValue("cpu_user", 3.1, cputags))
	assert.True(t, acc.CheckTaggedValue("cpu_system", 8.2, cputags))
	assert.True(t, acc.CheckTaggedValue("cpu_idle", 80.1, cputags))
	assert.True(t, acc.CheckTaggedValue("cpu_nice", 1.3, cputags))
	assert.True(t, acc.CheckTaggedValue("cpu_iowait", 0.2, cputags))
	assert.True(t, acc.CheckTaggedValue("cpu_irq", 0.1, cputags))
	assert.True(t, acc.CheckTaggedValue("cpu_softirq", 0.11, cputags))
	assert.True(t, acc.CheckTaggedValue("cpu_steal", 0.0001, cputags))
	assert.True(t, acc.CheckTaggedValue("cpu_guest", 8.1, cputags))
	assert.True(t, acc.CheckTaggedValue("cpu_guestNice", 0.324, cputags))
	assert.True(t, acc.CheckTaggedValue("cpu_stolen", 0.051, cputags))

	err = (&DiskStats{&mps}).Gather(&acc)
	require.NoError(t, err)

	tags := map[string]string{
		"path": "/",
	}

	assert.True(t, acc.CheckTaggedValue("disk_total", uint64(128), tags))
	assert.True(t, acc.CheckTaggedValue("disk_used", uint64(105), tags))
	assert.True(t, acc.CheckTaggedValue("disk_free", uint64(23), tags))
	assert.True(t, acc.CheckTaggedValue("disk_inodes_total", uint64(1234), tags))
	assert.True(t, acc.CheckTaggedValue("disk_inodes_free", uint64(234), tags))
	assert.True(t, acc.CheckTaggedValue("disk_inodes_used", uint64(1000), tags))

	err = (&NetIOStats{&mps}).Gather(&acc)
	require.NoError(t, err)

	ntags := map[string]string{
		"interface": "eth0",
	}

	assert.True(t, acc.CheckTaggedValue("net_bytes_sent", uint64(1123), ntags))
	assert.True(t, acc.CheckTaggedValue("net_bytes_recv", uint64(8734422), ntags))
	assert.True(t, acc.CheckTaggedValue("net_packets_sent", uint64(781), ntags))
	assert.True(t, acc.CheckTaggedValue("net_packets_recv", uint64(23456), ntags))
	assert.True(t, acc.CheckTaggedValue("net_err_in", uint64(832), ntags))
	assert.True(t, acc.CheckTaggedValue("net_err_out", uint64(8), ntags))
	assert.True(t, acc.CheckTaggedValue("net_drop_in", uint64(7), ntags))
	assert.True(t, acc.CheckTaggedValue("net_drop_out", uint64(1), ntags))

	err = (&DiskIOStats{&mps}).Gather(&acc)
	require.NoError(t, err)

	dtags := map[string]string{
		"name":   "sda1",
		"serial": "ab-123-ad",
	}

	assert.True(t, acc.CheckTaggedValue("io_reads", uint64(888), dtags))
	assert.True(t, acc.CheckTaggedValue("io_writes", uint64(5341), dtags))
	assert.True(t, acc.CheckTaggedValue("io_read_bytes", uint64(100000), dtags))
	assert.True(t, acc.CheckTaggedValue("io_write_bytes", uint64(200000), dtags))
	assert.True(t, acc.CheckTaggedValue("io_read_time", uint64(7123), dtags))
	assert.True(t, acc.CheckTaggedValue("io_write_time", uint64(9087), dtags))
	assert.True(t, acc.CheckTaggedValue("io_io_time", uint64(123552), dtags))

	err = (&MemStats{&mps}).Gather(&acc)
	require.NoError(t, err)

	vmtags := map[string]string(nil)

	assert.True(t, acc.CheckTaggedValue("mem_total", uint64(12400), vmtags))
	assert.True(t, acc.CheckTaggedValue("mem_available", uint64(7600), vmtags))
	assert.True(t, acc.CheckTaggedValue("mem_used", uint64(5000), vmtags))
	assert.True(t, acc.CheckTaggedValue("mem_used_prec", float64(47.1), vmtags))
	assert.True(t, acc.CheckTaggedValue("mem_free", uint64(1235), vmtags))
	assert.True(t, acc.CheckTaggedValue("mem_active", uint64(8134), vmtags))
	assert.True(t, acc.CheckTaggedValue("mem_inactive", uint64(1124), vmtags))
	assert.True(t, acc.CheckTaggedValue("mem_buffers", uint64(771), vmtags))
	assert.True(t, acc.CheckTaggedValue("mem_cached", uint64(4312), vmtags))
	assert.True(t, acc.CheckTaggedValue("mem_wired", uint64(134), vmtags))
	assert.True(t, acc.CheckTaggedValue("mem_shared", uint64(2142), vmtags))

	err = (&SwapStats{&mps}).Gather(&acc)
	require.NoError(t, err)

	swaptags := map[string]string(nil)

	assert.True(t, acc.CheckTaggedValue("swap_total", uint64(8123), swaptags))
	assert.True(t, acc.CheckTaggedValue("swap_used", uint64(1232), swaptags))
	assert.True(t, acc.CheckTaggedValue("swap_used_perc", float64(12.2), swaptags))
	assert.True(t, acc.CheckTaggedValue("swap_free", uint64(6412), swaptags))
	assert.True(t, acc.CheckTaggedValue("swap_in", uint64(7), swaptags))
	assert.True(t, acc.CheckTaggedValue("swap_out", uint64(830), swaptags))

	err = (&DockerStats{&mps}).Gather(&acc)
	require.NoError(t, err)

	dockertags := map[string]string{
		"id": "blah",
	}

	assert.True(t, acc.CheckTaggedValue("docker_user", 3.1, dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_system", 8.2, dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_idle", 80.1, dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_nice", 1.3, dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_iowait", 0.2, dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_irq", 0.1, dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_softirq", 0.11, dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_steal", 0.0001, dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_guest", 8.1, dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_guestNice", 0.324, dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_stolen", 0.051, dockertags))

	assert.True(t, acc.CheckTaggedValue("docker_cache", uint64(1), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_rss", uint64(2), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_rss_huge", uint64(3), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_mapped_file", uint64(4), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_swap_in", uint64(5), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_swap_out", uint64(6), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_page_fault", uint64(7), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_page_major_fault", uint64(8), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_inactive_anon", uint64(9), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_active_anon", uint64(10), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_inactive_file", uint64(11), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_active_file", uint64(12), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_unevictable", uint64(13), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_memory_limit", uint64(14), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_total_cache", uint64(15), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_total_rss", uint64(16), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_total_rss_huge", uint64(17), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_total_mapped_file", uint64(18), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_total_swap_in", uint64(19), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_total_swap_out", uint64(20), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_total_page_fault", uint64(21), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_total_page_major_fault", uint64(22), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_total_inactive_anon", uint64(23), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_total_active_anon", uint64(24), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_total_inactive_file", uint64(25), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_total_active_file", uint64(26), dockertags))
	assert.True(t, acc.CheckTaggedValue("docker_total_unevictable", uint64(27), dockertags))
}