-
Notifications
You must be signed in to change notification settings - Fork 15
/
Copy pathtransaction_test.go
117 lines (86 loc) · 2.99 KB
/
transaction_test.go
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
package lungo
import (
"testing"
"time"
"github.com/stretchr/testify/assert"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"github.com/256dpi/lungo/bsonkit"
)
func TestTransactionOplogCleaningBySize(t *testing.T) {
txn := NewTransaction(NewCatalog())
/* prepare */
id1 := primitive.NewObjectID()
_, err := txn.Insert(Handle{"foo", "bar"}, bsonkit.List{
bsonkit.MustConvert(bson.M{
"_id": id1,
"foo": "bar",
}),
}, true)
assert.NoError(t, err)
_, err = txn.Update(Handle{"foo", "bar"}, bsonkit.MustConvert(bson.M{
"_id": id1,
}), nil, bsonkit.MustConvert(bson.M{
"$set": bson.M{
"foo": "baz",
},
}), 0, 0, false, nil)
assert.NoError(t, err)
_, err = txn.Delete(Handle{"foo", "bar"}, bsonkit.MustConvert(bson.M{
"_id": id1,
}), nil, 0, 0)
assert.NoError(t, err)
assert.Len(t, txn.Catalog().Namespaces[Oplog].Documents.List, 3)
insert := txn.Catalog().Namespaces[Oplog].Documents.List[0]
update := txn.Catalog().Namespaces[Oplog].Documents.List[1]
delete := txn.Catalog().Namespaces[Oplog].Documents.List[2]
/* clean */
txn.Clean(3, 0, 0, time.Hour)
assert.Equal(t, bsonkit.List{insert, update, delete}, txn.Catalog().Namespaces[Oplog].Documents.List)
txn.Clean(2, 0, 0, time.Hour)
assert.Equal(t, bsonkit.List{update, delete}, txn.Catalog().Namespaces[Oplog].Documents.List)
txn.Clean(0, 1, 0, time.Hour)
assert.Equal(t, bsonkit.List{delete}, txn.Catalog().Namespaces[Oplog].Documents.List)
txn.Clean(0, 0, 0, time.Hour)
assert.Empty(t, txn.Catalog().Namespaces[Oplog].Documents.List)
}
func TestTransactionOplogCleaningByTime(t *testing.T) {
txn := NewTransaction(NewCatalog())
/* prepare */
id1 := primitive.NewObjectID()
_, err := txn.Insert(Handle{"foo", "bar"}, bsonkit.List{
bsonkit.MustConvert(bson.M{
"_id": id1,
"foo": "bar",
}),
}, true)
assert.NoError(t, err)
time.Sleep(time.Second)
_, err = txn.Update(Handle{"foo", "bar"}, bsonkit.MustConvert(bson.M{
"_id": id1,
}), nil, bsonkit.MustConvert(bson.M{
"$set": bson.M{
"foo": "baz",
},
}), 0, 0, false, nil)
assert.NoError(t, err)
time.Sleep(time.Second)
_, err = txn.Delete(Handle{"foo", "bar"}, bsonkit.MustConvert(bson.M{
"_id": id1,
}), nil, 0, 0)
assert.NoError(t, err)
time.Sleep(time.Second)
assert.Len(t, txn.Catalog().Namespaces[Oplog].Documents.List, 3)
insert := txn.Catalog().Namespaces[Oplog].Documents.List[0]
update := txn.Catalog().Namespaces[Oplog].Documents.List[1]
delete := txn.Catalog().Namespaces[Oplog].Documents.List[2]
/* clean */
txn.Clean(0, 100, 3*time.Second, 0)
assert.Equal(t, bsonkit.List{insert, update, delete}, txn.Catalog().Namespaces[Oplog].Documents.List)
txn.Clean(0, 100, 2*time.Second, 0)
assert.Equal(t, bsonkit.List{update, delete}, txn.Catalog().Namespaces[Oplog].Documents.List)
txn.Clean(0, 100, 0, 2*time.Second)
assert.Equal(t, bsonkit.List{delete}, txn.Catalog().Namespaces[Oplog].Documents.List)
txn.Clean(0, 100, 0, 0)
assert.Empty(t, txn.Catalog().Namespaces[Oplog].Documents.List)
}