43 lines
927 B
Go
43 lines
927 B
Go
package sandbox
|
|
|
|
import (
|
|
"math"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/joint-online-judge/JOJ3/internal/stage"
|
|
)
|
|
|
|
func TestExecRPCTimeout(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
cmds []stage.Cmd
|
|
want time.Duration
|
|
}{
|
|
{name: "margin only", want: rpcTimeoutMargin},
|
|
{
|
|
name: "largest clock limit",
|
|
cmds: []stage.Cmd{{ClockLimit: uint64(time.Minute)}, {ClockLimit: uint64(2 * time.Minute)}},
|
|
want: 2*time.Minute + rpcTimeoutMargin,
|
|
},
|
|
{
|
|
name: "cpu limit fallback",
|
|
cmds: []stage.Cmd{{CPULimit: uint64(time.Minute)}},
|
|
want: 2*time.Minute + rpcTimeoutMargin,
|
|
},
|
|
{
|
|
name: "duration overflow",
|
|
cmds: []stage.Cmd{{ClockLimit: math.MaxUint64}},
|
|
want: time.Duration(math.MaxInt64),
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := execRPCTimeout(tt.cmds); got != tt.want {
|
|
t.Fatalf("execRPCTimeout() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|