@@ -88,6 +88,9 @@ func TestSequenceIterator_Run(t *testing.T) {
88
88
ReadChunkMetasFn : func (f TSSPFile , idx int ) ([]byte , []uint32 , error ) { return make ([]byte , 10 ), make ([]uint32 , 10 ), nil },
89
89
}
90
90
91
+ errSearchSeriesKey := errno .NewError (errno .ErrSearchSeriesKey )
92
+ mockErr := errors .New ("mock error" )
93
+
91
94
for _ , testcase := range []struct {
92
95
Name string
93
96
Handler SequenceIteratorHandler
@@ -115,14 +118,38 @@ func TestSequenceIterator_Run(t *testing.T) {
115
118
},
116
119
Err : io .EOF ,
117
120
},
121
+ {
122
+ Name : "ErrSearchSeriesKey" ,
123
+ Handler : & MockSequenceIteratorHandler {
124
+ BeginFn : func () {},
125
+ FinishFn : func () {},
126
+ NextFileFn : func (file TSSPFile ) {},
127
+ NextChunkMetaFn : func (bytes []byte ) error { return errSearchSeriesKey },
128
+ LimitedFn : func () bool { return false },
129
+ },
130
+ Err : nil ,
131
+ },
132
+ {
133
+ Name : "Unknown Error" ,
134
+ Handler : & MockSequenceIteratorHandler {
135
+ BeginFn : func () {},
136
+ FinishFn : func () {},
137
+ NextFileFn : func (file TSSPFile ) {},
138
+ NextChunkMetaFn : func (bytes []byte ) error { return mockErr },
139
+ LimitedFn : func () bool { return false },
140
+ },
141
+ Err : mockErr ,
142
+ },
118
143
} {
119
- iterator := NewSequenceIterator (testcase .Handler , logger .NewLogger (errno .ModuleUnknown ).SetZapLogger (zap .NewNop ()))
120
- iterator .SetChunkMetasReader (metaReader )
121
- iterator .AddFiles (tsspFiles )
122
- err := iterator .Run ()
123
- if ! errors .Is (err , testcase .Err ) {
124
- t .Fatalf ("SequenceIterator.Run failed, actual %v, expected: %v" , err , testcase .Err )
125
- }
144
+ t .Run (testcase .Name , func (t * testing.T ) {
145
+ iterator := NewSequenceIterator (testcase .Handler , logger .NewLogger (errno .ModuleUnknown ).SetZapLogger (zap .NewNop ()))
146
+ iterator .SetChunkMetasReader (metaReader )
147
+ iterator .AddFiles (tsspFiles )
148
+ err := iterator .Run ()
149
+ if ! errors .Is (err , testcase .Err ) {
150
+ t .Fatalf ("SequenceIterator.Run failed, actual %v, expected: %v" , err , testcase .Err )
151
+ }
152
+ })
126
153
}
127
154
128
155
}
0 commit comments