Skip to content

Commit 8955b58

Browse files
committed
expose merkledb config defaults
Signed-off-by: Joshua Kim <[email protected]>
1 parent 03726dc commit 8955b58

File tree

2 files changed

+11
-23
lines changed

2 files changed

+11
-23
lines changed

x/merkledb/db_test.go

Lines changed: 7 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@ import (
1414
"testing"
1515
"time"
1616

17-
"github.com/prometheus/client_golang/prometheus"
1817
"github.com/stretchr/testify/require"
1918

2019
"github.com/ava-labs/avalanchego/database"
@@ -27,8 +26,6 @@ import (
2726
"github.com/ava-labs/avalanchego/utils/units"
2827
)
2928

30-
const defaultHistoryLength = 300
31-
3229
// newDB returns a new merkle database with the underlying type so that tests can access unexported fields
3330
func newDB(ctx context.Context, db database.Database, config Config) (*merkleDB, error) {
3431
db, err := New(ctx, db, config)
@@ -38,14 +35,6 @@ func newDB(ctx context.Context, db database.Database, config Config) (*merkleDB,
3835
return db.(*merkleDB), nil
3936
}
4037

41-
func newDefaultConfig() Config {
42-
config := NewConfig()
43-
config.HistoryLength = defaultHistoryLength
44-
config.Reg = prometheus.NewRegistry()
45-
46-
return config
47-
}
48-
4938
func Test_MerkleDB_Get_Safety(t *testing.T) {
5039
require := require.New(t)
5140

@@ -125,7 +114,7 @@ func Test_MerkleDB_DB_Load_Root_From_DB(t *testing.T) {
125114
db, err := New(
126115
context.Background(),
127116
baseDB,
128-
newDefaultConfig(),
117+
NewConfig(),
129118
)
130119
require.NoError(err)
131120

@@ -153,7 +142,7 @@ func Test_MerkleDB_DB_Load_Root_From_DB(t *testing.T) {
153142
db, err = New(
154143
context.Background(),
155144
baseDB,
156-
newDefaultConfig(),
145+
NewConfig(),
157146
)
158147
require.NoError(err)
159148

@@ -167,7 +156,7 @@ func Test_MerkleDB_DB_Rebuild(t *testing.T) {
167156

168157
initialSize := 5_000
169158

170-
config := newDefaultConfig()
159+
config := NewConfig()
171160
config.ValueNodeCacheSize = uint(initialSize)
172161
config.IntermediateNodeCacheSize = uint(initialSize)
173162

@@ -224,7 +213,7 @@ func Test_MerkleDB_Failed_Batch_Commit(t *testing.T) {
224213
db, err := New(
225214
context.Background(),
226215
memDB,
227-
newDefaultConfig(),
216+
NewConfig(),
228217
)
229218
require.NoError(err)
230219

@@ -245,7 +234,7 @@ func Test_MerkleDB_Value_Cache(t *testing.T) {
245234
db, err := New(
246235
context.Background(),
247236
memDB,
248-
newDefaultConfig(),
237+
NewConfig(),
249238
)
250239
require.NoError(err)
251240

@@ -1292,7 +1281,7 @@ func TestCrashRecovery(t *testing.T) {
12921281
merkleDB, err := newDatabase(
12931282
context.Background(),
12941283
baseDB,
1295-
newDefaultConfig(),
1284+
NewConfig(),
12961285
&mockMetrics{},
12971286
)
12981287
require.NoError(err)
@@ -1310,7 +1299,7 @@ func TestCrashRecovery(t *testing.T) {
13101299
newMerkleDB, err := newDatabase(
13111300
context.Background(),
13121301
baseDB,
1313-
newDefaultConfig(),
1302+
NewConfig(),
13141303
&mockMetrics{},
13151304
)
13161305
require.NoError(err)

x/merkledb/helpers_test.go

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,19 +20,18 @@ func getBasicDB() (*merkleDB, error) {
2020
return newDatabase(
2121
context.Background(),
2222
memdb.New(),
23-
newDefaultConfig(),
23+
NewConfig(),
2424
&mockMetrics{},
2525
)
2626
}
2727

28-
func getBasicDBWithBranchFactor(bf BranchFactor) (*merkleDB, error) {
29-
config := newDefaultConfig()
28+
func getBasicDBWithBranchFactor(bf BranchFactor) (MerkleDB, error) {
29+
config := NewConfig()
3030
config.BranchFactor = bf
31-
return newDatabase(
31+
return New(
3232
context.Background(),
3333
memdb.New(),
3434
config,
35-
&mockMetrics{},
3635
)
3736
}
3837

0 commit comments

Comments
 (0)