fix: QueryGetSecondarySpStorePriceByTime return data of other SPs in …

…some cases (#263)
bnb-chain · May 25, 2023 · 3cb644a · 3cb644a
1 parent 102ed9a
commit 3cb644a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/e2e/tests/sp_test.go b/e2e/tests/sp_test.go
@@ -252,6 +252,16 @@ func (s *StorageProviderTestSuite) TestSpStoragePrice() {
 	s.Require().Equal(newReadPrice, spStoragePrice2.SpStoragePrice.ReadPrice)
 	s.Require().Equal(newStorePrice, spStoragePrice2.SpStoragePrice.StorePrice)
 	s.CheckSecondarySpPrice()
+	// query sp storage price by time before it exists, expect error
+	_, err = s.Client.QueryGetSecondarySpStorePriceByTime(ctx, &sptypes.QueryGetSecondarySpStorePriceByTimeRequest{
+		Timestamp: 1,
+	})
+	s.Require().Error(err)
+	_, err = s.Client.QueryGetSpStoragePriceByTime(ctx, &sptypes.QueryGetSpStoragePriceByTimeRequest{
+		SpAddr:    spAddr,
+		Timestamp: 1,
+	})
+	s.Require().Error(err)
 }
 
 func (s *StorageProviderTestSuite) CheckSecondarySpPrice() {

diff --git a/x/sp/keeper/sp_storage_price.go b/x/sp/keeper/sp_storage_price.go
@@ -90,8 +90,11 @@ func (k Keeper) GetSpStoragePriceByTime(
 		return val, fmt.Errorf("no price found")
 	}
 
+	spAddrRes, UpdateTimeSec := types.ParseSpStoragePriceKey(iterator.Key())
+	if !spAddrRes.Equals(spAddr) {
+		return val, fmt.Errorf("no price found")
+	}
 	k.cdc.MustUnmarshal(iterator.Value(), &val)
-	_, UpdateTimeSec := types.ParseSpStoragePriceKey(iterator.Key())
 	val.SpAddress = spAddr.String()
 	val.UpdateTimeSec = UpdateTimeSec