| 186 | } |
| 187 | |
| 188 | func TestDB_Iterator(t *testing.T) { |
| 189 | tempDir := t.TempDir() |
| 190 | cfg := &config.Config{ |
| 191 | LevelDB: config.LevelDBConfig{ |
| 192 | CacheSize: 64 * 1024 * 1024, |
| 193 | BlockSize: 4 * 1024, |
| 194 | WriteBufferSize: 4 * 1024 * 1024, |
| 195 | MaxOpenFiles: 1000, |
| 196 | Compression: true, |
| 197 | }, |
| 198 | } |
| 199 | |
| 200 | db, err := Store{}.Open(tempDir, cfg) |
| 201 | require.NoError(t, err) |
| 202 | defer db.Close() |
| 203 | |
| 204 | // Insert test data |
| 205 | testData := map[string]string{ |
| 206 | "a": "1", |
| 207 | "b": "2", |
| 208 | "c": "3", |
| 209 | } |
| 210 | |
| 211 | for k, v := range testData { |
| 212 | err = db.Put([]byte(k), []byte(v)) |
| 213 | require.NoError(t, err) |
| 214 | } |
| 215 | |
| 216 | // Test iterator |
| 217 | iter := db.NewIterator() |
| 218 | defer iter.Close() |
| 219 | |
| 220 | count := 0 |
| 221 | for iter.First(); iter.Valid(); iter.Next() { |
| 222 | key := string(iter.Key()) |
| 223 | value := string(iter.Value()) |
| 224 | |
| 225 | expectedValue, exists := testData[key] |
| 226 | assert.True(t, exists) |
| 227 | assert.Equal(t, expectedValue, value) |
| 228 | count++ |
| 229 | } |
| 230 | |
| 231 | assert.Equal(t, len(testData), count) |
| 232 | } |
| 233 | |
| 234 | func TestDB_Snapshot(t *testing.T) { |
| 235 | tempDir := t.TempDir() |