You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
1.5 KiB
67 lines
1.5 KiB
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})
|
|
}
|
|
|