@@ -8,22 +8,22 @@ import (
88	"github.com/NethermindEth/juno/core" 
99	"github.com/NethermindEth/juno/core/felt" 
1010	"github.com/NethermindEth/juno/mocks" 
11- 	rpc "github.com/NethermindEth/juno/rpc/v8 " 
11+ 	rpc "github.com/NethermindEth/juno/rpc/v7 " 
1212	adaptfeeder "github.com/NethermindEth/juno/starknetdata/feeder" 
1313	"github.com/NethermindEth/juno/sync" 
1414	"github.com/NethermindEth/juno/utils" 
1515	"github.com/stretchr/testify/require" 
1616	"go.uber.org/mock/gomock" 
1717)
1818
19- func  TestPendingDataWrapper (t  * testing.T ) {
19+ func  TestPendingDataWrapper_PebndingData (t  * testing.T ) {
2020	mockCtrl  :=  gomock .NewController (t )
2121	t .Cleanup (mockCtrl .Finish )
2222	mockSyncReader  :=  mocks .NewMockSyncReader (mockCtrl )
2323	n  :=  utils .HeapPtr (utils .Sepolia )
2424	mockReader  :=  mocks .NewMockReader (mockCtrl )
2525	log  :=  utils .NewNopZapLogger ()
26- 	handler  :=  rpc .New (mockReader , mockSyncReader , nil , "" , log )
26+ 	handler  :=  rpc .New (mockReader , mockSyncReader , nil , "" , & utils . Sepolia ,  log )
2727
2828	client  :=  feeder .NewTestClient (t , n )
2929	gw  :=  adaptfeeder .New (client )
@@ -102,3 +102,55 @@ func TestPendingDataWrapper(t *testing.T) {
102102		))
103103	})
104104}
105+ 
106+ func  TestPendingDataWrapper_PendingState (t  * testing.T ) {
107+ 	mockCtrl  :=  gomock .NewController (t )
108+ 	t .Cleanup (mockCtrl .Finish )
109+ 	mockSyncReader  :=  mocks .NewMockSyncReader (mockCtrl )
110+ 	mockReader  :=  mocks .NewMockReader (mockCtrl )
111+ 	handler  :=  rpc .New (mockReader , mockSyncReader , nil , "" , & utils .Sepolia , nil )
112+ 
113+ 	mockState  :=  mocks .NewMockStateHistoryReader (mockCtrl )
114+ 	t .Run ("Returns pending state when starknet version < 0.14.0" , func (t  * testing.T ) {
115+ 		pending  :=  sync.Pending {}
116+ 		pendingData  :=  pending .AsPendingData ()
117+ 		mockSyncReader .EXPECT ().PendingData ().Return (
118+ 			& pendingData ,
119+ 			nil ,
120+ 		)
121+ 		mockSyncReader .EXPECT ().PendingState ().Return (mockState , nopCloser , nil )
122+ 		pendingState , closer , err  :=  handler .PendingState ()
123+ 
124+ 		require .NoError (t , err )
125+ 		require .NotNil (t , pendingState )
126+ 		require .NotNil (t , closer )
127+ 	})
128+ 
129+ 	t .Run ("Returns latest state starknet version >= 0.14.0" , func (t  * testing.T ) {
130+ 		preConfirmed  :=  core.PreConfirmed {}
131+ 		pendingData  :=  preConfirmed .AsPendingData ()
132+ 		mockSyncReader .EXPECT ().PendingData ().Return (
133+ 			& pendingData ,
134+ 			nil ,
135+ 		)
136+ 		mockReader .EXPECT ().HeadState ().Return (mockState , nopCloser , nil )
137+ 		pending , closer , err  :=  handler .PendingState ()
138+ 
139+ 		require .NoError (t , err )
140+ 		require .NotNil (t , pending )
141+ 		require .NotNil (t , closer )
142+ 	})
143+ 
144+ 	t .Run ("Returns latest state when pending data is nil" , func (t  * testing.T ) {
145+ 		mockSyncReader .EXPECT ().PendingData ().Return (
146+ 			nil ,
147+ 			sync .ErrPendingBlockNotFound ,
148+ 		)
149+ 		mockReader .EXPECT ().HeadState ().Return (mockState , nopCloser , nil )
150+ 		pending , closer , err  :=  handler .PendingState ()
151+ 
152+ 		require .NoError (t , err )
153+ 		require .NotNil (t , pending )
154+ 		require .NotNil (t , closer )
155+ 	})
156+ }
0 commit comments