testutil.go

package messaging

import (
	"context"
	"crypto/ed25519"
	"strings"
	"testing"
	"time"

	"github.com/mr-tron/base58"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"google.golang.org/grpc"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
	"google.golang.org/protobuf/proto"
	"google.golang.org/protobuf/types/known/timestamppb"

	commonpb "github.com/code-payments/code-protobuf-api/generated/go/common/v1"
	messagingpb "github.com/code-payments/code-protobuf-api/generated/go/messaging/v1"
	transactionpb "github.com/code-payments/code-protobuf-api/generated/go/transaction/v2"

	"github.com/code-payments/code-server/pkg/code/auth"
	"github.com/code-payments/code-server/pkg/code/common"
	code_data "github.com/code-payments/code-server/pkg/code/data"
	"github.com/code-payments/code-server/pkg/code/data/account"
	"github.com/code-payments/code-server/pkg/code/data/currency"
	"github.com/code-payments/code-server/pkg/code/data/messaging"
	"github.com/code-payments/code-server/pkg/code/data/paymentrequest"
	"github.com/code-payments/code-server/pkg/code/data/rendezvous"
	exchange_rate_util "github.com/code-payments/code-server/pkg/code/exchangerate"
	"github.com/code-payments/code-server/pkg/code/thirdparty"
	"github.com/code-payments/code-server/pkg/kin"
	"github.com/code-payments/code-server/pkg/pointer"
	"github.com/code-payments/code-server/pkg/testutil"
)

type testEnv struct {
	client1 *clientEnv
	client2 *clientEnv
	server1 *serverEnv
	server2 *serverEnv
}

func setup(t *testing.T, enableMultiServer bool) (env testEnv, cleanup func()) {
	conn1, serv1, err := testutil.NewServer()
	require.NoError(t, err)

	conn2, serv2, err := testutil.NewServer()
	require.NoError(t, err)

	data := code_data.NewTestDataProvider()

	env.client1 = &clientEnv{
		ctx:              context.Background(),
		client:           messagingpb.NewMessagingClient(conn1),
		conf:             &clientConf{},
		streams:          make(map[string][]*cancellableStream),
		directDataAccess: data,
	}
	env.client2 = &clientEnv{
		ctx:              context.Background(),
		client:           messagingpb.NewMessagingClient(conn1),
		conf:             &clientConf{},
		streams:          make(map[string][]*cancellableStream),
		directDataAccess: data,
	}
	if enableMultiServer {
		env.client2.client = messagingpb.NewMessagingClient(conn2)
	}

	subsidizer := testutil.SetupRandomSubsidizer(t, data)

	require.NoError(t, data.ImportExchangeRates(context.Background(), &currency.MultiRateRecord{
		Time: exchange_rate_util.GetLatestExchangeRateTime(),
		Rates: map[string]float64{
			"usd": 0.1,
		},
	}))

	s1 := NewMessagingClientAndServer(data, auth.NewRPCSignatureVerifier(data), conn1.Target(), withManualTestOverrides(&testOverrides{}))
	s1.domainVerifier = mockDomainVerifier
	env.server1 = &serverEnv{
		ctx:        context.Background(),
		server:     s1,
		subsidizer: subsidizer,
	}

	s2 := NewMessagingClientAndServer(data, auth.NewRPCSignatureVerifier(data), conn2.Target(), withManualTestOverrides(&testOverrides{}))
	s2.domainVerifier = mockDomainVerifier
	env.server2 = &serverEnv{
		ctx:        context.Background(),
		server:     s2,
		subsidizer: subsidizer,
	}

	serv1.RegisterService(func(server *grpc.Server) {
		messagingpb.RegisterMessagingServer(server, s1)
	})
	serv2.RegisterService(func(server *grpc.Server) {
		messagingpb.RegisterMessagingServer(server, s2)
	})

	cleanup1, err := serv1.Serve()
	require.NoError(t, err)

	cleanup2, err := serv2.Serve()
	require.NoError(t, err)

	return env, func() {
		cleanup1()
		cleanup2()
	}
}

type serverEnv struct {
	ctx        context.Context
	server     *server
	subsidizer *common.Account
}

func (s *serverEnv) getMessages(t *testing.T, rendezvousKey *common.Account) []*messaging.Record {
	messages, err := s.server.data.GetMessages(s.ctx, rendezvousKey.PublicKey().ToBase58())
	require.NoError(t, err)
	return messages
}

func (s *serverEnv) assertNoMessages(t *testing.T, rendezvousKey *common.Account) {
	messages, err := s.server.data.GetMessages(s.ctx, rendezvousKey.PublicKey().ToBase58())
	require.NoError(t, err)
	assert.Empty(t, messages)
}

func (s *serverEnv) assertPaymentRequestRecordSaved(t *testing.T, rendezvousKey *common.Account, msg *messagingpb.RequestToReceiveBill) {
	var asciiBaseDomain string
	var err error
	if msg.Domain != nil {
		asciiBaseDomain, err = thirdparty.GetAsciiBaseDomain(msg.Domain.Value)
		require.NoError(t, err)
	}

	requestRecord, err := s.server.data.GetRequest(s.ctx, rendezvousKey.PublicKey().ToBase58())
	require.NoError(t, err)

	assert.Equal(t, requestRecord.Intent, rendezvousKey.PublicKey().ToBase58())
	assert.Equal(t, base58.Encode(msg.RequestorAccount.Value), *requestRecord.DestinationTokenAccount)

	switch typed := msg.ExchangeData.(type) {
	case *messagingpb.RequestToReceiveBill_Exact:
		assert.EqualValues(t, typed.Exact.Currency, *requestRecord.ExchangeCurrency)
		assert.Equal(t, typed.Exact.NativeAmount, *requestRecord.NativeAmount)
		require.NotNil(t, requestRecord.ExchangeRate)
		assert.Equal(t, typed.Exact.ExchangeRate, *requestRecord.ExchangeRate)
		require.NotNil(t, requestRecord.Quantity)
		assert.Equal(t, typed.Exact.Quarks, *requestRecord.Quantity)
	case *messagingpb.RequestToReceiveBill_Partial:
		assert.EqualValues(t, typed.Partial.Currency, *requestRecord.ExchangeCurrency)
		assert.Equal(t, typed.Partial.NativeAmount, *requestRecord.NativeAmount)
		assert.Nil(t, requestRecord.ExchangeRate)
		assert.Nil(t, requestRecord.Quantity)
	default:
		require.Fail(t, "unhandled exchange data type")
	}

	if len(asciiBaseDomain) == 0 {
		assert.Nil(t, requestRecord.Domain)
		assert.False(t, requestRecord.IsVerified)
	} else {
		require.NotNil(t, requestRecord.Domain)
		assert.Equal(t, asciiBaseDomain, *requestRecord.Domain)
		assert.Equal(t, msg.Verifier != nil, requestRecord.IsVerified)
	}

	require.Len(t, requestRecord.Fees, len(msg.AdditionalFees))
	for i, expectedFee := range msg.AdditionalFees {
		assert.Equal(t, base58.Encode(expectedFee.Destination.Value), requestRecord.Fees[i].DestinationTokenAccount)
		assert.EqualValues(t, expectedFee.FeeBps, requestRecord.Fees[i].BasisPoints)
	}
}

func (s *serverEnv) assertLoginRequestRecordSaved(t *testing.T, rendezvousKey *common.Account, msg *messagingpb.RequestToLogin) {
	asciiBaseDomain, err := thirdparty.GetAsciiBaseDomain(msg.Domain.Value)
	require.NoError(t, err)

	requestRecord, err := s.server.data.GetRequest(s.ctx, rendezvousKey.PublicKey().ToBase58())
	require.NoError(t, err)

	assert.Equal(t, requestRecord.Intent, rendezvousKey.PublicKey().ToBase58())

	require.NotNil(t, requestRecord.Domain)
	assert.Equal(t, asciiBaseDomain, *requestRecord.Domain)
	assert.True(t, requestRecord.IsVerified)

	assert.Nil(t, requestRecord.DestinationTokenAccount)
	assert.Nil(t, requestRecord.ExchangeCurrency)
	assert.Nil(t, requestRecord.NativeAmount)
	assert.Nil(t, requestRecord.ExchangeRate)
	assert.Nil(t, requestRecord.Quantity)
}

func (s *serverEnv) assertRequestRecordNotSaved(t *testing.T, rendezvousKey *common.Account) {
	_, err := s.server.data.GetRequest(s.ctx, rendezvousKey.PublicKey().ToBase58())
	assert.Equal(t, paymentrequest.ErrPaymentRequestNotFound, err)
}

func (s *serverEnv) assertInitialRendezvousRecordSaved(t *testing.T, rendezvousKey *common.Account) {
	start := time.Now()

	for i := 0; i < 5; i++ {
		rendezvousRecord, err := s.server.data.GetRendezvous(s.ctx, rendezvousKey.PublicKey().ToBase58())
		if err == rendezvous.ErrNotFound {
			time.Sleep(10 * time.Millisecond)
			continue
		}
		require.NoError(t, err)

		assert.Equal(t, rendezvousKey.PublicKey().ToBase58(), rendezvousRecord.Key)
		assert.Equal(t, s.server.broadcastAddress, rendezvousRecord.Location) // Note: assertion must be called on the expected server
		assert.True(t, start.Sub(rendezvousRecord.CreatedAt) <= 50*time.Millisecond)
		assert.True(t, start.Sub(rendezvousRecord.CreatedAt) >= -50*time.Millisecond)
		assert.Equal(t, rendezvousRecord.CreatedAt.Unix(), rendezvousRecord.LastUpdatedAt.Unix())
		return
	}

	require.Fail(t, "rendezvous record not saved")
}

func (s *serverEnv) assertRendezvousRecordRefreshed(t *testing.T, rendezvousKey *common.Account) {
	rendezvousRecord, err := s.server.data.GetRendezvous(s.ctx, rendezvousKey.PublicKey().ToBase58())
	require.NoError(t, err)

	// Assumes it's the first time for simplicity
	assert.True(t, rendezvousRecord.LastUpdatedAt.Sub(rendezvousRecord.CreatedAt) > rendezvousRecordMaxAge/2)
	assert.True(t, rendezvousRecord.LastUpdatedAt.Sub(rendezvousRecord.CreatedAt) < rendezvousRecordMaxAge)
}

func (s *serverEnv) assertRendezvousRecordDeleted(t *testing.T, rendezvousKey *common.Account) {
	for i := 0; i < 5; i++ {
		_, err := s.server.data.GetRendezvous(s.ctx, rendezvousKey.PublicKey().ToBase58())
		if err == nil {
			time.Sleep(10 * time.Millisecond)
			continue
		}
		assert.Equal(t, rendezvous.ErrNotFound, err)
	}
}

type cancellableStream struct {
	stream               messagingpb.Messaging_OpenMessageStreamClient
	streamWithKeepAlives messagingpb.Messaging_OpenMessageStreamWithKeepAliveClient
	cancel               func()
}

type clientConf struct {
	// Simulations for invalid account

	simulateStaleRequestorAccountType bool
	simulateInvalidAccountType        bool
	simulateAccountNotCodeAccount     bool

	// Simulations for invalid exchange data

	simulateInvalidCurrency        bool
	simulateInvalidExchangeRate    bool
	simulateInvalidNativeAmount    bool
	simulateSmallNativeAmount      bool
	simulateLargeNativeAmount      bool
	simulateFractionalNativeAmount bool
	simulateFractionalQuarkAmount  bool

	// Simulations for invalid relationships

	simulateInvalidRelationship bool
	simulateInvalidDomain       bool
	simulateDoesntOwnDomain     bool

	// Simulations for invalid signatures

	simulateInvalidRequestSignature bool
	simulateInvalidMessageSignature bool

	// Simulations for invalid rendezvous keys

	simulateInvalidRendezvousKey bool

	// Simulations for invalid fee structures
	simulateLargeFeePercentage           bool
	simulateInvalidFeeCodeAccount        bool
	simulateInvalidFeeRelationship       bool
	simulateDuplicatedFeeTaker           bool
	simulateFeeTakerIsPaymentDestination bool
}

type clientEnv struct {
	ctx     context.Context
	client  messagingpb.MessagingClient
	conf    *clientConf
	streams map[string][]*cancellableStream

	// Direct data access to help test/pass validation checks
	directDataAccess code_data.Provider
}

func (c *clientEnv) openMessageStream(t *testing.T, rendezvousKey *common.Account, enableKeepAlive bool) {
	cancellableCtx, cancel := context.WithCancel(c.ctx)

	req := &messagingpb.OpenMessageStreamRequest{
		RendezvousKey: &messagingpb.RendezvousKey{
			Value: rendezvousKey.PublicKey().ToBytes(),
		},
	}

	reqBytes, err := proto.Marshal(req)
	require.NoError(t, err)

	req.Signature = &commonpb.Signature{
		Value: ed25519.Sign(rendezvousKey.PrivateKey().ToBytes(), reqBytes),
	}

	if enableKeepAlive {
		streamer, err := c.client.OpenMessageStreamWithKeepAlive(cancellableCtx)
		require.NoError(t, err)

		require.NoError(t, streamer.Send(&messagingpb.OpenMessageStreamWithKeepAliveRequest{
			RequestOrPong: &messagingpb.OpenMessageStreamWithKeepAliveRequest_Request{
				Request: req,
			},
		}))

		c.streams[rendezvousKey.PublicKey().ToBase58()] = append(c.streams[rendezvousKey.PublicKey().ToBase58()], &cancellableStream{
			streamWithKeepAlives: streamer,
			cancel:               cancel,
		})
	} else {
		streamer, err := c.client.OpenMessageStream(cancellableCtx, req)
		require.NoError(t, err)
		c.streams[rendezvousKey.PublicKey().ToBase58()] = append(c.streams[rendezvousKey.PublicKey().ToBase58()], &cancellableStream{
			stream: streamer,
			cancel: cancel,
		})
	}
}

func (c *clientEnv) closeMessageStream(t *testing.T, rendezvousKey *common.Account) {
	streamers, ok := c.streams[rendezvousKey.PublicKey().ToBase58()]
	require.True(t, ok)
	for _, streamer := range streamers {
		streamer.cancel()
	}
	delete(c.streams, rendezvousKey.PublicKey().ToBase58())
}

func (c *clientEnv) receiveMessagesInRealTime(t *testing.T, rendezvousKey *common.Account) []*messagingpb.Message {
	streamers, ok := c.streams[rendezvousKey.PublicKey().ToBase58()]
	require.True(t, ok)

	for _, streamer := range streamers {
		if streamer.streamWithKeepAlives != nil {
			for {
				resp, err := streamer.streamWithKeepAlives.Recv()

				status, ok := status.FromError(err)
				if ok && status.Code() == codes.Aborted {
					// Try the next open stream
					break
				}

				require.NoError(t, err)

				switch typed := resp.ResponseOrPing.(type) {
				case *messagingpb.OpenMessageStreamWithKeepAliveResponse_Response:
					return typed.Response.Messages
				case *messagingpb.OpenMessageStreamWithKeepAliveResponse_Ping:
					require.NoError(t, streamer.streamWithKeepAlives.Send(&messagingpb.OpenMessageStreamWithKeepAliveRequest{
						RequestOrPong: &messagingpb.OpenMessageStreamWithKeepAliveRequest_Pong{
							Pong: &messagingpb.ClientPong{
								Timestamp: timestamppb.Now(),
							},
						},
					}))
				default:
					require.Fail(t, "response and ping wasn't set")
				}
			}
		} else {
			resp, err := streamer.stream.Recv()

			status, ok := status.FromError(err)
			if ok && status.Code() == codes.Aborted {
				// Try the next open stream
				continue
			}

			require.NoError(t, err)
			return resp.Messages
		}
	}

	require.Fail(t, "all active streams have been aborted")
	return nil
}

func (c *clientEnv) pollForMessages(t *testing.T, rendezvousKey *common.Account) []*messagingpb.Message {
	req := &messagingpb.PollMessagesRequest{
		RendezvousKey: &messagingpb.RendezvousKey{
			Value: rendezvousKey.PublicKey().ToBytes(),
		},
	}

	reqBytes, err := proto.Marshal(req)
	require.NoError(t, err)

	req.Signature = &commonpb.Signature{
		Value: ed25519.Sign(rendezvousKey.PrivateKey().ToBytes(), reqBytes),
	}

	resp, err := c.client.PollMessages(c.ctx, req)
	require.NoError(t, err)
	return resp.Messages
}

func (c *clientEnv) waitUntilStreamTerminationOrTimeout(t *testing.T, rendezvousKey *common.Account, keepStreamAlive bool, timeout time.Duration) int {
	streamers, ok := c.streams[rendezvousKey.PublicKey().ToBase58()]
	require.True(t, ok)
	require.Len(t, streamers, 1)

	streamer := streamers[0]
	require.NotNil(t, streamer.streamWithKeepAlives)

	var pingCount int
	var lastPingTs time.Time
	start := time.Now()
	for {
		resp, err := streamer.streamWithKeepAlives.Recv()

		status, ok := status.FromError(err)
		if ok && status.Code() == codes.Aborted {
			return pingCount
		}

		require.NoError(t, err)

		switch typed := resp.ResponseOrPing.(type) {
		case *messagingpb.OpenMessageStreamWithKeepAliveResponse_Ping:
			pingTimeEsimate := start
			if pingCount > 0 {
				pingTimeEsimate = lastPingTs.Add(messageStreamPingDelay)
			}

			deltaPingTime := typed.Ping.Timestamp.AsTime().Sub(pingTimeEsimate)
			assert.True(t, deltaPingTime <= 50*time.Millisecond)
			assert.True(t, deltaPingTime >= -50*time.Millisecond)

			assert.Equal(t, messageStreamPingDelay, typed.Ping.PingDelay.AsDuration())
			if pingCount == 0 {
				// First ping should come immediately
				assert.True(t, time.Since(start) <= 50*time.Millisecond)
			} else {
				// Every other ping comes at the defined time interval
				assert.True(t, time.Since(lastPingTs) >= messageStreamPingDelay-50*time.Millisecond)
				assert.True(t, time.Since(lastPingTs) <= messageStreamPingDelay+50*time.Millisecond)
			}

			pingCount += 1
			lastPingTs = time.Now()

			if keepStreamAlive {
				require.NoError(t, streamer.streamWithKeepAlives.Send(&messagingpb.OpenMessageStreamWithKeepAliveRequest{
					RequestOrPong: &messagingpb.OpenMessageStreamWithKeepAliveRequest_Pong{
						Pong: &messagingpb.ClientPong{
							Timestamp: timestamppb.Now(),
						},
					},
				}))
			}

			if time.Since(start) > timeout {
				return pingCount
			}
		case *messagingpb.OpenMessageStreamWithKeepAliveResponse_Response:
		default:
			require.Fail(t, "response and ping wasn't set")
		}
	}
}

type sendMessageCallMetadata struct {
	req  *messagingpb.SendMessageRequest
	resp *messagingpb.SendMessageResponse
	err  error
}

func (c *sendMessageCallMetadata) requireSuccess(t *testing.T) {
	require.NoError(t, c.err)
	require.Equal(t, messagingpb.SendMessageResponse_OK, c.resp.Result)
}

func (c *sendMessageCallMetadata) assertInvalidMessageError(t *testing.T, message string) {
	require.Error(t, c.err)
	require.Nil(t, c.resp)

	status, ok := status.FromError(c.err)
	require.True(t, ok)
	assert.Equal(t, codes.InvalidArgument, status.Code())
	assert.True(t, strings.Contains(strings.ToLower(status.Message()), strings.ToLower(message)))
}

func (c *sendMessageCallMetadata) assertUnauthenticatedError(t *testing.T, message string) {
	require.Error(t, c.err)
	require.Nil(t, c.resp)

	status, ok := status.FromError(c.err)
	require.True(t, ok)
	assert.Equal(t, codes.Unauthenticated, status.Code())
	assert.True(t, strings.Contains(strings.ToLower(status.Message()), strings.ToLower(message)))
}

func (c *sendMessageCallMetadata) assertPermissionDeniedError(t *testing.T, message string) {
	require.Error(t, c.err)
	require.Nil(t, c.resp)

	status, ok := status.FromError(c.err)
	require.True(t, ok)
	assert.Equal(t, codes.PermissionDenied, status.Code())
	assert.True(t, strings.Contains(strings.ToLower(status.Message()), strings.ToLower(message)))
}

func (c *sendMessageCallMetadata) assertNoActiveStreamError(t *testing.T) {
	require.NoError(t, c.err)
	require.Equal(t, messagingpb.SendMessageResponse_NO_ACTIVE_STREAM, c.resp.Result)
}

func (c *clientEnv) sendRequestToGrabBillMessage(t *testing.T, rendezvousKey *common.Account) *sendMessageCallMetadata {
	destination := testutil.NewRandomAccount(t)

	accountInfoRecord := &account.Record{
		OwnerAccount:     testutil.NewRandomAccount(t).PublicKey().ToBase58(),
		AuthorityAccount: testutil.NewRandomAccount(t).PublicKey().ToBase58(),
		TokenAccount:     destination.PublicKey().ToBase58(),
		MintAccount:      common.KinMintAccount.PublicKey().ToBase58(),
		AccountType:      commonpb.AccountType_TEMPORARY_INCOMING,
		Index:            0,
	}
	if c.conf.simulateInvalidAccountType {
		accountInfoRecord.AccountType = commonpb.AccountType_TEMPORARY_OUTGOING
	}
	if !c.conf.simulateAccountNotCodeAccount {
		require.NoError(t, c.directDataAccess.CreateAccountInfo(c.ctx, accountInfoRecord))
	}

	if c.conf.simulateStaleRequestorAccountType {
		latestAccountInfoRecord := &account.Record{
			OwnerAccount:     accountInfoRecord.OwnerAccount,
			AuthorityAccount: testutil.NewRandomAccount(t).PublicKey().ToBase58(),
			TokenAccount:     testutil.NewRandomAccount(t).PublicKey().ToBase58(),
			MintAccount:      common.KinMintAccount.PublicKey().ToBase58(),
			AccountType:      commonpb.AccountType_TEMPORARY_INCOMING,
			Index:            accountInfoRecord.Index + 1,
		}
		require.NoError(t, c.directDataAccess.CreateAccountInfo(c.ctx, latestAccountInfoRecord))
	}

	req := &messagingpb.SendMessageRequest{
		Message: &messagingpb.Message{
			Kind: &messagingpb.Message_RequestToGrabBill{
				RequestToGrabBill: &messagingpb.RequestToGrabBill{
					RequestorAccount: destination.ToProto(),
				},
			},
		},
		RendezvousKey: &messagingpb.RendezvousKey{
			Value: rendezvousKey.PublicKey().ToBytes(),
		},
	}

	return c.sendMessage(t, req, rendezvousKey)
}

type testRequestToReceiveBillConf struct {
	usePrimaryAccount         bool
	useRelationshipAccount    bool
	disableDomainVerification bool
}

// todo: code duplication with fiat variant
func (c *clientEnv) sendRequestToReceiveKinBillMessage(
	t *testing.T,
	rendezvousKey *common.Account,
	conf *testRequestToReceiveBillConf,
) *sendMessageCallMetadata {
	authority, err := common.NewAccountFromPrivateKeyString("dr2MUzL4NCS45qyp16vDXiSdHqqdg2DF79xKaYMB1vzVtDDjPvyQ8xTH4VsTWXSDP3NFzsdCV6gEoChKftzwLno")
	require.NoError(t, err)

	if c.conf.simulateDoesntOwnDomain {
		authority = testutil.NewRandomAccount(t)
	}

	destination := testutil.NewRandomAccount(t)

	if conf.usePrimaryAccount {
		owner := testutil.NewRandomAccount(t)
		accountInfoRecord := &account.Record{
			OwnerAccount:     owner.PublicKey().ToBase58(),
			AuthorityAccount: owner.PublicKey().ToBase58(),
			TokenAccount:     destination.PublicKey().ToBase58(),
			MintAccount:      common.KinMintAccount.PublicKey().ToBase58(),
			AccountType:      commonpb.AccountType_PRIMARY,
			Index:            0,
		}
		require.NoError(t, c.directDataAccess.CreateAccountInfo(c.ctx, accountInfoRecord))
	} else if conf.useRelationshipAccount {
		accountInfoRecord := &account.Record{
			OwnerAccount:     testutil.NewRandomAccount(t).PublicKey().ToBase58(),
			AuthorityAccount: testutil.NewRandomAccount(t).PublicKey().ToBase58(),
			TokenAccount:     destination.PublicKey().ToBase58(),
			MintAccount:      common.KinMintAccount.PublicKey().ToBase58(),
			AccountType:      commonpb.AccountType_RELATIONSHIP,
			Index:            0,
			RelationshipTo:   pointer.String("getcode.com"),
		}
		if c.conf.simulateInvalidRelationship {
			accountInfoRecord.RelationshipTo = pointer.String("example.com")
		}
		require.NoError(t, c.directDataAccess.CreateAccountInfo(c.ctx, accountInfoRecord))
	}

	if c.conf.simulateInvalidAccountType {
		accountInfoRecord := &account.Record{
			OwnerAccount:     testutil.NewRandomAccount(t).PublicKey().ToBase58(),
			AuthorityAccount: testutil.NewRandomAccount(t).PublicKey().ToBase58(),
			TokenAccount:     destination.PublicKey().ToBase58(),
			MintAccount:      common.KinMintAccount.PublicKey().ToBase58(),
			AccountType:      commonpb.AccountType_TEMPORARY_INCOMING,
			Index:            0,
		}
		require.NoError(t, c.directDataAccess.CreateAccountInfo(c.ctx, accountInfoRecord))
	}

	exchangeData := &transactionpb.ExchangeData{
		Currency:     "kin",
		ExchangeRate: 1.0,
		NativeAmount: 10_001,
		Quarks:       kin.ToQuarks(10_001),
	}

	if c.conf.simulateInvalidCurrency {
		exchangeData.Currency = "usd"
	}
	if c.conf.simulateInvalidExchangeRate {
		exchangeData.ExchangeRate *= 2.0
		exchangeData.NativeAmount *= 2.0
	}
	if c.conf.simulateInvalidNativeAmount {
		exchangeData.NativeAmount += 2
	}
	if c.conf.simulateSmallNativeAmount {
		exchangeData.NativeAmount = 1
		exchangeData.Quarks = kin.ToQuarks(1)
	}
	if c.conf.simulateLargeNativeAmount {
		exchangeData.NativeAmount = 100_001
		exchangeData.Quarks = kin.ToQuarks(100_001)
	}
	if c.conf.simulateFractionalNativeAmount {
		exchangeData.NativeAmount += 0.1
	}
	if c.conf.simulateFractionalQuarkAmount {
		exchangeData.Quarks += kin.QuarksPerKin / 10
	}

	additionalFees := []*transactionpb.AdditionalFeePayment{
		{
			Destination: testutil.NewRandomAccount(t).ToProto(),
			FeeBps:      1234,
		},
		{
			Destination: testutil.NewRandomAccount(t).ToProto(),
			FeeBps:      56,
		},
		{
			Destination: testutil.NewRandomAccount(t).ToProto(),
			FeeBps:      789,
		},
	}

	if c.conf.simulateLargeFeePercentage {
		additionalFees[0].FeeBps = 8000
	}
	if c.conf.simulateDuplicatedFeeTaker {
		additionalFees[0].Destination = additionalFees[2].Destination
	}
	if c.conf.simulateFeeTakerIsPaymentDestination {
		additionalFees[0].Destination = destination.ToProto()
	}

	feeCodeAccountOwner := testutil.NewRandomAccount(t)
	feeCodeAccountAuthority := feeCodeAccountOwner
	feeCodeAccountType := commonpb.AccountType_PRIMARY
	if c.conf.simulateInvalidFeeCodeAccount {
		feeCodeAccountType = commonpb.AccountType_TEMPORARY_INCOMING
		feeCodeAccountAuthority = testutil.NewRandomAccount(t)
	}
	require.NoError(t, c.directDataAccess.CreateAccountInfo(c.ctx, &account.Record{
		OwnerAccount:     feeCodeAccountOwner.PublicKey().ToBase58(),
		AuthorityAccount: feeCodeAccountAuthority.PublicKey().ToBase58(),
		TokenAccount:     base58.Encode(additionalFees[0].Destination.Value),
		MintAccount:      common.KinMintAccount.PublicKey().ToBase58(),
		AccountType:      feeCodeAccountType,
		Index:            0,
	}))
	if !conf.disableDomainVerification {
		feeRelationship := "getcode.com"
		if c.conf.simulateInvalidFeeRelationship {
			feeRelationship = "example.com"
		}
		require.NoError(t, c.directDataAccess.CreateAccountInfo(c.ctx, &account.Record{
			OwnerAccount:     feeCodeAccountOwner.PublicKey().ToBase58(),
			AuthorityAccount: testutil.NewRandomAccount(t).PublicKey().ToBase58(),
			TokenAccount:     base58.Encode(additionalFees[1].Destination.Value),
			MintAccount:      common.KinMintAccount.PublicKey().ToBase58(),
			AccountType:      commonpb.AccountType_RELATIONSHIP,
			Index:            0,
			RelationshipTo:   &feeRelationship,
		}))
	}

	msg := &messagingpb.RequestToReceiveBill{
		RequestorAccount: destination.ToProto(),
		ExchangeData: &messagingpb.RequestToReceiveBill_Exact{
			Exact: exchangeData,
		},
		Domain: &commonpb.Domain{
			Value: "app.getcode.com",
		},
		Verifier: authority.ToProto(),
		RendezvousKey: &messagingpb.RendezvousKey{
			Value: rendezvousKey.PublicKey().ToBytes(),
		},
		AdditionalFees: additionalFees,
	}

	if conf.disableDomainVerification {
		msg.Verifier = nil
		msg.RendezvousKey = nil
	}

	if c.conf.simulateInvalidDomain {
		msg.Domain.Value = "localhost"
	}

	if c.conf.simulateInvalidRendezvousKey {
		msg.RendezvousKey.Value = testutil.NewRandomAccount(t).PublicKey().ToBytes()
	}

	messageBytes, err := proto.Marshal(msg)
	require.NoError(t, err)

	if !conf.disableDomainVerification {
		signer := authority
		if c.conf.simulateInvalidMessageSignature {
			signer = testutil.NewRandomAccount(t)
		}
		msg.Signature = &commonpb.Signature{
			Value: ed25519.Sign(signer.PrivateKey().ToBytes(), messageBytes),
		}
	}

	req := &messagingpb.SendMessageRequest{
		Message: &messagingpb.Message{
			Kind: &messagingpb.Message_RequestToReceiveBill{
				RequestToReceiveBill: msg,
			},
		},
		RendezvousKey: &messagingpb.RendezvousKey{
			Value: rendezvousKey.PublicKey().ToBytes(),
		},
	}

	return c.sendMessage(t, req, rendezvousKey)
}

// todo: code duplication with kin variant
func (c *clientEnv) sendRequestToReceiveFiatBillMessage(
	t *testing.T,
	rendezvousKey *common.Account,
	conf *testRequestToReceiveBillConf,
) *sendMessageCallMetadata {
	authority, err := common.NewAccountFromPrivateKeyString("dr2MUzL4NCS45qyp16vDXiSdHqqdg2DF79xKaYMB1vzVtDDjPvyQ8xTH4VsTWXSDP3NFzsdCV6gEoChKftzwLno")
	require.NoError(t, err)

	if c.conf.simulateDoesntOwnDomain {
		authority = testutil.NewRandomAccount(t)
	}

	destination := testutil.NewRandomAccount(t)

	if conf.usePrimaryAccount {
		owner := testutil.NewRandomAccount(t)
		require.NoError(t, c.directDataAccess.CreateAccountInfo(c.ctx, &account.Record{
			OwnerAccount:     owner.PublicKey().ToBase58(),
			AuthorityAccount: owner.PublicKey().ToBase58(),
			TokenAccount:     destination.PublicKey().ToBase58(),
			MintAccount:      common.KinMintAccount.PublicKey().ToBase58(),
			AccountType:      commonpb.AccountType_PRIMARY,
			Index:            0,
		}))
	} else if conf.useRelationshipAccount {
		accountInfoRecord := &account.Record{
			OwnerAccount:     testutil.NewRandomAccount(t).PublicKey().ToBase58(),
			AuthorityAccount: testutil.NewRandomAccount(t).PublicKey().ToBase58(),
			TokenAccount:     destination.PublicKey().ToBase58(),
			MintAccount:      common.KinMintAccount.PublicKey().ToBase58(),
			AccountType:      commonpb.AccountType_RELATIONSHIP,
			Index:            0,
			RelationshipTo:   pointer.String("getcode.com"),
		}
		if c.conf.simulateInvalidRelationship {
			accountInfoRecord.RelationshipTo = pointer.String("example.com")
		}
		require.NoError(t, c.directDataAccess.CreateAccountInfo(c.ctx, accountInfoRecord))
	}

	if c.conf.simulateInvalidAccountType {
		accountInfoRecord := &account.Record{
			OwnerAccount:     testutil.NewRandomAccount(t).PublicKey().ToBase58(),
			AuthorityAccount: testutil.NewRandomAccount(t).PublicKey().ToBase58(),
			TokenAccount:     destination.PublicKey().ToBase58(),
			MintAccount:      common.KinMintAccount.PublicKey().ToBase58(),
			AccountType:      commonpb.AccountType_TEMPORARY_INCOMING,
			Index:            0,
		}
		require.NoError(t, c.directDataAccess.CreateAccountInfo(c.ctx, accountInfoRecord))
	}

	exchangeData := &transactionpb.ExchangeDataWithoutRate{
		Currency:     "usd",
		NativeAmount: .50,
	}
	if c.conf.simulateInvalidCurrency {
		exchangeData.Currency = "kin"
	}
	if c.conf.simulateSmallNativeAmount {
		exchangeData.NativeAmount = 0.01
	}
	if c.conf.simulateLargeNativeAmount {
		exchangeData.NativeAmount = 1.01
	}

	additionalFees := []*transactionpb.AdditionalFeePayment{
		{
			Destination: testutil.NewRandomAccount(t).ToProto(),
			FeeBps:      1234,
		},
		{
			Destination: testutil.NewRandomAccount(t).ToProto(),
			FeeBps:      56,
		},
		{
			Destination: testutil.NewRandomAccount(t).ToProto(),
			FeeBps:      789,
		},
	}

	if c.conf.simulateLargeFeePercentage {
		additionalFees[0].FeeBps = 8000
	}
	if c.conf.simulateDuplicatedFeeTaker {
		additionalFees[0].Destination = additionalFees[2].Destination
	}
	if c.conf.simulateFeeTakerIsPaymentDestination {
		additionalFees[0].Destination = destination.ToProto()
	}

	feeCodeAccountOwner := testutil.NewRandomAccount(t)
	feeCodeAccountAuthority := feeCodeAccountOwner
	feeCodeAccountType := commonpb.AccountType_PRIMARY
	if c.conf.simulateInvalidFeeCodeAccount {
		feeCodeAccountType = commonpb.AccountType_TEMPORARY_INCOMING
		feeCodeAccountAuthority = testutil.NewRandomAccount(t)
	}
	require.NoError(t, c.directDataAccess.CreateAccountInfo(c.ctx, &account.Record{
		OwnerAccount:     feeCodeAccountOwner.PublicKey().ToBase58(),
		AuthorityAccount: feeCodeAccountAuthority.PublicKey().ToBase58(),
		TokenAccount:     base58.Encode(additionalFees[0].Destination.Value),
		MintAccount:      common.KinMintAccount.PublicKey().ToBase58(),
		AccountType:      feeCodeAccountType,
		Index:            0,
	}))
	if !conf.disableDomainVerification {
		feeRelationship := "getcode.com"
		if c.conf.simulateInvalidFeeRelationship {
			feeRelationship = "example.com"
		}
		require.NoError(t, c.directDataAccess.CreateAccountInfo(c.ctx, &account.Record{
			OwnerAccount:     feeCodeAccountOwner.PublicKey().ToBase58(),
			AuthorityAccount: testutil.NewRandomAccount(t).PublicKey().ToBase58(),
			TokenAccount:     base58.Encode(additionalFees[1].Destination.Value),
			MintAccount:      common.KinMintAccount.PublicKey().ToBase58(),
			AccountType:      commonpb.AccountType_RELATIONSHIP,
			Index:            0,
			RelationshipTo:   &feeRelationship,
		}))
	}

	msg := &messagingpb.RequestToReceiveBill{
		RequestorAccount: destination.ToProto(),
		ExchangeData: &messagingpb.RequestToReceiveBill_Partial{
			Partial: exchangeData,
		},
		Domain: &commonpb.Domain{
			Value: "app.getcode.com",
		},
		Verifier: authority.ToProto(),
		RendezvousKey: &messagingpb.RendezvousKey{
			Value: rendezvousKey.PublicKey().ToBytes(),
		},
		AdditionalFees: additionalFees,
	}

	if conf.disableDomainVerification {
		msg.Verifier = nil
		msg.RendezvousKey = nil
	}

	if c.conf.simulateInvalidDomain {
		msg.Domain.Value = "localhost"
	}

	if c.conf.simulateInvalidRendezvousKey {
		msg.RendezvousKey.Value = testutil.NewRandomAccount(t).PublicKey().ToBytes()
	}

	messageBytes, err := proto.Marshal(msg)
	require.NoError(t, err)

	if !conf.disableDomainVerification {
		signer := authority
		if c.conf.simulateInvalidMessageSignature {
			signer = testutil.NewRandomAccount(t)
		}
		msg.Signature = &commonpb.Signature{
			Value: ed25519.Sign(signer.PrivateKey().ToBytes(), messageBytes),
		}
	}

	req := &messagingpb.SendMessageRequest{
		Message: &messagingpb.Message{
			Kind: &messagingpb.Message_RequestToReceiveBill{
				RequestToReceiveBill: msg,
			},
		},
		RendezvousKey: &messagingpb.RendezvousKey{
			Value: rendezvousKey.PublicKey().ToBytes(),
		},
	}

	return c.sendMessage(t, req, rendezvousKey)
}

func (c *clientEnv) sendRequestToLoginMessage(t *testing.T, rendezvousKey *common.Account) *sendMessageCallMetadata {
	authority, err := common.NewAccountFromPrivateKeyString("dr2MUzL4NCS45qyp16vDXiSdHqqdg2DF79xKaYMB1vzVtDDjPvyQ8xTH4VsTWXSDP3NFzsdCV6gEoChKftzwLno")
	require.NoError(t, err)

	if c.conf.simulateDoesntOwnDomain {
		authority = testutil.NewRandomAccount(t)
	}

	msg := &messagingpb.RequestToLogin{
		Verifier: authority.ToProto(),
		Domain: &commonpb.Domain{
			Value: "app.getcode.com",
		},
		RendezvousKey: &messagingpb.RendezvousKey{
			Value: rendezvousKey.PublicKey().ToBytes(),
		},
	}

	if c.conf.simulateInvalidDomain {
		msg.Domain.Value = "localhost"
	}

	if c.conf.simulateInvalidRendezvousKey {
		msg.RendezvousKey.Value = testutil.NewRandomAccount(t).PublicKey().ToBytes()
	}

	messageBytes, err := proto.Marshal(msg)
	require.NoError(t, err)

	signer := authority
	if c.conf.simulateInvalidMessageSignature {
		signer = testutil.NewRandomAccount(t)
	}
	msg.Signature = &commonpb.Signature{
		Value: ed25519.Sign(signer.PrivateKey().ToBytes(), messageBytes),
	}

	req := &messagingpb.SendMessageRequest{
		Message: &messagingpb.Message{
			Kind: &messagingpb.Message_RequestToLogin{
				RequestToLogin: msg,
			},
		},
		RendezvousKey: &messagingpb.RendezvousKey{
			Value: rendezvousKey.PublicKey().ToBytes(),
		},
	}
	return c.sendMessage(t, req, rendezvousKey)
}

func (c *clientEnv) sendMessage(t *testing.T, req *messagingpb.SendMessageRequest, rendezvousKey *common.Account) *sendMessageCallMetadata {
	messageBytes, err := proto.Marshal(req.Message)
	require.NoError(t, err)

	signer := rendezvousKey
	if c.conf.simulateInvalidRequestSignature {
		signer = testutil.NewRandomAccount(t)
	}

	req.Signature = &commonpb.Signature{
		Value: ed25519.Sign(signer.PrivateKey().ToBytes(), messageBytes),
	}

	resp, err := c.client.SendMessage(c.ctx, req)
	return &sendMessageCallMetadata{
		req:  req,
		resp: resp,
		err:  err,
	}
}

func (c *clientEnv) ackMessages(t *testing.T, rendezvousKey *common.Account, ids ...*messagingpb.MessageId) {
	resp, err := c.client.AckMessages(c.ctx, &messagingpb.AckMessagesRequest{
		RendezvousKey: &messagingpb.RendezvousKey{
			Value: rendezvousKey.PublicKey().ToBytes(),
		},
		MessageIds: ids,
	})
	require.NoError(t, err)
	assert.Equal(t, messagingpb.AckMesssagesResponse_OK, resp.Result)
}

func (c *clientEnv) resetConf() {
	c.conf = &clientConf{}
}

func mockDomainVerifier(ctx context.Context, owner *common.Account, domain string) (bool, error) {
	// Private key: dr2MUzL4NCS45qyp16vDXiSdHqqdg2DF79xKaYMB1vzVtDDjPvyQ8xTH4VsTWXSDP3NFzsdCV6gEoChKftzwLno
	return owner.PublicKey().ToBase58() == "AiXmGd1DkRbVyfiLLNxC6EFF9ZidCdGpyVY9QFH966Bm", nil
}