refactor: code cleanup

Author: alex-sandrk

proxy-router/internal/blockchainapi/compute_session_token_test.go

@@ -0,0 +1,92 @@
+package blockchainapi
+
+import (
+	"math/big"
+	"testing"
+
+	"github.com/MorpheusAIs/Morpheus-Lumerin-Node/proxy-router/internal/blockchainapi/structs"
+	"github.com/MorpheusAIs/Morpheus-Lumerin-Node/proxy-router/internal/lib"
+)
+
+func TestComputeSessionTokenAmount(t *testing.T) {
+	makeBid := func(pricePerSecond int64) *structs.Bid {
+		return &structs.Bid{
+			PricePerSecond: &lib.BigInt{Int: *big.NewInt(pricePerSecond)},
+		}
+	}
+
+	tests := []struct {
+		name          string
+		bid           *structs.Bid
+		duration      *big.Int
+		supply        *big.Int
+		budget        *big.Int
+		directPayment bool
+		want          *big.Int
+		wantErr       bool
+	}{
+		{
+			name:          "direct payment",
+			bid:           makeBid(100),
+			duration:      big.NewInt(3600),
+			supply:        big.NewInt(1_000_000),
+			budget:        big.NewInt(50_000),
+			directPayment: true,
+			want:          big.NewInt(360_000),
+		},
+		{
+			name:          "staked",
+			bid:           makeBid(100),
+			duration:      big.NewInt(3600),
+			supply:        big.NewInt(1_000_000),
+			budget:        big.NewInt(50_000),
+			directPayment: false,
+			want:          big.NewInt(7_200_000),
+		},
+		{
+			name:          "nil bid",
+			bid:           nil,
+			duration:      big.NewInt(3600),
+			supply:        big.NewInt(1_000_000),
+			budget:        big.NewInt(50_000),
+			directPayment: false,
+			wantErr:       true,
+		},
+		{
+			name:          "zero budget",
+			bid:           makeBid(100),
+			duration:      big.NewInt(3600),
+			supply:        big.NewInt(1_000_000),
+			budget:        big.NewInt(0),
+			directPayment: false,
+			wantErr:       true,
+		},
+		{
+			name:          "nil budget",
+			bid:           makeBid(100),
+			duration:      big.NewInt(3600),
+			supply:        big.NewInt(1_000_000),
+			budget:        nil,
+			directPayment: false,
+			wantErr:       true,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := computeSessionTokenAmount(tt.bid, tt.duration, tt.supply, tt.budget, tt.directPayment)
+			if tt.wantErr {
+				if err == nil {
+					t.Fatalf("expected error, got result %s", got)
+				}
+				return
+			}
+			if err != nil {
+				t.Fatalf("unexpected error: %v", err)
+			}
+			if got.Cmp(tt.want) != 0 {
+				t.Errorf("got %s, want %s", got, tt.want)
+			}
+		})
+	}
+}

proxy-router/internal/repositories/ethclient/rpcclientmultiple_test.go

@@ -0,0 +1,117 @@
+package ethclient
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/url"
+	"testing"
+)
+
+func TestShouldRetryRPCError(t *testing.T) {
+	tests := []struct {
+		name     string
+		err      error
+		expected bool
+	}{
+		{
+			name:     "nil error",
+			err:      nil,
+			expected: false,
+		},
+		{
+			name:     "context canceled",
+			err:      context.Canceled,
+			expected: false,
+		},
+		{
+			name:     "context deadline",
+			err:      context.DeadlineExceeded,
+			expected: false,
+		},
+		{
+			name:     "rate limit text",
+			err:      fmt.Errorf("429 too many requests"),
+			expected: true,
+		},
+		{
+			name:     "rate limit code",
+			err:      fmt.Errorf("-32005"),
+			expected: true,
+		},
+		{
+			name:     "execution reverted",
+			err:      fmt.Errorf("execution reverted"),
+			expected: false,
+		},
+		{
+			name:     "revert",
+			err:      fmt.Errorf("revert"),
+			expected: false,
+		},
+		{
+			name:     "timeout",
+			err:      fmt.Errorf("connection timeout"),
+			expected: true,
+		},
+		{
+			name:     "connection refused",
+			err:      fmt.Errorf("connection refused"),
+			expected: true,
+		},
+		{
+			name:     "connection reset",
+			err:      fmt.Errorf("connection reset"),
+			expected: true,
+		},
+		{
+			name:     "eof",
+			err:      fmt.Errorf("eof"),
+			expected: true,
+		},
+		{
+			name:     "cloudflare block",
+			err:      fmt.Errorf("just a moment"),
+			expected: true,
+		},
+		{
+			name:     "quota exceeded",
+			err:      fmt.Errorf("quota exceeded"),
+			expected: true,
+		},
+		{
+			name:     "insufficient funds",
+			err:      fmt.Errorf("insufficient funds for gas"),
+			expected: false,
+		},
+		{
+			name:     "usage limit",
+			err:      fmt.Errorf("usage limit"),
+			expected: true,
+		},
+		{
+			name:     "method not found code",
+			err:      fmt.Errorf("-32601"),
+			expected: true,
+		},
+		{
+			name:     "json syntax error",
+			err:      &json.SyntaxError{},
+			expected: false,
+		},
+		{
+			name:     "url error",
+			err:      &url.Error{Op: "Get", URL: "http://example.com", Err: fmt.Errorf("dial failed")},
+			expected: true,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := shouldRetryRPCError(tt.err)
+			if got != tt.expected {
+				t.Errorf("shouldRetryRPCError(%v) = %v, want %v", tt.err, got, tt.expected)
+			}
+		})
+	}
+}