61 lines
1.6 KiB
Go
61 lines
1.6 KiB
Go
package scaleset
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGenerateJitRunnerConfig(t *testing.T) {
|
|
ctx := context.Background()
|
|
auth := &ActionsAuth{
|
|
Token: "token",
|
|
}
|
|
|
|
t.Run("Get JIT Config for Runner", func(t *testing.T) {
|
|
want := &RunnerScaleSetJitRunnerConfig{}
|
|
response := []byte(`{"count":1,"value":[{"id":1,"name":"scale-set-name"}]}`)
|
|
|
|
runnerSettings := &RunnerScaleSetJitRunnerSetting{}
|
|
|
|
server := newActionsServer(t, http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
|
w.Write(response)
|
|
}))
|
|
client, err := NewClient(server.configURLForOrg("my-org"), auth)
|
|
require.NoError(t, err)
|
|
|
|
got, err := client.GenerateJitRunnerConfig(ctx, runnerSettings, 1)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, want, got)
|
|
})
|
|
|
|
t.Run("Default retries on server error", func(t *testing.T) {
|
|
runnerSettings := &RunnerScaleSetJitRunnerSetting{}
|
|
|
|
retryMax := 1
|
|
actualRetry := 0
|
|
expectedRetry := retryMax + 1
|
|
|
|
server := newActionsServer(t, http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
|
w.WriteHeader(http.StatusServiceUnavailable)
|
|
actualRetry++
|
|
}))
|
|
|
|
client, err := NewClient(
|
|
server.configURLForOrg("my-org"),
|
|
auth,
|
|
WithRetryMax(1),
|
|
WithRetryWaitMax(1*time.Millisecond),
|
|
)
|
|
require.NoError(t, err)
|
|
|
|
_, err = client.GenerateJitRunnerConfig(ctx, runnerSettings, 1)
|
|
assert.NotNil(t, err)
|
|
assert.Equalf(t, actualRetry, expectedRetry, "A retry was expected after the first request but got: %v", actualRetry)
|
|
})
|
|
}
|