github.com/mysteriumnetwork/node@v0.0.0-20240516044423-365054f76801/core/quality/morqa_transport_test.go

/*
 * Copyright (C) 2019 The "MysteriumNetwork/node" Authors.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package quality

import (
	"fmt"
	"io"
	"net/http"
	"net/http/httptest"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"google.golang.org/protobuf/proto"

	"github.com/mysteriumnetwork/metrics"
	"github.com/mysteriumnetwork/node/core/location/locationstate"
	"github.com/mysteriumnetwork/node/identity"
)

var (
	eventStartup = Event{
		EventName:   unlockEventName,
		Application: appInfo{Version: "test version"},
		Context:     "0x1234567890abcdef",
	}

	signerFactory = func(id identity.Identity) identity.Signer {
		return &identity.SignerFake{}
	}
)

func TestMORQATransport_SendEvent_HandlesSuccess(t *testing.T) {
	var events metrics.SignedBatch

	server := httptest.NewServer(http.HandlerFunc(func(response http.ResponseWriter, request *http.Request) {
		body, _ := io.ReadAll(request.Body)
		_ = proto.Unmarshal(body, &events)
		response.WriteHeader(http.StatusAccepted)
	}))

	morqa := NewMorqaClient(httpClient, server.URL, signerFactory)

	go morqa.Start()
	defer morqa.Stop()

	transport := &morqaTransport{morqaClient: morqa, lp: &mockLocationResolver{}}

	err := transport.SendEvent(eventStartup)
	assert.NoError(t, err)

	assert.Eventually(t, func() bool {
		morqa.sendAll()

		return len(events.Batch.Events) > 0
	}, 2*time.Second, 10*time.Millisecond)

	assert.Exactly(
		t,
		"c2lnbmVkChAiDgoMdGVzdCB2ZXJzaW9u",
		events.Signature,
	)

	assert.Exactly(
		t,
		&metrics.Event{
			IsProvider: false,
			TargetId:   "",
			Version: &metrics.VersionPayload{
				Version: "test version",
			},
		},
		events.Batch.Events[0],
	)
}

func TestMORQAT_sendMetrics_HandlesErrorsWithMessages(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusBadRequest)
		_, _ = w.Write([]byte(`{
			"message": "invalid payload given"
		}`))
	}))

	morqa := NewMorqaClient(httpClient, server.URL, signerFactory)
	morqa.addMetric(metric{
		event: &metrics.Event{},
	})
	err := morqa.sendMetrics("")

	assert.EqualError(t, err, fmt.Sprintf(
		"server response invalid: 400 Bad Request (%s/batch). Possible error: invalid payload given",
		server.URL,
	))
}

func TestMORQATransport_SendEvent_HandlesValidationErrors(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusUnprocessableEntity)
		_, _ = w.Write([]byte(`{
			"message": "validation problems",
			"errors": {
				"field": [ {"code": "required", "message": "Field is required"} ]
			}
		}`))
	}))

	morqa := NewMorqaClient(httpClient, server.URL, signerFactory)
	morqa.addMetric(metric{
		event: &metrics.Event{},
	})
	err := morqa.sendMetrics("")

	assert.EqualError(t, err, fmt.Sprintf(
		"server response invalid: 422 Unprocessable Entity (%s/batch). Possible error: validation problems",
		server.URL,
	))
}

func TestMORQA_ProposalQuality(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		_, _ = w.Write([]byte(`[{
			"proposalId": { "providerId": "0x61400b27616f3ce15a86e4cd12c27c7a4d1c545c", "serviceType": "openvpn" },
			"quality": 2
		}, {
			"proposalId": { "providerId": "0xb724ba4f646babdebaaad1d1aea6b26df568e8f6", "serviceType": "openvpn" },
			"quality": 1
		}, {
			"proposalId": { "providerId": "0x093285d0a05ad5d9a05e0dae1eb69e7437fa02c6", "serviceType": "openvpn" },
			"quality": 0
		}]`))
	}))

	morqa := NewMorqaClient(httpClient, server.URL, signerFactory)
	proposalMetrics := morqa.ProposalsQuality()

	assert.Equal(t,
		[]ProposalQuality{
			{
				ProposalID: ProposalID{ProviderID: "0x61400b27616f3ce15a86e4cd12c27c7a4d1c545c", ServiceType: "openvpn"},
				Quality:    2,
			},
			{
				ProposalID: ProposalID{ProviderID: "0xb724ba4f646babdebaaad1d1aea6b26df568e8f6", ServiceType: "openvpn"},
				Quality:    1,
			},
			{
				ProposalID: ProposalID{ProviderID: "0x093285d0a05ad5d9a05e0dae1eb69e7437fa02c6", ServiceType: "openvpn"},
				Quality:    0,
			},
		},
		proposalMetrics,
	)
}

type mockLocationResolver struct{}

func (mlr *mockLocationResolver) GetOrigin() locationstate.Location {
	return locationstate.Location{}
}