102 lines
3.1 KiB
Go
102 lines
3.1 KiB
Go
package runner
|
|
|
|
import (
|
|
"context"
|
|
"github.com/openHPI/poseidon/pkg/dto"
|
|
"github.com/openHPI/poseidon/tests"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
"testing"
|
|
)
|
|
|
|
func TestAWSRunnerManager_EnvironmentAccessor(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
m := NewAWSRunnerManager(ctx)
|
|
|
|
environments := m.ListEnvironments()
|
|
assert.Empty(t, environments)
|
|
|
|
environment := createBasicEnvironmentMock(defaultEnvironmentID)
|
|
m.StoreEnvironment(environment)
|
|
|
|
environments = m.ListEnvironments()
|
|
assert.Len(t, environments, 1)
|
|
assert.Equal(t, environments[0].ID(), dto.EnvironmentID(tests.DefaultEnvironmentIDAsInteger))
|
|
|
|
e, ok := m.GetEnvironment(tests.DefaultEnvironmentIDAsInteger)
|
|
assert.True(t, ok)
|
|
assert.Equal(t, environment, e)
|
|
|
|
_, ok = m.GetEnvironment(tests.AnotherEnvironmentIDAsInteger)
|
|
assert.False(t, ok)
|
|
}
|
|
|
|
func TestAWSRunnerManager_Claim(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
m := NewAWSRunnerManager(ctx)
|
|
environment := createBasicEnvironmentMock(defaultEnvironmentID)
|
|
r, err := NewAWSFunctionWorkload(environment, nil)
|
|
assert.NoError(t, err)
|
|
environment.On("Sample").Return(r, true)
|
|
m.StoreEnvironment(environment)
|
|
|
|
t.Run("returns runner for AWS environment", func(t *testing.T) {
|
|
r, err := m.Claim(tests.DefaultEnvironmentIDAsInteger, 60)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, r)
|
|
})
|
|
|
|
t.Run("forwards request for non-AWS environments", func(t *testing.T) {
|
|
nextHandler := &ManagerMock{}
|
|
nextHandler.On("Claim", mock.AnythingOfType("dto.EnvironmentID"), mock.AnythingOfType("int")).
|
|
Return(nil, nil)
|
|
m.SetNextHandler(nextHandler)
|
|
|
|
_, err := m.Claim(tests.AnotherEnvironmentIDAsInteger, 60)
|
|
assert.Nil(t, err)
|
|
nextHandler.AssertCalled(t, "Claim", dto.EnvironmentID(tests.AnotherEnvironmentIDAsInteger), 60)
|
|
})
|
|
}
|
|
|
|
func TestAWSRunnerManager_Return(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
m := NewAWSRunnerManager(ctx)
|
|
environment := createBasicEnvironmentMock(defaultEnvironmentID)
|
|
m.StoreEnvironment(environment)
|
|
r, err := NewAWSFunctionWorkload(environment, nil)
|
|
assert.NoError(t, err)
|
|
|
|
t.Run("removes usedRunner", func(t *testing.T) {
|
|
m.usedRunners.Add(r.ID(), r)
|
|
assert.Contains(t, m.usedRunners.List(), r)
|
|
|
|
err := m.Return(r)
|
|
assert.NoError(t, err)
|
|
assert.NotContains(t, m.usedRunners.List(), r)
|
|
})
|
|
|
|
t.Run("calls nextHandler for non-AWS runner", func(t *testing.T) {
|
|
nextHandler := &ManagerMock{}
|
|
nextHandler.On("Return", mock.AnythingOfType("*runner.NomadJob")).Return(nil)
|
|
m.SetNextHandler(nextHandler)
|
|
|
|
nonAWSRunner := NewNomadJob(tests.DefaultRunnerID, nil, nil, nil)
|
|
err := m.Return(nonAWSRunner)
|
|
assert.NoError(t, err)
|
|
nextHandler.AssertCalled(t, "Return", nonAWSRunner)
|
|
})
|
|
}
|
|
|
|
func createBasicEnvironmentMock(id dto.EnvironmentID) *ExecutionEnvironmentMock {
|
|
environment := &ExecutionEnvironmentMock{}
|
|
environment.On("ID").Return(id)
|
|
environment.On("Image").Return("")
|
|
environment.On("CPULimit").Return(uint(0))
|
|
environment.On("MemoryLimit").Return(uint(0))
|
|
environment.On("NetworkAccess").Return(false, nil)
|
|
return environment
|
|
}
|