tests: add more test cases for the rewards

Author: 0oM4R

node-registrar/client/node_test.go

@@ -201,7 +201,7 @@ func TestGetNodeCapacityRewards(t *testing.T) {
 		request = getNodeCapacityRewardsWithStatusOK
 		resp, err := c.GetNodeCapacityRewards(nodeID)
 		require.NoError(err)
-		require.Equal(resp, NodeCapacityReward{})
+		require.Equal(NodeCapacityReward{}, resp)
 	})
 
 	t.Run("get node rewards for non-existing node", func(t *testing.T) {
@@ -212,8 +212,41 @@ func TestGetNodeCapacityRewards(t *testing.T) {
 
 	t.Run("no reports available, status UnprocessableEntity", func(t *testing.T) {
 		request = getNodeCapacityRewardsWithStatusUnprocessableEntity
-		res, err := c.GetNodeCapacityRewards(1)
+		res, err := c.GetNodeCapacityRewards(nodeID)
 		require.Error(err)
-		require.Equal(res, NodeCapacityReward{})
+		require.Equal(NodeCapacityReward{}, res)
+
+	})
+
+	t.Run("node with partial uptime rewards calculation", func(t *testing.T) {
+		request = getNodeCapacityRewardsWithPartialUptime
+		res, err := c.GetNodeCapacityRewards(nodeID)
+		require.NoError(err)
+		expected := NodeCapacityReward{
+			FarmerReward:     60.0,
+			TfReward:         20.0,
+			FpReward:         20.0,
+			Total:            100.0,
+			UpTimePercentage: 75.0,
+		}
+		require.Equal(expected, res)
+		// Verify reward distribution percentages are correct
+		require.InDelta(0.6, res.FarmerReward/res.Total, 0.001)
+		require.InDelta(0.2, res.TfReward/res.Total, 0.001)
+		require.InDelta(0.2, res.FpReward/res.Total, 0.001)
+	})
+
+	t.Run("bad request due to invalid node ID format", func(t *testing.T) {
+		request = getNodeCapacityRewardsWithBadRequest
+		res, err := c.GetNodeCapacityRewards(nodeID)
+		require.Error(err)
+		require.Equal(NodeCapacityReward{}, res)
+	})
+
+	t.Run("internal server error when calculating rewards", func(t *testing.T) {
+		request = getNodeCapacityRewardsWithServerError
+		res, err := c.GetNodeCapacityRewards(nodeID)
+		require.Error(err)
+		require.Equal(NodeCapacityReward{}, res)
 	})
 }

node-registrar/client/utils_test.go

@@ -47,6 +47,9 @@ const (
 	getNodeCapacityRewardsWithStatusOK
 	getNodeCapacityRewardsWithStatusNotFound
 	getNodeCapacityRewardsWithStatusUnprocessableEntity
+	getNodeCapacityRewardsWithPartialUptime
+	getNodeCapacityRewardsWithBadRequest
+	getNodeCapacityRewardsWithServerError
 
 	testMnemonic = "bottom drive obey lake curtain smoke basket hold race lonely fit walk"
 
@@ -239,9 +242,31 @@ func serverHandler(r *http.Request, request, count int, require *require.Asserti
 	case getNodeCapacityRewardsWithStatusUnprocessableEntity:
 		require.Equal("/v1/nodes/1/rewards", r.URL.Path)
 		require.Equal(http.MethodGet, r.Method)
-		resp, err := json.Marshal(NodeCapacityReward{TfReward: 239843})
+		return http.StatusUnprocessableEntity, nil
+
+	case getNodeCapacityRewardsWithPartialUptime:
+		require.Equal("/v1/nodes/1/rewards", r.URL.Path)
+		require.Equal(http.MethodGet, r.Method)
+		resp, err := json.Marshal(NodeCapacityReward{
+			FarmerReward:     60.0,
+			TfReward:         20.0,
+			FpReward:         20.0,
+			Total:            100.0,
+			UpTimePercentage: 75.0,
+		})
 		require.NoError(err)
-		return http.StatusUnprocessableEntity, resp
+		return http.StatusOK, resp
+
+	case getNodeCapacityRewardsWithBadRequest:
+		require.Equal("/v1/nodes/1/rewards", r.URL.Path)
+		require.Equal(http.MethodGet, r.Method)
+
+		return http.StatusBadRequest, nil
+
+	case getNodeCapacityRewardsWithServerError:
+		require.Equal("/v1/nodes/1/rewards", r.URL.Path)
+		require.Equal(http.MethodGet, r.Method)
+		return http.StatusInternalServerError, nil
 
 	// unauthorized requests
 	case newClientWithNoAccount,

node-registrar/test/main.go

@@ -0,0 +1,31 @@
+package main
+
+import (
+	"fmt"
+	"log"
+
+	"github.com/threefoldtech/tfgrid4-sdk-go/node-registrar/client"
+)
+
+func main() {
+	// Create a new client with the registrar server URL
+	registrarClient, err := client.NewRegistrarClient("http://localhost:8080/api/v1")
+	if err != nil {
+		log.Fatalf("Failed to create client: %v", err)
+	}
+
+	// Get node capacity rewards by node ID
+	nodeID := uint64(34) // Replace with actual node ID
+	rewards, err := registrarClient.GetNodeCapacityRewards(nodeID)
+	if err != nil {
+		log.Fatalf("Failed to get node capacity rewards: %v", err)
+	}
+
+	// Print the rewards information
+	fmt.Printf("Node %d Rewards:\n", nodeID)
+	fmt.Printf("Total: %f\n", rewards.Total)
+	fmt.Printf("Farmer Reward: %f\n", rewards.FarmerReward)
+	fmt.Printf("TF Reward: %f\n", rewards.TfReward)
+	fmt.Printf("FP Reward: %f\n", rewards.FpReward)
+	fmt.Printf("Uptime Percentage: %f%%\n", rewards.UpTimePercentage)
+}