File tree Expand file tree Collapse file tree 1 file changed +6
-11
lines changed Expand file tree Collapse file tree 1 file changed +6
-11
lines changed Original file line number Diff line number Diff line change @@ -27,19 +27,14 @@ func NewDummyStore(t *testing.T) *Store[*DummyHeader] {
27
27
// NewStore creates a generic mock store supporting different type of Headers based on Generator.
28
28
func NewStore [H header.Header [H ]](_ * testing.T , gen Generator [H ], numHeaders int ) * Store [H ] {
29
29
store := & Store [H ]{
30
- Headers : make (map [uint64 ]H ),
31
- HeadHeight : 1 ,
32
- TailHeight : 1 ,
30
+ Headers : make (map [uint64 ]H ),
33
31
}
34
32
35
- for i := 0 ; i < numHeaders ; i ++ {
36
- header := gen .NextHeader ()
37
- store .Headers [header .Height ()] = header
38
-
39
- if header .Height () > store .HeadHeight {
40
- store .HeadHeight = header .Height ()
41
- }
33
+ for range numHeaders {
34
+ h := gen .NextHeader ()
35
+ _ = store .Append (context .Background (), h )
42
36
}
37
+
43
38
return store
44
39
}
45
40
@@ -134,7 +129,7 @@ func (m *Store[H]) Append(_ context.Context, headers ...H) error {
134
129
if height > m .HeadHeight {
135
130
m .HeadHeight = height
136
131
}
137
- if height < m .TailHeight {
132
+ if height < m .TailHeight || m . TailHeight == 0 {
138
133
m .TailHeight = height
139
134
}
140
135
}
You can’t perform that action at this time.
0 commit comments