Add EntityStore interface

environment/runner_pool_test.go

@@ -0,0 +1,66 @@
+package environment
+
+import (
+	"github.com/stretchr/testify/suite"
+	"gitlab.hpi.de/codeocean/codemoon/poseidon/runner"
+	"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.Runner
+}
+
+func (suite *RunnerPoolTestSuite) SetupTest() {
+	suite.runnerPool = NewLocalRunnerPool()
+	suite.runner = CreateTestRunner()
+}
+
+func (suite *RunnerPoolTestSuite) TestAddInvalidEntityTypeReturnsError() {
+	dummyEntity := DummyEntity{}
+	err := suite.runnerPool.Add(dummyEntity)
+	suite.Error(err)
+}
+
+func (suite *RunnerPoolTestSuite) TestAddValidEntityReturnsNoError() {
+	err := suite.runnerPool.Add(suite.runner)
+	suite.NoError(err)
+}
+
+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 := runner.NewExerciseRunner(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")
+}