bstore/context_test.go at master · mjl-/bstore · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
package bstore

import (
	"context"
	"os"
	"testing"
	"time"

	bolt "go.etcd.io/bbolt"
)

func TestContext(t *testing.T) {
	type X struct {
		ID int
	}

	const path = "testdata/tmp.context.db"
	os.Remove(path)
	db, err := topen(t, path, nil, X{})
	tcheck(t, err, "open")

	err = db.Insert(ctxbg, &X{}, &X{})
	tcheck(t, err, "insert values")

	canceled, cancel := context.WithCancel(context.Background())
	cancel()

	// Early cancel, don't start transaction.
	_, err = db.Begin(canceled, false)
	tneed(t, err, context.Canceled, "begin canceled")

	// Cancel during transaction.
	ctx, cancel := context.WithCancel(context.Background())
	tx, err := db.Begin(ctx, false)
	tcheck(t, err, "begin")
	cancel()
	err = tx.Get(&X{ID: 1})
	tneed(t, err, context.Canceled, "begin and canceled get")
	tx.Rollback()

	// Cancel before transaction.
	err = db.Read(canceled, func(tx *Tx) error {
		return nil
	})
	tneed(t, err, context.Canceled, "read tx canceled")

	// Cancel during transaction.
	ctx, cancel = context.WithCancel(context.Background())
	err = db.Read(ctx, func(tx *Tx) error {
		cancel()
		return tx.Get(&X{ID: 1})
	})
	tneed(t, err, context.Canceled, "read tx and canceled get")

	// Canceled Get
	err = db.Get(canceled, &X{ID: 1})
	tneed(t, err, context.Canceled, "canceled db get")

	// Canceled DB Query
	q := QueryDB[X](canceled, db)
	err = q.Err()
	tneed(t, err, context.Canceled, "canceled db query")

	// DB Query with canceled List
	ctx, cancel = context.WithCancel(context.Background())
	q = QueryDB[X](ctx, db)
	err = q.Err()
	tcheck(t, err, "query error")
	cancel()
	_, err = q.List()
	tneed(t, err, context.Canceled, "canceled db query list")

	// DB Query with cancel during ForEach.
	ctx, cancel = context.WithCancel(context.Background())
	q = QueryDB[X](ctx, db)
	err = q.Err()
	tcheck(t, err, "query error")
	err = q.ForEach(func(x X) error {
		cancel()
		return nil
	})
	tneed(t, err, context.Canceled, "canceled db query foreach")

	// DB Query with cancel after Next.
	ctx, cancel = context.WithCancel(context.Background())
	q = QueryDB[X](ctx, db)
	err = q.Err()
	tcheck(t, err, "query error")
	var id int
	err = q.NextID(&id)
	tcheck(t, err, "next id")
	cancel()
	err = q.NextID(&id)
	tneed(t, err, context.Canceled, "db query next id after cancel")
	q.Close()

	// Canceled Tx Query
	err = db.Read(canceled, func(tx *Tx) error {
		q := QueryTx[X](tx)
		return q.Err()
	})
	tneed(t, err, context.Canceled, "canceled tx query")

	// Tx Query with cancel.
	ctx, cancel = context.WithCancel(context.Background())
	err = db.Read(ctx, func(tx *Tx) error {
		q := QueryTx[X](tx)
		err = q.Err()
		tcheck(t, err, "query error")
		cancel()
		_, err = q.List()
		return err
	})
	tneed(t, err, context.Canceled, "canceled tx query list")

	// Open with context with timeout.
	now := time.Now()
	ctx, cancel = context.WithTimeout(ctxbg, 0)
	defer cancel()
	_, err = Open(ctx, path, nil, X{})
	tneed(t, err, bolt.ErrTimeout, "open with context with exceeded deadline")
	if time.Since(now) > time.Second {
		t.Fatalf("timeout took more than 1 second, should be immediate")
	}

	tclose(t, db)
}