| 144 | } |
| 145 | |
| 146 | func TestDB_WriteBatch(t *testing.T) { |
| 147 | tempDir := t.TempDir() |
| 148 | cfg := &config.Config{ |
| 149 | LevelDB: config.LevelDBConfig{ |
| 150 | CacheSize: 64 * 1024 * 1024, |
| 151 | BlockSize: 4 * 1024, |
| 152 | WriteBufferSize: 4 * 1024 * 1024, |
| 153 | MaxOpenFiles: 1000, |
| 154 | Compression: true, |
| 155 | }, |
| 156 | } |
| 157 | |
| 158 | db, err := Store{}.Open(tempDir, cfg) |
| 159 | require.NoError(t, err) |
| 160 | defer db.Close() |
| 161 | |
| 162 | wb := db.NewWriteBatch() |
| 163 | defer wb.Close() |
| 164 | |
| 165 | key1 := []byte("batch-key1") |
| 166 | value1 := []byte("batch-value1") |
| 167 | key2 := []byte("batch-key2") |
| 168 | value2 := []byte("batch-value2") |
| 169 | |
| 170 | // Add operations to batch |
| 171 | wb.Put(key1, value1) |
| 172 | wb.Put(key2, value2) |
| 173 | |
| 174 | // Commit batch |
| 175 | err = wb.Commit() |
| 176 | require.NoError(t, err) |
| 177 | |
| 178 | // Verify values |
| 179 | retrieved1, err := db.Get(key1) |
| 180 | require.NoError(t, err) |
| 181 | assert.Equal(t, value1, retrieved1) |
| 182 | |
| 183 | retrieved2, err := db.Get(key2) |
| 184 | require.NoError(t, err) |
| 185 | assert.Equal(t, value2, retrieved2) |
| 186 | } |
| 187 | |
| 188 | func TestDB_Iterator(t *testing.T) { |
| 189 | tempDir := t.TempDir() |