WIP: add rewards tests in client

Author: 0oM4R

node-registrar/client/node_test.go

@@ -2,6 +2,7 @@ package client
 
 import (
 	"encoding/base64"
+	"fmt"
 	"net/http"
 	"net/http/httptest"
 	"net/url"
@@ -159,6 +160,7 @@ func TestGetNode(t *testing.T) {
 	t.Run("test get node with twin id", func(t *testing.T) {
 		request = getNodeWithTwinID
 		result, err := c.GetNodeByTwinID(twinID)
+		fmt.Println(result)
 		require.NoError(err)
 		require.Equal(node, result)
 	})
@@ -171,3 +173,46 @@ func TestGetNode(t *testing.T) {
 		require.Equal([]Node{node}, result)
 	})
 }
+
+func TestGetNodeCapacityRewards(t *testing.T) {
+	var request int
+	var count int
+	require := require.New(t)
+
+	testServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		statusCode, body := serverHandler(r, request, count, require)
+		w.WriteHeader(statusCode)
+		_, err := w.Write(body)
+		require.NoError(err)
+		count++
+	}))
+	defer testServer.Close()
+
+	baseURL, err := url.JoinPath(testServer.URL, "v1")
+	require.NoError(err)
+	request = newClientWithNoAccount
+	c, err := NewRegistrarClient(baseURL)
+	require.NoError(err)
+
+	t.Run("test get node capacity rewards, status ok", func(t *testing.T) {
+		request = getNodeCapacityRewardsWithStatusOK
+		resp, err := c.GetNodeCapacityRewards(nodeID)
+		require.NoError(err)
+		require.Equal(resp, NodeCapacityReward{})
+	})
+
+	t.Run("get node rewards for non-existing node", func(t *testing.T) {
+		request = getNodeCapacityRewardsWithStatusNotFound
+		_, err := c.GetNodeCapacityRewards(nodeID)
+		require.Error(err)
+	})
+
+	t.Run("No reports available, status UnprocessableEntity", func(t *testing.T) {
+		request = getNodeCapacityRewardsWithStatusUnprocessableEntity
+		res, err := c.GetNodeCapacityRewards(1)
+		fmt.Println(res)
+		fmt.Println(err.Error())
+		require.Error(err)
+		require.Equal(res, NodeCapacityReward{TfReward: 323})
+	})
+}

node-registrar/client/utils_test.go

@@ -11,7 +11,7 @@ import (
 var (
 	account = Account{TwinID: 1, Relays: []string{}, RMBEncKey: ""}
 	farm    = Farm{FarmID: 1, FarmName: "freeFarm", TwinID: 1}
-	node    = Node{NodeID: 1, FarmID: farmID, TwinID: twinID}
+	node    = Node{NodeID: 1, FarmID: farmID, TwinID: twinID, Resources: Resources{CRU: 2342}}
 )
 
 const (
@@ -44,6 +44,10 @@ const (
 	getNodeWithTwinID
 	listNodesInFarm
 
+	getNodeCapacityRewardsWithStatusOK
+	getNodeCapacityRewardsWithStatusNotFound
+	getNodeCapacityRewardsWithStatusUnprocessableEntity
+
 	testMnemonic = "bottom drive obey lake curtain smoke basket hold race lonely fit walk"
 
 	farmID uint64 = 1
@@ -220,6 +224,24 @@ func serverHandler(r *http.Request, request, count int, require *require.Asserti
 		require.NoError(err)
 		return http.StatusOK, resp
 
+	case getNodeCapacityRewardsWithStatusOK:
+		require.Equal("/v1/nodes/1/rewards", r.URL.Path)
+		require.Equal(http.MethodGet, r.Method)
+		resp, err := json.Marshal(NodeCapacityReward{})
+		require.NoError(err)
+		return http.StatusOK, resp
+
+	case getNodeCapacityRewardsWithStatusNotFound:
+		require.Equal("/v1/nodes/1/rewards", r.URL.Path)
+		require.Equal(http.MethodGet, r.Method)
+		return http.StatusNotFound, nil
+
+	case getNodeCapacityRewardsWithStatusUnprocessableEntity:
+		require.Equal("/v1/nodes/90/rewards", r.URL.Path)
+		require.Equal(http.MethodGet, r.Method)
+		resp, err := json.Marshal(NodeCapacityReward{TfReward: 239843})
+		require.NoError(err)
+		return http.StatusUnprocessableEntity, resp
 	// unauthorized requests
 	case newClientWithNoAccount,
 		getAccountWithPKStatusNotFount,
@@ -230,7 +252,6 @@ func serverHandler(r *http.Request, request, count int, require *require.Asserti
 		require.Equal(account.PublicKey, r.URL.Query().Get("public_key"))
 		require.Equal(http.MethodGet, r.Method)
 		return http.StatusNotFound, nil
-
 	}
 
 	return http.StatusNotAcceptable, nil