@@ -14,20 +14,32 @@ See the License for the specific language governing permissions and
1414limitations under the License.
1515*/
1616
17- package solo
17+ package deliver
1818
1919import (
2020 "fmt"
2121 "testing"
2222 "time"
2323
24- "google.golang.org/grpc"
25-
24+ "github.com/hyperledger/fabric/orderer/common/bootstrap/static"
2625 "github.com/hyperledger/fabric/orderer/rawledger/ramledger"
2726 cb "github.com/hyperledger/fabric/protos/common"
2827 ab "github.com/hyperledger/fabric/protos/orderer"
28+
29+ "google.golang.org/grpc"
2930)
3031
32+ var genesisBlock * cb.Block
33+
34+ func init () {
35+ bootstrapper := static .New ()
36+ var err error
37+ genesisBlock , err = bootstrapper .GenesisBlock ()
38+ if err != nil {
39+ panic ("Error intializing static bootstrap genesis block" )
40+ }
41+ }
42+
3143// MagicLargestWindow is used as the default max window size for initializing the deliver service
3244const MagicLargestWindow int = 1000
3345
@@ -66,9 +78,9 @@ func TestOldestSeek(t *testing.T) {
6678
6779 m := newMockD ()
6880 defer close (m .recvChan )
69- ds := NewDeliverServer (rl , MagicLargestWindow )
81+ ds := NewHandlerImpl (rl , MagicLargestWindow )
7082
71- go ds .HandleDeliver (m )
83+ go ds .Handle (m )
7284
7385 m .recvChan <- & ab.DeliverUpdate {Type : & ab.DeliverUpdate_Seek {Seek : & ab.SeekInfo {WindowSize : uint64 (MagicLargestWindow ), Start : ab .SeekInfo_OLDEST }}}
7486
@@ -98,9 +110,9 @@ func TestNewestSeek(t *testing.T) {
98110
99111 m := newMockD ()
100112 defer close (m .recvChan )
101- ds := NewDeliverServer (rl , MagicLargestWindow )
113+ ds := NewHandlerImpl (rl , MagicLargestWindow )
102114
103- go ds .HandleDeliver (m )
115+ go ds .Handle (m )
104116
105117 m .recvChan <- & ab.DeliverUpdate {Type : & ab.DeliverUpdate_Seek {Seek : & ab.SeekInfo {WindowSize : uint64 (MagicLargestWindow ), Start : ab .SeekInfo_NEWEST }}}
106118
@@ -126,9 +138,9 @@ func TestSpecificSeek(t *testing.T) {
126138 }
127139
128140 m := newMockD ()
129- ds := NewDeliverServer (rl , MagicLargestWindow )
141+ ds := NewHandlerImpl (rl , MagicLargestWindow )
130142
131- go ds .HandleDeliver (m )
143+ go ds .Handle (m )
132144
133145 m .recvChan <- & ab.DeliverUpdate {Type : & ab.DeliverUpdate_Seek {Seek : & ab.SeekInfo {WindowSize : uint64 (MagicLargestWindow ), Start : ab .SeekInfo_SPECIFIED , SpecifiedNumber : uint64 (ledgerSize - 1 )}}}
134146
@@ -155,9 +167,9 @@ func TestBadSeek(t *testing.T) {
155167
156168 m := newMockD ()
157169 defer close (m .recvChan )
158- ds := NewDeliverServer (rl , MagicLargestWindow )
170+ ds := NewHandlerImpl (rl , MagicLargestWindow )
159171
160- go ds .HandleDeliver (m )
172+ go ds .Handle (m )
161173
162174 m .recvChan <- & ab.DeliverUpdate {Type : & ab.DeliverUpdate_Seek {Seek : & ab.SeekInfo {WindowSize : uint64 (MagicLargestWindow ), Start : ab .SeekInfo_SPECIFIED , SpecifiedNumber : uint64 (ledgerSize - 1 )}}}
163175
@@ -188,9 +200,9 @@ func TestBadWindow(t *testing.T) {
188200
189201 m := newMockD ()
190202 defer close (m .recvChan )
191- ds := NewDeliverServer (rl , MagicLargestWindow )
203+ ds := NewHandlerImpl (rl , MagicLargestWindow )
192204
193- go ds .HandleDeliver (m )
205+ go ds .Handle (m )
194206
195207 m .recvChan <- & ab.DeliverUpdate {Type : & ab.DeliverUpdate_Seek {Seek : & ab.SeekInfo {WindowSize : uint64 (MagicLargestWindow ) * 2 , Start : ab .SeekInfo_OLDEST }}}
196208
@@ -214,9 +226,9 @@ func TestAck(t *testing.T) {
214226
215227 m := newMockD ()
216228 defer close (m .recvChan )
217- ds := NewDeliverServer (rl , MagicLargestWindow )
229+ ds := NewHandlerImpl (rl , MagicLargestWindow )
218230
219- go ds .HandleDeliver (m )
231+ go ds .Handle (m )
220232
221233 m .recvChan <- & ab.DeliverUpdate {Type : & ab.DeliverUpdate_Seek {Seek : & ab.SeekInfo {WindowSize : windowSize , Start : ab .SeekInfo_OLDEST }}}
222234
0 commit comments