client_test.go

package ocm_test

import (
	"context"
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/stretchr/testify/assert"

	"code.vereign.com/gaiax/tsa/golib/ocm"
)

func Test_GetLoginProofInvitationSuccess(t *testing.T) {
	expected := &ocm.LoginProofInvitationResponse{
		StatusCode: 200,
		Message:    "success",
		Data:       ocm.LoginProofInvitationResponseData{},
	}

	ocmServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusOK)
		_ = json.NewEncoder(w).Encode(expected)
	}))

	client := ocm.New(ocmServer.URL)
	res, err := client.GetLoginProofInvitation(context.Background(), []string{"principalMembershipCredential"})

	assert.NoError(t, err)
	assert.Equal(t, expected.StatusCode, res.StatusCode)
	assert.Equal(t, expected.Message, res.Message)
	assert.Equal(t, expected.Data, res.Data)
}

func Test_GetLoginProofInvitationErr(t *testing.T) {
	ocmServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusInternalServerError)
	}))

	client := ocm.New(ocmServer.URL)
	res, err := client.GetLoginProofInvitation(context.Background(), []string{"principalMembershipCredential"})

	assert.Nil(t, res)
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "unexpected response code")
}

func TestClient_GetLoginProofResultSuccess(t *testing.T) {
	expected := &ocm.LoginProofResultResponse{
		StatusCode: 200,
		Message:    "success",
		Data:       ocm.LoginProofResultResponseData{},
	}

	ocmServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusOK)
		_ = json.NewEncoder(w).Encode(expected)
	}))

	client := ocm.New(ocmServer.URL)
	res, err := client.GetLoginProofResult(context.Background(), "2cf01406-b15f-4960-a6a7-7bc62cd37a3c")

	assert.NoError(t, err)
	assert.Equal(t, expected.StatusCode, res.StatusCode)
	assert.Equal(t, expected.Message, res.Message)
	assert.Equal(t, expected.Data, res.Data)
}

func Test_GetLoginProofResultErr(t *testing.T) {
	ocmServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusInternalServerError)
	}))

	client := ocm.New(ocmServer.URL)
	res, err := client.GetLoginProofResult(context.Background(), "2cf01406-b15f-4960-a6a7-7bc62cd37a3c")

	assert.Nil(t, res)
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "unexpected response code")
}