woop/api/service/manager_test.go

package service

import (
	"fmt"
	"testing"
	"time"

	msg_pb "github.com/harmony-one/harmony/api/proto/message"
)

type SupportSyncingTest struct {
	msgChan chan *msg_pb.Message
}

func (s *SupportSyncingTest) StartService() {
	fmt.Println("SupportSyncingTest starting")
}

func (s *SupportSyncingTest) StopService() {
	fmt.Println("SupportSyncingTest stopping")
}

func (s *SupportSyncingTest) NotifyService(data map[string]interface{}) {
	t := data["test"].(*testing.T)

	if s.msgChan != data["chan"].(chan *msg_pb.Message) {
		t.Error("message chan is not equal to the one from the passing params")
	}
}
func (s *SupportSyncingTest) SetMessageChan(msgChan chan *msg_pb.Message) {
	s.msgChan = msgChan
}

// Test TakeAction.
func TestTakeAction(t *testing.T) {
	m := &Manager{}
	m.SetupServiceManager()
	m.RegisterService(SupportSyncing, &SupportSyncingTest{})

	for i := 0; i < 2; i++ {
		select {
		case <-time.After(100 * time.Millisecond):
			m.SendAction(&Action{Action: Start, ServiceType: SupportSyncing})
		}
	}

	m.SendAction(&Action{ServiceType: Done})
}

func TestMessageChan(t *testing.T) {
	m := &Manager{}
	m.SetupServiceManager()
	m.RegisterService(SupportSyncing, &SupportSyncingTest{})
	msgChans := make(map[Type]chan *msg_pb.Message)
	m.SetupServiceMessageChan(msgChans)

	m.SendAction(&Action{
		Action:      Notify,
		ServiceType: SupportSyncing,
		Params: map[string]interface{}{
			"chan": msgChans[SupportSyncing],
			"test": t,
		},
	})

	m.SendAction(&Action{ServiceType: Done})
}