parties_should_have_the_following_position_status.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 steps

import (
	"fmt"

	"code.vegaprotocol.io/vega/core/integration/stubs"
	proto "code.vegaprotocol.io/vega/protos/vega"

	"github.com/cucumber/godog"
)

func PartiesShouldHaveTheFollowingPositionStatus(
	broker *stubs.BrokerStub,
	market string,
	table *godog.Table,
) error {
	return positionEvents(broker, table, market, false)
}

func PartiesShouldHaveTheFollowingPositionStatusAgg(
	broker *stubs.BrokerStub,
	market string,
	table *godog.Table,
) error {
	return positionEvents(broker, table, market, true)
}

func positionEvents(broker *stubs.BrokerStub, table *godog.Table, market string, aggregate bool) error {
	orders := broker.GetDistressedOrders()
	closed := broker.GetSettleDistressed()
	// get a status map by party
	partyStatus := map[string]proto.PositionStatus{}
	for _, o := range orders {
		if o.MarketID() != market {
			continue
		}
		for _, p := range o.Parties() {
			partyStatus[p] = proto.PositionStatus_POSITION_STATUS_ORDERS_CLOSED
		}
	}
	for _, c := range closed {
		if c.MarketID() != market {
			continue
		}
		partyStatus[c.PartyID()] = proto.PositionStatus_POSITION_STATUS_CLOSED_OUT
	}
	lossSoc := broker.GetLossSocializationEvents()
	partyLSA := map[string]int64{}
	for _, le := range lossSoc {
		if le.MarketID() != market {
			continue
		}
		amt := le.AmountLost()
		party := le.PartyID()
		// aggregate if required
		if aggregate {
			amt += partyLSA[party]
		}
		partyLSA[party] = amt
	}
	for _, r := range parsePositionStatusRow(table) {
		row := positionStatusRow{row: r}
		party := row.Party()
		if row.HasStatus() {
			exp := row.Status()
			// status is not unspecified (ie we expect to see an event with this information
			if exp == proto.PositionStatus_POSITION_STATUS_UNSPECIFIED {
				continue
			}
			status, ok := partyStatus[party]
			if !ok {
				return fmt.Errorf("no status update found for party %s", party)
			}
			if status != row.Status() {
				return fmt.Errorf("expected status %s for party %s, instead got %s", row.Status(), party, status)
			}
		}
		if row.HasAmount() {
			exp := row.Amount()
			got, ok := partyLSA[party]
			if !ok && exp != 0 {
				return fmt.Errorf("expected a loss socialisation amount of %d, none found", exp)
			}
			if got != exp {
				return fmt.Errorf("expected a loss socialisation amount of %d, instead saw %d", exp, got)
			}
		}
	}
	return nil
}

func parsePositionStatusRow(table *godog.Table) []RowWrapper {
	return StrictParseTable(table, []string{
		"party",
	}, []string{
		"status",
		"amount",
	})
}

type positionStatusRow struct {
	row RowWrapper
}

func (p positionStatusRow) HasStatus() bool {
	return p.row.HasColumn("status")
}

func (p positionStatusRow) HasAmount() bool {
	return p.row.HasColumn("amount")
}

func (p positionStatusRow) Party() string {
	return p.row.MustStr("party")
}

func (p positionStatusRow) Status() proto.PositionStatus {
	return p.row.MustPositionStatus("status")
}

func (p positionStatusRow) Amount() int64 {
	return p.row.MustI64("amount")
}