Cache ContextDoneVars

Note that nothing currently flushes them. Could probably flush them when they're done, and the early return will take care of the rest.

1 files changed, 20 insertions, 0 deletions

diff --git a/stmutil/context_test.go b/stmutil/context_test.go
new file mode 100644
index 0000000..0a6b7c0
--- /dev/null
+++ b/stmutil/context_test.go
@@ -0,0 +1,20 @@
+package stmutil
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestContextEquality(t *testing.T) {
+	ctx := context.Background()
+	assert.True(t, ctx == context.Background())
+	childCtx, cancel := context.WithCancel(ctx)
+	assert.True(t, childCtx != ctx)
+	assert.True(t, childCtx != ctx)
+	assert.Equal(t, context.Background(), ctx)
+	cancel()
+	assert.Equal(t, context.Background(), ctx)
+	assert.NotEqual(t, ctx, childCtx)
+}