api_payments_test.go

package lightning

import (
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"os"
	"testing"
	"time"

	entities "github.com/bolt-observer/go_common/entities"
	"github.com/stretchr/testify/assert"
)

func getAPI(t *testing.T, name string, typ APIType) LightingAPICalls {
	var data entities.Data

	if _, err := os.Stat(name); errors.Is(err, os.ErrNotExist) {
		// If file with credentials does not exist succeed
		return nil
	}

	content, err := os.ReadFile(name)
	if err != nil {
		t.Fatalf("Error when opening file: %v", err)
		return nil
	}

	if _, err := os.Stat(FixtureDir); errors.Is(err, os.ErrNotExist) {
		err := os.Mkdir(FixtureDir, os.ModePerm)
		if err != nil {
			t.Fatalf("Could not create directory: %v", err)
			return nil
		}
	}

	err = json.Unmarshal(content, &data)
	if err != nil {
		t.Fatalf("Error during Unmarshal(): %v", err)
		return nil
	}

	api, err := NewAPI(typ, func() (*entities.Data, error) {
		return &data, nil
	})
	assert.NoError(t, err)

	return api
}

func TestGetInvoicesRest(t *testing.T) {

	api := getAPI(t, "fixture.secret", LndRest)
	if api == nil {
		return
	}

	resp, err := api.GetInvoices(context.Background(), false, Pagination{Reversed: true, BatchSize: 10})
	if err != nil {
		t.Fatalf("Error received %v\n", err)
	}

	fmt.Printf("%+v\n", resp)

	//t.Fail()
}

func TestGetInvoicesGrpc(t *testing.T) {

	api := getAPI(t, "fixture-grpc.secret", LndGrpc)
	if api == nil {
		return
	}

	resp, err := api.GetInvoices(context.Background(), false, Pagination{Reversed: true, BatchSize: 10})
	if err != nil {
		t.Fatalf("Error received %v\n", err)
	}

	fmt.Printf("%+v\n", resp)

	//t.Fail()
}

func TestSubscribeFailedForwards(t *testing.T) {
	var err error
	rest := getAPI(t, "fixture.secret", LndRest)
	if rest == nil {
		return
	}
	grpc := getAPI(t, "fixture-grpc.secret", LndGrpc)
	if grpc == nil {
		return
	}

	a := rest.(*LndRestLightningAPI)
	b := grpc.(*LndGrpcLightningAPI)

	assert.NotNil(t, a)
	assert.NotNil(t, b)

	ctx, cancel := context.WithTimeout(context.Background(), 8*time.Second)
	defer cancel()

	outchan := make(chan RawMessage)

	err = a.SubscribeFailedForwards(ctx, outchan)
	assert.NoError(t, err)
	err = b.SubscribeFailedForwards(ctx, outchan)
	assert.NoError(t, err)

	for {
		select {
		case <-ctx.Done():
			return
		case data := <-outchan:
			fmt.Printf("Received %v\n", data)
		}
	}
}