fixed tests

NethermindEth · Jan 9, 2025 · 5badae0 · 5badae0
1 parent dbb495f
commit 5badae0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 9 deletions.
diff --git a/mempool/init_test.go b/mempool/init_test.go
diff --git a/mempool/mempool.go b/mempool/mempool.go
@@ -158,12 +158,10 @@ func (p *Pool) handleTransaction(userTxn *BroadcastedTransaction) error {
 		if tailValue != nil {
 			// Update old tail to point to the new item
 			var oldTailElem storageElem
-			fmt.Println("  - error here")
 			oldTailElem, err := p.dbElem(dbTxn, tailValue)
 			if err != nil {
 				return err
 			}
-			fmt.Println("  - error here solved")
 			oldTailElem.NextHash = userTxn.Transaction.Hash()
 			if err = p.putdbElem(dbTxn, tailValue, &oldTailElem); err != nil {
 				return err

diff --git a/mempool/mempool_test.go b/mempool/mempool_test.go
@@ -9,6 +9,7 @@ import (
 	"github.com/NethermindEth/juno/core/felt"
 	"github.com/NethermindEth/juno/db"
 	"github.com/NethermindEth/juno/db/pebble"
+	_ "github.com/NethermindEth/juno/encoder/registry"
 	"github.com/NethermindEth/juno/mempool"
 	"github.com/NethermindEth/juno/mocks"
 	"github.com/NethermindEth/juno/utils"
@@ -150,7 +151,7 @@ func TestRestoreMempool(t *testing.T) {
 	}
 
 	// check the db has stored the transactions
-	time.Sleep(4000 * time.Millisecond)
+	time.Sleep(100 * time.Millisecond)
 	lenDB, err = pool.LenDB()
 	require.NoError(t, err)
 	require.Equal(t, uint16(3), lenDB)
@@ -181,7 +182,9 @@ func TestRestoreMempool(t *testing.T) {
 
 func TestWait(t *testing.T) {
 	log := utils.NewNopZapLogger()
-	testDB := pebble.NewMemTest(t)
+	testDB, dbCloser, err := setupDatabase("testwait", true)
+	require.NoError(t, err)
+	defer dbCloser()
 	mockCtrl := gomock.NewController(t)
 	t.Cleanup(mockCtrl.Finish)
 	state := mocks.NewMockStateHistoryReader(mockCtrl)