goleg/goleg_test.go at master · infoforcefeed/goleg · 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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
package goleg

import (
	"bytes"
	"io/ioutil"
	"os"
	"strconv"
	"testing"
)

func openRandomDB(features int) (Database, string, error) {
	name, err := ioutil.TempDir("/tmp", "goleg")
	if err != nil {
		return Database{}, "", err
	}

	//F_APPENDONLY|F_AOL_FFLUSH|F_LZ4|F_SPLAYTREE
	database, err := Open(name, "test", features)
	if err != nil {
		return Database{}, "", err
	}

	return database, name, nil
}

func cleanTemp(dir string) {
	os.RemoveAll(dir)
}

func TestOpen(t *testing.T) {
	if testing.Short() {
		t.Skip("Skipping in short mode")
	}

	database, dir, err := openRandomDB(F_APPENDONLY)
	if err != nil {
		t.Fatalf("Can't open database: %s", err.Error())
	}

	database.Close()
	cleanTemp(dir)
}

const JARN = 10

func TestJar(t *testing.T) {
	if testing.Short() {
		t.Skip("Skipping in short mode")
	}

	database, dir, err := openRandomDB(F_LZ4 | F_SPLAYTREE)
	if err != nil {
		t.Fatalf("Can't open database: %s", err.Error())
	}

	for i := 0; i < JARN; i++ {
		if database.Jar("record"+strconv.Itoa(i), []byte("value"+strconv.Itoa(i))) != 0 {
			t.Fatalf("Can't jar value #%d", i)
		}
	}

	database.Close()
	cleanTemp(dir)
}

func TestUnjar(t *testing.T) {
	database, dir, err := openRandomDB(F_LZ4 | F_SPLAYTREE)
	if err != nil {
		t.Fatalf("Can't open database: %s", err.Error())
	}

	for i := 0; i < JARN; i++ {
		if database.Jar("record"+strconv.Itoa(i), []byte("value"+strconv.Itoa(i))) != 0 {
			t.Fatalf("Can't jar value #%d", i)
		}
	}

	for i := 0; i < JARN; i++ {
		val := database.Unjar("record" + strconv.Itoa(i))
		if !bytes.Equal(val, []byte("value"+strconv.Itoa(i))) {
			t.Errorf("Value #%d doesn't match", i)
		}
	}

	database.Close()
	cleanTemp(dir)
}

func TestFullKeyDump(t *testing.T) {
	database, _, err := openRandomDB(F_LZ4 | F_SPLAYTREE)
	if err != nil {
		t.Fatalf("Can't open database: %s", err.Error())
	}

	for i := 0; i < JARN; i++ {
		if database.Jar("record"+strconv.Itoa(i), []byte("value"+strconv.Itoa(i))) != 0 {
			t.Fatalf("Can't jar value #%d", i)
		}
	}

	gotKeys, keys := database.DumpKeys()

	if !gotKeys {
		t.Fatal("Didn't get keys and should have")
	}

	var j int
	for i := 0; i <= JARN; i++ {
		for _, key := range keys {
			if key == "record"+strconv.Itoa(i) {
				j++
			}
		}
	}
	if j != JARN {
		t.Fatal("One or more keys did not dump")
	}
}

func TestBulkUnjarOnlyReturnsKeysWeGiveIt(t *testing.T) {
	database, _, err := openRandomDB(F_LZ4 | F_SPLAYTREE)
	if err != nil {
		t.Fatalf("Can't open database: %s", err.Error())
	}

	keys := []string{"key0", "key1", "key2", "key3"}

	for i, key := range keys {
		if database.Jar(key, []byte("value"+strconv.Itoa(i))) != 0 {
			t.Fatalf("Can't jar value #%d", i)
		}
	}

	subset := keys[1:] //sans key0

	values := database.BulkUnjar(subset)

	if l := len(values); l != 3 {
		t.Fatalf("Expected a length of 3, got %d", l)
	}

	for i, value := range values {
		if subset[i][3] != string(value)[5] {
			t.Fatalf("Expected %s, got %s", subset[i][3], string(value)[5])
		}
	}
}