135 lines
3.1 KiB
Go
135 lines
3.1 KiB
Go
package marathon
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gambol99/go-marathon"
|
|
)
|
|
|
|
func testReadinessChecker() *readinessChecker {
|
|
return defaultReadinessChecker(false)
|
|
}
|
|
|
|
func TestDisabledReadinessChecker(t *testing.T) {
|
|
var rc *readinessChecker
|
|
tsk := task()
|
|
app := application(
|
|
deployments("deploymentId"),
|
|
readinessCheck(0),
|
|
readinessCheckResult(testTaskName, false),
|
|
)
|
|
|
|
if ready := rc.Do(tsk, app); !ready {
|
|
t.Error("expected ready = true")
|
|
}
|
|
}
|
|
|
|
func TestEnabledReadinessChecker(t *testing.T) {
|
|
tests := []struct {
|
|
desc string
|
|
task marathon.Task
|
|
app marathon.Application
|
|
rc readinessChecker
|
|
expectedReady bool
|
|
}{
|
|
{
|
|
desc: "no deployment running",
|
|
task: task(),
|
|
app: application(),
|
|
expectedReady: true,
|
|
},
|
|
{
|
|
desc: "no readiness checks defined",
|
|
task: task(),
|
|
app: application(deployments("deploymentId")),
|
|
expectedReady: true,
|
|
},
|
|
{
|
|
desc: "readiness check result negative",
|
|
task: task(),
|
|
app: application(
|
|
deployments("deploymentId"),
|
|
readinessCheck(0),
|
|
readinessCheckResult("otherTaskID", true),
|
|
readinessCheckResult(testTaskName, false),
|
|
),
|
|
expectedReady: false,
|
|
},
|
|
{
|
|
desc: "readiness check result positive",
|
|
task: task(),
|
|
app: application(
|
|
deployments("deploymentId"),
|
|
readinessCheck(0),
|
|
readinessCheckResult("otherTaskID", false),
|
|
readinessCheckResult(testTaskName, true),
|
|
),
|
|
expectedReady: true,
|
|
},
|
|
{
|
|
desc: "no readiness check result with default timeout",
|
|
task: task(startedAtFromNow(3 * time.Minute)),
|
|
app: application(
|
|
deployments("deploymentId"),
|
|
readinessCheck(0),
|
|
),
|
|
rc: readinessChecker{
|
|
checkDefaultTimeout: 5 * time.Minute,
|
|
},
|
|
expectedReady: false,
|
|
},
|
|
{
|
|
desc: "no readiness check result with readiness check timeout",
|
|
task: task(startedAtFromNow(4 * time.Minute)),
|
|
app: application(
|
|
deployments("deploymentId"),
|
|
readinessCheck(3*time.Minute),
|
|
),
|
|
rc: readinessChecker{
|
|
checkSafetyMargin: 3 * time.Minute,
|
|
},
|
|
expectedReady: false,
|
|
},
|
|
{
|
|
desc: "invalid task start time",
|
|
task: task(startedAt("invalid")),
|
|
app: application(
|
|
deployments("deploymentId"),
|
|
readinessCheck(0),
|
|
),
|
|
expectedReady: false,
|
|
},
|
|
{
|
|
desc: "task not involved in deployment",
|
|
task: task(startedAtFromNow(1 * time.Hour)),
|
|
app: application(
|
|
deployments("deploymentId"),
|
|
readinessCheck(0),
|
|
),
|
|
rc: readinessChecker{
|
|
checkDefaultTimeout: 10 * time.Second,
|
|
},
|
|
expectedReady: true,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
test := test
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
rc := testReadinessChecker()
|
|
if test.rc.checkDefaultTimeout > 0 {
|
|
rc.checkDefaultTimeout = test.rc.checkDefaultTimeout
|
|
}
|
|
if test.rc.checkSafetyMargin > 0 {
|
|
rc.checkSafetyMargin = test.rc.checkSafetyMargin
|
|
}
|
|
actualReady := test.rc.Do(test.task, test.app)
|
|
if actualReady != test.expectedReady {
|
|
t.Errorf("actual ready = %t, expected ready = %t", actualReady, test.expectedReady)
|
|
}
|
|
})
|
|
}
|
|
}
|