thiagodeev/rpcv08-websocket

.github/workflows/main_ci_check.yml

@@ -71,3 +71,7 @@ jobs:
           cd ../simpleCall && go build
           cd ../simpleInvoke && go build
           cd ../deployContractUDC && go build
+
+      # Test client on mock
+      - name: Test client with mocks
+        run: cd client && go test -v

account/account_test.go

@@ -4,7 +4,6 @@ import (
 	"context"
 	"encoding/json"
 	"flag"
-	"fmt"
 	"math/big"
 	"os"
 	"testing"
@@ -18,7 +17,6 @@ import (
 	"github.com/NethermindEth/starknet.go/mocks"
 	"github.com/NethermindEth/starknet.go/rpc"
 	"github.com/NethermindEth/starknet.go/utils"
-	"github.com/joho/godotenv"
 	"github.com/stretchr/testify/require"
 	"go.uber.org/mock/gomock"
 )
@@ -43,18 +41,12 @@ var (
 //
 //	none
 func TestMain(m *testing.M) {
-	flag.StringVar(&testEnv, "env", "mock", "set the test environment")
+	flag.StringVar(&testEnv, "env", "devnet", "set the test environment")
 	flag.Parse()
 	if testEnv == "mock" {
 		return
 	}
-	base = os.Getenv("INTEGRATION_BASE")
-	if base == "" {
-		if err := godotenv.Load(fmt.Sprintf(".env.%s", testEnv)); err != nil {
-			panic(fmt.Sprintf("Failed to load .env.%s, err: %s", testEnv, err))
-		}
-		base = os.Getenv("INTEGRATION_BASE")
-	}
+	base = "http://localhost:5050"
 	os.Exit(m.Run())
 }
 
@@ -1076,7 +1068,7 @@ func TestWaitForTransactionReceipt(t *testing.T) {
 				Timeout:         3, // Should poll 3 times
 				Hash:            new(felt.Felt).SetUint64(100),
 				ExpectedReceipt: rpc.TransactionReceipt{},
-				ExpectedErr:     rpc.Err(rpc.InternalError, &rpc.RPCData{Message: "Post \"http://localhost:5050\": context deadline exceeded"}),
+				ExpectedErr:     rpc.Err(rpc.InternalError, &rpc.RPCData{Message: "context deadline exceeded"}),
 			},
 		},
 	}[testEnv]
@@ -1090,7 +1082,7 @@ func TestWaitForTransactionReceipt(t *testing.T) {
 			rpcErr, ok := err.(*rpc.RPCError)
 			require.True(t, ok)
 			require.Equal(t, test.ExpectedErr.Code, rpcErr.Code)
-			require.Equal(t, test.ExpectedErr.Data.Message, rpcErr.Data.Message)
+			require.Contains(t, rpcErr.Data.Message, test.ExpectedErr.Data.Message) // sometimes the error message starts with "Post \"http://localhost:5050\":..."
 		} else {
 			require.Equal(t, test.ExpectedReceipt.ExecutionStatus, (*resp).ExecutionStatus)
 		}