80 lines
2.3 KiB
Go
80 lines
2.3 KiB
Go
package runner
|
|
|
|
import (
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/sirupsen/logrus/hooks/test"
|
|
"github.com/stretchr/testify/suite"
|
|
"testing"
|
|
)
|
|
|
|
type DummyEntity struct{}
|
|
|
|
func (DummyEntity) Id() string {
|
|
return ""
|
|
}
|
|
|
|
func TestRunnerPoolTestSuite(t *testing.T) {
|
|
suite.Run(t, new(RunnerPoolTestSuite))
|
|
}
|
|
|
|
type RunnerPoolTestSuite struct {
|
|
suite.Suite
|
|
runnerPool *localRunnerPool
|
|
runner Runner
|
|
}
|
|
|
|
func (suite *RunnerPoolTestSuite) SetupTest() {
|
|
suite.runnerPool = NewLocalRunnerPool()
|
|
suite.runner = CreateTestRunner()
|
|
}
|
|
|
|
func (suite *RunnerPoolTestSuite) TestAddInvalidEntityTypeThrowsFatal() {
|
|
var hook *test.Hook
|
|
logger, hook := test.NewNullLogger()
|
|
// don't terminate program on fatal log entry
|
|
logger.ExitFunc = func(int) {}
|
|
log = logger.WithField("pkg", "environment")
|
|
|
|
dummyEntity := DummyEntity{}
|
|
suite.runnerPool.Add(dummyEntity)
|
|
suite.Equal(logrus.FatalLevel, hook.LastEntry().Level)
|
|
suite.Equal(dummyEntity, hook.LastEntry().Data["entity"])
|
|
}
|
|
|
|
func (suite *RunnerPoolTestSuite) TestAddValidEntityThrowsFatal() {
|
|
var hook *test.Hook
|
|
logger, hook := test.NewNullLogger()
|
|
log = logger.WithField("pkg", "environment")
|
|
|
|
suite.runnerPool.Add(suite.runner)
|
|
// currently, the Add method does not log anything else. adjust if necessary
|
|
suite.Nil(hook.LastEntry())
|
|
}
|
|
|
|
func (suite *RunnerPoolTestSuite) TestAddedRunnerCanBeRetrieved() {
|
|
suite.runnerPool.Add(suite.runner)
|
|
retrievedRunner, ok := suite.runnerPool.Get(suite.runner.Id())
|
|
suite.True(ok, "A saved runner should be retrievable")
|
|
suite.Equal(suite.runner, retrievedRunner)
|
|
}
|
|
|
|
func (suite *RunnerPoolTestSuite) TestRunnerWithSameIdOverwritesOldOne() {
|
|
otherRunnerWithSameId := NewRunner(suite.runner.Id())
|
|
// assure runner is actually different
|
|
suite.NotEqual(suite.runner, otherRunnerWithSameId)
|
|
|
|
suite.runnerPool.Add(suite.runner)
|
|
suite.runnerPool.Add(otherRunnerWithSameId)
|
|
retrievedRunner, _ := suite.runnerPool.Get(suite.runner.Id())
|
|
suite.NotEqual(suite.runner, retrievedRunner)
|
|
suite.Equal(otherRunnerWithSameId, retrievedRunner)
|
|
}
|
|
|
|
func (suite *RunnerPoolTestSuite) TestDeletedRunnersAreNotAccessible() {
|
|
suite.runnerPool.Add(suite.runner)
|
|
suite.runnerPool.Delete(suite.runner.Id())
|
|
retrievedRunner, ok := suite.runnerPool.Get(suite.runner.Id())
|
|
suite.Nil(retrievedRunner)
|
|
suite.False(ok, "A deleted runner should not be accessible")
|
|
}
|