75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
|
package overwatch
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/md5"
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
type mockService struct {
|
||
|
|
serviceName string
|
||
|
|
serviceType string
|
||
|
|
runError error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *mockService) Name() string {
|
||
|
|
return s.serviceName
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *mockService) Type() string {
|
||
|
|
return s.serviceType
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *mockService) Hash() string {
|
||
|
|
h := md5.New()
|
||
|
|
io.WriteString(h, s.serviceName)
|
||
|
|
io.WriteString(h, s.serviceType)
|
||
|
|
return fmt.Sprintf("%x", h.Sum(nil))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *mockService) Shutdown() {
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *mockService) Run() error {
|
||
|
|
return s.runError
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestManagerAddAndRemove(t *testing.T) {
|
||
|
|
m := NewAppManager(nil)
|
||
|
|
|
||
|
|
first := &mockService{serviceName: "first", serviceType: "mock"}
|
||
|
|
second := &mockService{serviceName: "second", serviceType: "mock"}
|
||
|
|
m.Add(first)
|
||
|
|
m.Add(second)
|
||
|
|
assert.Len(t, m.Services(), 2, "expected 2 services in the list")
|
||
|
|
|
||
|
|
m.Remove(first.Name())
|
||
|
|
services := m.Services()
|
||
|
|
assert.Len(t, services, 1, "expected 1 service in the list")
|
||
|
|
assert.Equal(t, second.Hash(), services[0].Hash(), "hashes should match. Wrong service was removed")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestManagerDuplicate(t *testing.T) {
|
||
|
|
m := NewAppManager(nil)
|
||
|
|
|
||
|
|
first := &mockService{serviceName: "first", serviceType: "mock"}
|
||
|
|
m.Add(first)
|
||
|
|
m.Add(first)
|
||
|
|
assert.Len(t, m.Services(), 1, "expected 1 service in the list")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestManagerErrorChannel(t *testing.T) {
|
||
|
|
errChan := make(chan error)
|
||
|
|
m := NewAppManager(errChan)
|
||
|
|
|
||
|
|
err := errors.New("test error")
|
||
|
|
first := &mockService{serviceName: "first", serviceType: "mock", runError: err}
|
||
|
|
m.Add(first)
|
||
|
|
respErr := <-errChan
|
||
|
|
assert.Equal(t, err, respErr, "errors don't match")
|
||
|
|
}
|