| 119 | } |
| 120 | |
| 121 | func TestIterator_Close(t *testing.T) { |
| 122 | tempDir := t.TempDir() |
| 123 | cfg := &config.Config{ |
| 124 | LevelDB: config.LevelDBConfig{ |
| 125 | CacheSize: 64 * 1024 * 1024, |
| 126 | BlockSize: 4 * 1024, |
| 127 | WriteBufferSize: 4 * 1024 * 1024, |
| 128 | MaxOpenFiles: 1000, |
| 129 | Compression: true, |
| 130 | }, |
| 131 | } |
| 132 | |
| 133 | db, err := Store{}.Open(tempDir, cfg) |
| 134 | require.NoError(t, err) |
| 135 | defer db.Close() |
| 136 | |
| 137 | iter := db.NewIterator() |
| 138 | |
| 139 | // Should be able to use iterator before closing |
| 140 | iter.First() |
| 141 | assert.False(t, iter.Valid()) // Empty DB |
| 142 | |
| 143 | // Close should work without errors |
| 144 | err = iter.Close() |
| 145 | assert.NoError(t, err) |
| 146 | |
| 147 | // Double close should not panic |
| 148 | err = iter.Close() |
| 149 | assert.NoError(t, err) |
| 150 | } |