Skip to content

Commit dbcfd74

Browse files
committed
refactor: reader casting
1 parent 94b1722 commit dbcfd74

File tree

3 files changed

+17
-17
lines changed

3 files changed

+17
-17
lines changed

core/eth/reader.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -738,8 +738,8 @@ func (t *Reader) GetVersionedBlobParams(ctx context.Context, blobVersion uint16)
738738
}
739739
return &core.BlobVersionParameters{
740740
CodingRate: uint32(params.CodingRate),
741-
NumChunks: uint32(params.NumChunks),
742-
MaxNumOperators: uint32(params.MaxNumOperators),
741+
NumChunks: params.NumChunks,
742+
MaxNumOperators: params.MaxNumOperators,
743743
}, nil
744744
}
745745

core/meterer/meterer_test.go

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -177,10 +177,10 @@ func teardown() {
177177

178178
func TestMetererReservations(t *testing.T) {
179179
ctx := context.Background()
180-
paymentChainState.On("GetReservationWindow", testifymock.Anything).Return(uint32(5), nil)
180+
paymentChainState.On("GetReservationWindow", testifymock.Anything).Return(uint64(5), nil)
181181
paymentChainState.On("GetGlobalSymbolsPerSecond", testifymock.Anything).Return(uint64(1009), nil)
182-
paymentChainState.On("GetGlobalRatePeriodInterval", testifymock.Anything).Return(uint32(1), nil)
183-
paymentChainState.On("GetMinNumSymbols", testifymock.Anything).Return(uint32(3), nil)
182+
paymentChainState.On("GetGlobalRatePeriodInterval", testifymock.Anything).Return(uint64(1), nil)
183+
paymentChainState.On("GetMinNumSymbols", testifymock.Anything).Return(uint64(3), nil)
184184

185185
now := time.Now().UnixNano()
186186
reservationPeriod := meterer.GetReservationPeriodByNanosecond(now, mt.ChainPaymentState.GetReservationWindow())
@@ -281,8 +281,8 @@ func TestMetererReservations(t *testing.T) {
281281
func TestMetererOnDemand(t *testing.T) {
282282
ctx := context.Background()
283283
quorumNumbers := []uint8{0, 1}
284-
paymentChainState.On("GetPricePerSymbol", testifymock.Anything).Return(uint32(2), nil)
285-
paymentChainState.On("GetMinNumSymbols", testifymock.Anything).Return(uint32(3), nil)
284+
paymentChainState.On("GetPricePerSymbol", testifymock.Anything).Return(uint64(2), nil)
285+
paymentChainState.On("GetMinNumSymbols", testifymock.Anything).Return(uint64(3), nil)
286286
now := time.Now().Unix()
287287

288288
paymentChainState.On("GetOnDemandPaymentByAccount", testifymock.Anything, testifymock.MatchedBy(func(account gethcommon.Address) bool {
@@ -328,7 +328,7 @@ func TestMetererOnDemand(t *testing.T) {
328328
header = createPaymentHeader(now, priceCharged, accountID2)
329329
symbolsCharged, err := mt.MeterRequest(ctx, *header, symbolLength, quorumNumbers)
330330
assert.NoError(t, err)
331-
assert.Equal(t, uint32(102), symbolsCharged)
331+
assert.Equal(t, uint64(102), symbolsCharged)
332332
header = createPaymentHeader(now, priceCharged, accountID2)
333333
_, err = mt.MeterRequest(ctx, *header, symbolLength, quorumNumbers)
334334
assert.ErrorContains(t, err, "exact payment already exists")
@@ -339,7 +339,7 @@ func TestMetererOnDemand(t *testing.T) {
339339
header = createPaymentHeader(now, new(big.Int).Mul(priceCharged, big.NewInt(int64(i+1))), accountID2)
340340
symbolsCharged, err = mt.MeterRequest(ctx, *header, symbolLength, quorumNumbers)
341341
assert.NoError(t, err)
342-
assert.Equal(t, uint32(102), symbolsCharged)
342+
assert.Equal(t, uint64(102), symbolsCharged)
343343
}
344344

345345
// test cumulative payment on-chain constraint
@@ -384,8 +384,8 @@ func TestMeterer_paymentCharged(t *testing.T) {
384384
tests := []struct {
385385
name string
386386
symbolLength uint64
387-
pricePerSymbol uint32
388-
minNumSymbols uint32
387+
pricePerSymbol uint64
388+
minNumSymbols uint64
389389
expected *big.Int
390390
}{
391391
{
@@ -427,8 +427,8 @@ func TestMeterer_paymentCharged(t *testing.T) {
427427

428428
paymentChainState := &mock.MockOnchainPaymentState{}
429429
for _, tt := range tests {
430-
paymentChainState.On("GetPricePerSymbol", testifymock.Anything).Return(uint32(tt.pricePerSymbol), nil)
431-
paymentChainState.On("GetMinNumSymbols", testifymock.Anything).Return(uint32(tt.minNumSymbols), nil)
430+
paymentChainState.On("GetPricePerSymbol", testifymock.Anything).Return(uint64(tt.pricePerSymbol), nil)
431+
paymentChainState.On("GetMinNumSymbols", testifymock.Anything).Return(uint64(tt.minNumSymbols), nil)
432432
t.Run(tt.name, func(t *testing.T) {
433433
m := &meterer.Meterer{
434434
ChainPaymentState: paymentChainState,

disperser/apiserver/server_v2_test.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -455,11 +455,11 @@ func newTestServerV2(t *testing.T) *testComponents {
455455
// append test name to each table name for an unique store
456456
mockState := &mock.MockOnchainPaymentState{}
457457
mockState.On("RefreshOnchainPaymentState", tmock.Anything).Return(nil).Maybe()
458-
mockState.On("GetReservationWindow", tmock.Anything).Return(uint32(1), nil)
459-
mockState.On("GetPricePerSymbol", tmock.Anything).Return(uint32(2), nil)
458+
mockState.On("GetReservationWindow", tmock.Anything).Return(uint64(1), nil)
459+
mockState.On("GetPricePerSymbol", tmock.Anything).Return(uint64(2), nil)
460460
mockState.On("GetGlobalSymbolsPerSecond", tmock.Anything).Return(uint64(1009), nil)
461-
mockState.On("GetGlobalRatePeriodInterval", tmock.Anything).Return(uint32(1), nil)
462-
mockState.On("GetMinNumSymbols", tmock.Anything).Return(uint32(3), nil)
461+
mockState.On("GetGlobalRatePeriodInterval", tmock.Anything).Return(uint64(1), nil)
462+
mockState.On("GetMinNumSymbols", tmock.Anything).Return(uint64(3), nil)
463463

464464
now := uint64(time.Now().Unix())
465465
mockState.On("GetReservedPaymentByAccount", tmock.Anything, tmock.Anything).Return(&core.ReservedPayment{SymbolsPerSecond: 100, StartTimestamp: now + 1200, EndTimestamp: now + 1800, QuorumSplits: []byte{50, 50}, QuorumNumbers: []uint8{0, 1}}, nil)

0 commit comments

Comments
 (0)