add context fixture

Author: rekby

sf/context.go

@@ -0,0 +1,13 @@
+package sf
+
+import (
+	"context"
+	"github.com/rekby/fixenv"
+)
+
+func Context(e fixenv.Env) context.Context {
+	return fixenv.CacheWithCleanup(e, nil, nil, func() (context.Context, fixenv.FixtureCleanupFunc, error) {
+		ctx, ctxCancel := context.WithCancel(context.Background())
+		return ctx, fixenv.FixtureCleanupFunc(ctxCancel), nil
+	})
+}

sf/context_test.go

@@ -0,0 +1,23 @@
+package sf
+
+import (
+	"context"
+	"github.com/rekby/fixenv"
+	"github.com/rekby/fixenv/internal"
+	"testing"
+)
+
+func TestContext(t *testing.T) {
+	tm := &internal.TestMock{}
+
+	e := fixenv.New(tm)
+	ctx := Context(e)
+	if ctx.Err() != nil {
+		t.Fatal(ctx.Err())
+	}
+
+	tm.CallCleanup()
+	if ctx.Err() != context.Canceled {
+		t.Fatal(ctx.Err())
+	}
+}