term/event/nats/nats_test.go

package nats

import (
	"bytes"
	"context"
	"testing"
	"time"

	nats "github.com/nats-io/nats.go"
	"github.com/nats-io/nats.go/jetstream"
	dataserviceapi "github.com/grassrootseconomics/ussd-data-service/pkg/api"
	"git.defalsify.org/vise.git/db"
	memdb "git.defalsify.org/vise.git/db/mem"
	"git.grassecon.net/urdt/ussd/common"
	"git.grassecon.net/urdt/ussd/config"
	"git.grassecon.net/urdt/ussd/models"
	"git.grassecon.net/term/lookup"
	"git.grassecon.net/term/event"
)

func init() {
}

const (
	aliceChecksum = "0xeae046BF396e91f5A8D74f863dC57c107c8a4a70"
)

// TODO: jetstream, would have been nice of you to provide an easier way to make a mock msg
type testMsg struct {
	data []byte
}

func(m *testMsg) Ack() error {
	return nil
}

func(m *testMsg) Nak() error {
	return nil
}

func(m *testMsg) NakWithDelay(time.Duration) error {
	return nil
}

func(m *testMsg) Data() []byte {
	return m.data
}

func(m *testMsg) Reply() string {
	return ""
}

func(m *testMsg) Subject() string {
	return ""
}

func(m *testMsg) Term() error {
	return nil
}

func(m *testMsg) TermWithReason(string) error {
	return nil
}

func(m *testMsg) DoubleAck(ctx context.Context) error {
	return nil
}

func(m *testMsg) Headers() nats.Header {
	return nats.Header{}
}

func(m *testMsg) InProgress() error {
	return nil
}

func(m *testMsg) Metadata() (*jetstream.MsgMetadata, error) {
	return nil, nil
}

type mockApi struct {
}

func(m mockApi) CheckBalance(ctx context.Context, publicKey string) (*models.BalanceResult, error) {
	return nil, nil
}

func(m mockApi) CreateAccount(ctx context.Context) (*models.AccountResult, error) {
	return nil, nil
}

func(m mockApi) TrackAccountStatus(ctx context.Context, publicKey string) (*models.TrackStatusResult, error) {
	return nil, nil
}

func(m mockApi) FetchVouchers(ctx context.Context, publicKey string) ([]dataserviceapi.TokenHoldings, error) {
	logg.DebugCtxf(ctx, "mockapi fetchvouchers", "key", publicKey)
	return []dataserviceapi.TokenHoldings{
		dataserviceapi.TokenHoldings{
			ContractAddress: "0xeE0A29AE1BB7a033c8277C04780c4aBcf4388E93",
			TokenSymbol: "FOO",
			TokenDecimals: "6",
			Balance: "362436",
		},
	}, nil
}

func(m mockApi) FetchTransactions(ctx context.Context, publicKey string) ([]dataserviceapi.Last10TxResponse, error) {
	logg.DebugCtxf(ctx, "mockapi fetchtransactions", "key", publicKey)
	return []dataserviceapi.Last10TxResponse{
		dataserviceapi.Last10TxResponse{
			Sender: "0xeae046BF396e91f5A8D74f863dC57c107c8a4a70",
			Recipient: "B3117202371853e24B725d4169D87616A7dDb127",
			TransferValue: "1337",
			ContractAddress: "0x765DE816845861e75A25fCA122bb6898B8B1282a",
			TxHash: "0xabcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789",
			DateBlock: time.Unix(1730592500, 0),
			TokenSymbol: "FOO",
			TokenDecimals: "6",
		},
	}, nil
}

func(m mockApi) VoucherData(ctx context.Context, address string) (*models.VoucherDataResult, error) {
	return &models.VoucherDataResult{
		TokenSymbol: "FOO",
		TokenName: "Foo Token",
		TokenDecimals: "6",
		SinkAddress: "0xb42C5920014eE152F2225285219407938469BBfA",
	}, nil
}

func TestHandleMsg(t *testing.T) {
	err := config.LoadConfig()
	if err != nil {
		t.Fatal(err)
	}

	lookup.Api = mockApi{}

	ctx := context.Background()
	userDb := memdb.NewMemDb()
	err = userDb.Connect(ctx, "")
	if err != nil {
		panic(err)
	}

	alice, err := common.NormalizeHex(aliceChecksum)
	if err != nil {
		t.Fatal(err)
	}

	aliceSession := "5553425"
	userDb.SetSession(alice)
	userDb.SetPrefix(db.DATATYPE_USERDATA)
	err = userDb.Put(ctx, common.PackKey(common.DATA_PUBLIC_KEY_REVERSE, []byte{}), []byte(aliceSession))
	if err != nil {
		t.Fatal(err)
	}

	sub := NewNatsSubscription(userDb)

	data := `{
	"block": 42,
	"contractAddress": "0x765DE816845861e75A25fCA122bb6898B8B1282a",
	"success": true,
	"timestamp": 1730592500,
	"transactionHash": "0xabcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789",
	"transactionType": "TOKEN_TRANSFER",
	"payload": {
		"from": "0xeae046BF396e91f5A8D74f863dC57c107c8a4a70",
		"to": "B3117202371853e24B725d4169D87616A7dDb127",
		"value": "1337"
	}
}`
	msg := &testMsg{
		data: []byte(data),
	}
	sub.handleEvent(msg)

	store := common.UserDataStore{
		Db: userDb,
	}
	v, err := store.ReadEntry(ctx, aliceSession, common.DATA_ACTIVE_SYM)
	if err != nil {
		t.Fatal(err)
	}
	if !bytes.Equal(v, []byte("FOO")) {
		t.Fatalf("expected 'FOO', got %s", v)
	}

	v, err = store.ReadEntry(ctx, aliceSession, common.DATA_ACTIVE_BAL)
	if err != nil {
		t.Fatal(err)
	}
	if !bytes.Equal(v, []byte("362436")) {
		t.Fatalf("expected '362436', got %s", v)
	}

	v, err = store.ReadEntry(ctx, aliceSession, common.DATA_TRANSACTIONS)
	if err != nil {
		t.Fatal(err)
	}
	if !bytes.Contains(v, []byte("abcdef")) {
		t.Fatal("no transaction data")
	}

	userDb.SetPrefix(event.DATATYPE_USERSUB)
	userDb.SetSession(aliceSession)
	k := append([]byte("vouchers"), []byte("sym")...)
	v, err = userDb.Get(ctx, k)
	if err != nil {
		t.Fatal(err)
	}
	if !bytes.Contains(v, []byte("1:FOO")) {
		t.Fatalf("expected '1:FOO', got %s", v)
	}
}