You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
648 B
Go
28 lines
648 B
Go
4 years ago
|
package contextx
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestShrinkDeadlineLess(t *testing.T) {
|
||
|
|
deadline := time.Now().Add(time.Second)
|
||
|
|
ctx, _ := context.WithDeadline(context.Background(), deadline)
|
||
|
|
ctx, _ = ShrinkDeadline(ctx, time.Minute)
|
||
|
|
dl, ok := ctx.Deadline()
|
||
|
|
assert.True(t, ok)
|
||
|
|
assert.Equal(t, deadline, dl)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestShrinkDeadlineMore(t *testing.T) {
|
||
|
|
deadline := time.Now().Add(time.Minute)
|
||
|
|
ctx, _ := context.WithDeadline(context.Background(), deadline)
|
||
|
|
ctx, _ = ShrinkDeadline(ctx, time.Second)
|
||
|
|
dl, ok := ctx.Deadline()
|
||
|
|
assert.True(t, ok)
|
||
|
|
assert.True(t, dl.Before(deadline))
|
||
|
|
}
|