This repository has been archived by the owner on May 3, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 3
/
iterators_test.go
109 lines (95 loc) · 2.28 KB
/
iterators_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
package gotinydb
import (
"bytes"
"crypto/rand"
"fmt"
"reflect"
"testing"
)
func TestCollectionIterator(t *testing.T) {
defer clean()
err := openT(t)
if err != nil {
return
}
iter := testCol.GetIterator()
n := 0
for ; iter.Valid(); iter.Next() {
n++
}
iter.Close()
if n != 2 {
t.Errorf("regular iterator returned %d document but expect %d", n, 2)
}
iter = testCol.GetRevertedIterator()
n = 0
for ; iter.Valid(); iter.Next() {
n++
}
iter.Close()
if n != 2 {
t.Errorf("reverted iterator returned %d document but expect %d", n, 2)
}
iter = testCol.GetIterator()
n = 0
last := new(testUserStruct)
for iter.Seek(cloneTestUserID); iter.Valid(); iter.Next() {
iter.GetValue(last)
n++
}
if n != 1 {
t.Errorf("regular iterator after seek returned %d document but expect %d", n, 1)
}
if !reflect.DeepEqual(last, cloneTestUser) {
t.Errorf("regular iterator after seek returned %v document but expect %v", last, cloneTestUser)
}
iter.Close()
}
func TestFileIterator(t *testing.T) {
defer clean()
err := openT(t)
if err != nil {
return
}
for i := 0; i < 5; i++ {
randBuff := make([]byte, 20*1000*1000)
rand.Read(randBuff)
buff := bytes.NewBuffer(randBuff)
_, err := testDB.GetFileStore().PutFile(fmt.Sprint(i), fmt.Sprint(i), buff)
if err != nil {
t.Fatalf("can't write the file %d: %s", i, err.Error())
}
}
iter := testDB.GetFileStore().GetFileIterator()
defer iter.Close()
expectedOrder := []string{"0", "2", "3", "1", "4"}
n := 0
for ; iter.Valid(); iter.Next() {
meta := iter.GetMeta()
if expectedOrder[n] != meta.ID {
t.Errorf("the expected ID is %q but got %q", expectedOrder[n], meta.ID)
}
if expectedOrder[n] != meta.Name {
t.Errorf("the expected name is %q but got %q", expectedOrder[n], meta.Name)
}
n++
}
if n != 5 {
t.Errorf("this test must loop %d times but it looks like it did only %d", 5, n)
}
iter.Seek("3")
n = 2
for ; iter.Valid(); iter.Next() {
meta := iter.GetMeta()
if expectedOrder[n] != meta.ID {
t.Errorf("the expected ID is %q but got %q", expectedOrder[n], meta.ID)
}
if expectedOrder[n] != meta.Name {
t.Errorf("the expected name is %q but got %q", expectedOrder[n], meta.Name)
}
n++
}
if n != 5 {
t.Errorf("this test must loop %d times but it looks like it did only %d", 5, n)
}
}