package gobot

import (
	"errors"
	"log"
	"os"
	"testing"

	"github.com/hybridgroup/gobot/gobottest"
)

func TestConnectionEach(t *testing.T) {
	r := newTestRobot("Robot1")

	i := 0
	r.Connections().Each(func(conn Connection) {
		i++
	})
	gobottest.Assert(t, r.Connections().Len(), i)
}

func initTestMaster() *Master {
	log.SetOutput(&NullReadWriteCloser{})
	g := NewMaster()
	g.trap = func(c chan os.Signal) {
		c <- os.Interrupt
	}
	g.AddRobot(newTestRobot("Robot1"))
	g.AddRobot(newTestRobot("Robot2"))
	g.AddRobot(newTestRobot(""))
	return g
}

func initTestMaster1Robot() *Master {
	log.SetOutput(&NullReadWriteCloser{})
	g := NewMaster()
	g.trap = func(c chan os.Signal) {
		c <- os.Interrupt
	}
	g.AddRobot(newTestRobot("Robot99"))

	return g
}

func TestVersion(t *testing.T) {
	gobottest.Assert(t, version, Version())
}

func TestNullReadWriteCloser(t *testing.T) {
	n := &NullReadWriteCloser{}
	i, _ := n.Write([]byte{1, 2, 3})
	gobottest.Assert(t, i, 3)
	i, _ = n.Read(make([]byte, 10))
	gobottest.Assert(t, i, 10)
	gobottest.Assert(t, n.Close(), nil)
}

func TestGobotRobot(t *testing.T) {
	g := initTestMaster()
	gobottest.Assert(t, g.Robot("Robot1").Name, "Robot1")
	gobottest.Assert(t, g.Robot("Robot4"), (*Robot)(nil))
	gobottest.Assert(t, g.Robot("Robot4").Device("Device1"), (Device)(nil))
	gobottest.Assert(t, g.Robot("Robot4").Connection("Connection1"), (Connection)(nil))
	gobottest.Assert(t, g.Robot("Robot1").Device("Device4"), (Device)(nil))
	gobottest.Assert(t, g.Robot("Robot1").Device("Device1").Name(), "Device1")
	gobottest.Assert(t, g.Robot("Robot1").Devices().Len(), 3)
	gobottest.Assert(t, g.Robot("Robot1").Connection("Connection4"), (Connection)(nil))
	gobottest.Assert(t, g.Robot("Robot1").Connections().Len(), 3)
}

func TestGobotToJSON(t *testing.T) {
	g := initTestMaster()
	g.AddCommand("test_function", func(params map[string]interface{}) interface{} {
		return nil
	})
	json := NewJSONMaster(g)
	gobottest.Assert(t, len(json.Robots), g.Robots().Len())
	gobottest.Assert(t, len(json.Commands), len(g.Commands()))
}

func TestMasterStart(t *testing.T) {
	g := initTestMaster()
	gobottest.Assert(t, len(g.Start()), 0)
	gobottest.Assert(t, len(g.Stop()), 0)
}

func TestMasterStartDriverErrors(t *testing.T) {
	g := initTestMaster1Robot()

	testDriverStart = func() (errs []error) {
		return []error{
			errors.New("driver start error 1"),
		}
	}

	gobottest.Assert(t, len(g.Start()), 1)
	gobottest.Assert(t, len(g.Stop()), 0)

	testDriverStart = func() (errs []error) { return }
}

func TestMasterStartAdaptorErrors(t *testing.T) {
	g := initTestMaster1Robot()

	testAdaptorConnect = func() (errs []error) {
		return []error{
			errors.New("adaptor start error 1"),
		}
	}

	gobottest.Assert(t, len(g.Start()), 1)
	gobottest.Assert(t, len(g.Stop()), 0)

	testAdaptorConnect = func() (errs []error) { return }
}

func TestMasterHaltErrors(t *testing.T) {
	g := initTestMaster1Robot()

	testDriverHalt = func() (errs []error) {
		return []error{
			errors.New("driver halt error 1"),
		}
	}

	testAdaptorFinalize = func() (errs []error) {
		return []error{
			errors.New("adaptor finalize error 1"),
		}
	}

	gobottest.Assert(t, len(g.Start()), 0)
	gobottest.Assert(t, len(g.Stop()), 6)
}