package hddtemp

import (
	"net"
	"reflect"
	"testing"
)

func TestFetch(t *testing.T) {
	l := serve(t, []byte("|/dev/sda|foobar|36|C|"))
	defer l.Close()

	disks, err := Fetch(l.Addr().String())

	if err != nil {
		t.Error("expecting err to be nil")
	}

	expected := []disk{
		{
			DeviceName:  "sda",
			Model:       "foobar",
			Temperature: 36,
			Unit:        "C",
		},
	}

	if !reflect.DeepEqual(expected, disks) {
		t.Error("disks' slice is different from expected")
	}
}

func TestFetchWrongAddress(t *testing.T) {
	_, err := Fetch("127.0.0.1:1")

	if err == nil {
		t.Error("expecting err to be non-nil")
	}
}

func TestFetchStatus(t *testing.T) {
	l := serve(t, []byte("|/dev/sda|foobar|SLP|C|"))
	defer l.Close()

	disks, err := Fetch(l.Addr().String())

	if err != nil {
		t.Error("expecting err to be nil")
	}

	expected := []disk{
		{
			DeviceName:  "sda",
			Model:       "foobar",
			Temperature: 0,
			Unit:        "C",
			Status:      "SLP",
		},
	}

	if !reflect.DeepEqual(expected, disks) {
		t.Error("disks' slice is different from expected")
	}
}

func TestFetchTwoDisks(t *testing.T) {
	l := serve(t, []byte("|/dev/hda|ST380011A|46|C||/dev/hdd|ST340016A|SLP|*|"))
	defer l.Close()

	disks, err := Fetch(l.Addr().String())

	if err != nil {
		t.Error("expecting err to be nil")
	}

	expected := []disk{
		{
			DeviceName:  "hda",
			Model:       "ST380011A",
			Temperature: 46,
			Unit:        "C",
		},
		{
			DeviceName:  "hdd",
			Model:       "ST340016A",
			Temperature: 0,
			Unit:        "*",
			Status:      "SLP",
		},
	}

	if !reflect.DeepEqual(expected, disks) {
		t.Error("disks' slice is different from expected")
	}
}

func serve(t *testing.T, data []byte) net.Listener {
	l, err := net.Listen("tcp", "127.0.0.1:0")

	if err != nil {
		t.Fatal(err)
	}

	go func(t *testing.T) {
		conn, err := l.Accept()

		if err != nil {
			t.Fatal(err)
		}

		conn.Write(data)
		conn.Close()
	}(t)

	return l
}