code.vegaprotocol.io/vega@v0.79.0/commands/liquidity_provision_submission_test.go

// Copyright (C) 2023 Gobalsky Labs Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero 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 Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

package commands_test

import (
	"fmt"
	"testing"

	"code.vegaprotocol.io/vega/commands"
	commandspb "code.vegaprotocol.io/vega/protos/vega/commands/v1"

	"github.com/stretchr/testify/assert"
)

func TestNilLiquidityProvisionSubmissionFails(t *testing.T) {
	err := commands.CheckLiquidityProvisionSubmission(nil)

	assert.Error(t, err)
}

func TestLiquidityProvisionSubmission(t *testing.T) {
	cases := []struct {
		lp        commandspb.LiquidityProvisionSubmission
		errString string
	}{
		{
			// this is a valid cancellation.
			lp: commandspb.LiquidityProvisionSubmission{
				CommitmentAmount: "0",
				MarketId:         "08dce6ebf50e34fedee32860b6f459824e4b834762ea66a96504fdc57a9c4741",
			},
			errString: "liquidity_provision_submission.commitment_amount (is not a valid number)",
		},
		{
			lp: commandspb.LiquidityProvisionSubmission{
				CommitmentAmount: "100",
				Fee:              "abcd",
				MarketId:         "08dce6ebf50e34fedee32860b6f459824e4b834762ea66a96504fdc57a9c4741",
			},
			errString: "liquidity_provision_submission.fee (is not a valid value)",
		},
		{
			lp: commandspb.LiquidityProvisionSubmission{
				CommitmentAmount: "100",
				Fee:              "-1",
				MarketId:         "08dce6ebf50e34fedee32860b6f459824e4b834762ea66a96504fdc57a9c4741",
			},
			errString: "liquidity_provision_submission.fee (must be positive)",
		},
		{
			lp: commandspb.LiquidityProvisionSubmission{
				CommitmentAmount: "100",
				Fee:              "0.1",
			},
			errString: "liquidity_provision_submission.market_id (is required)",
		},

		{
			lp: commandspb.LiquidityProvisionSubmission{
				CommitmentAmount: "100",
				Fee:              "0.1",
				MarketId:         "08dce6ebf50e34fedee32860b6f459824e4b834762ea66a96504fdc57a9c4741",
			},
		},
	}

	for i, c := range cases {
		err := commands.CheckLiquidityProvisionSubmission(&c.lp)
		if len(c.errString) <= 0 {
			assert.NoErrorf(t, err, "unexpected error on position: %d", i)
			continue
		}

		assert.Errorf(t, err, "expected error on position: %d", i)
		assert.EqualErrorf(t, err, c.errString, "expected error to match on position: %d", i)
	}
}

func TestCheckLiquidityProvisionCancellation(t *testing.T) {
	type args struct {
		cmd *commandspb.LiquidityProvisionCancellation
	}
	tests := []struct {
		name      string
		args      args
		wantErr   assert.ErrorAssertionFunc
		errString string
	}{
		{
			name: "Should return an error if request is nil",
			args: args{
				cmd: nil,
			},
			wantErr:   assert.Error,
			errString: "liquidity_provision_cancellation (is required)",
		},
		{
			name: "Should return an error if market_id is not provided",
			args: args{
				cmd: &commandspb.LiquidityProvisionCancellation{
					MarketId: "",
				},
			},
			wantErr:   assert.Error,
			errString: "liquidity_provision_cancellation.market_id (is required)",
		},
		{
			name: "Should succeed if market id is provided",
			args: args{
				cmd: &commandspb.LiquidityProvisionCancellation{
					MarketId: "08dce6ebf50e34fedee32860b6f459824e4b834762ea66a96504fdc57a9c4741",
				},
			},
			wantErr: assert.NoError,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			gotErr := commands.CheckLiquidityProvisionCancellation(tt.args.cmd)
			tt.wantErr(t, gotErr, fmt.Sprintf("CheckLiquidityProvisionCancellation(%v)", tt.args.cmd))
			if tt.errString != "" {
				assert.EqualError(t, gotErr, tt.errString)
			}
		})
	}
}

func TestCheckLiquidityProvisionAmendment(t *testing.T) {
	type args struct {
		cmd *commandspb.LiquidityProvisionAmendment
	}
	tests := []struct {
		name      string
		args      args
		wantErr   assert.ErrorAssertionFunc
		errString string
	}{
		{
			name: "Should return an error when the command is nil",
			args: args{
				cmd: nil,
			},
			wantErr:   assert.Error,
			errString: "liquidity_provision_amendment (is required)",
		},
		{
			name: "Should return an error when market_id is not provided",
			args: args{
				cmd: &commandspb.LiquidityProvisionAmendment{
					MarketId: "",
				},
			},
			wantErr:   assert.Error,
			errString: "liquidity_provision_amendment.market_id (is required)",
		},
		{
			name: "Should return an error if amendment changes nothing",
			args: args{
				cmd: &commandspb.LiquidityProvisionAmendment{
					MarketId: "abcd",
				},
			},
			wantErr:   assert.Error,
			errString: "liquidity_provision_amendment (is required)",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			gotErr := commands.CheckLiquidityProvisionAmendment(tt.args.cmd)
			tt.wantErr(t, gotErr, fmt.Sprintf("CheckLiquidityProvisionAmendment(%v)", tt.args.cmd))

			if tt.errString != "" {
				assert.EqualError(t, gotErr, tt.errString)
			}
		})
	}
}