htwkmooc/poseidon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor NewAbstractManager

Browse Source 
to require a context used for the monitoring.
This commit is contained in:
Maximilian Paß
2023-09-04 11:44:36 +02:00
parent b28b87d56f
commit 0d6b4f660c
7 changed files with 34 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cmd/poseidon/main.go
View File

@@ -169,8 +169,9 @@ func createNomadManager(ctx context.Context) (
return runnerManager, environmentManager
}
func createAWSManager(_ context.Context) (runnerManager runner.Manager, environmentManager environment.ManagerHandler) {
runnerManager = runner.NewAWSRunnerManager()
func createAWSManager(ctx context.Context) (
runnerManager runner.Manager, environmentManager environment.ManagerHandler) {
runnerManager = runner.NewAWSRunnerManager(ctx)
return runnerManager, environment.NewAWSEnvironmentManager(runnerManager)
}

4
cmd/poseidon/main_test.go
View File

@@ -17,7 +17,7 @@ func TestAWSDisabledUsesNomadManager(t *testing.T) {
cancel()
runnerManager, environmentManager := createManagerHandler(createNomadManager, true,
runner.NewAbstractManager(), &environment.AbstractManager{}, disableRecovery)
runner.NewAbstractManager(disableRecovery), &environment.AbstractManager{}, disableRecovery)
awsRunnerManager, awsEnvironmentManager := createManagerHandler(createAWSManager, false,
runnerManager, environmentManager, disableRecovery)
assert.Equal(t, runnerManager, awsRunnerManager)
@@ -29,7 +29,7 @@ func TestAWSEnabledWrappesNomadManager(t *testing.T) {
cancel()
runnerManager, environmentManager := createManagerHandler(createNomadManager, true,
runner.NewAbstractManager(), &environment.AbstractManager{}, disableRecovery)
runner.NewAbstractManager(disableRecovery), &environment.AbstractManager{}, disableRecovery)
awsRunnerManager, awsEnvironmentManager := createManagerHandler(createAWSManager,
true, runnerManager, environmentManager, disableRecovery)
assert.NotEqual(t, runnerManager, awsRunnerManager)