code.vegaprotocol.io/vega@v0.79.0/core/integration/steps/the_following_events_should_be_emitted.go (about)

     1  // Copyright (C) 2023 Gobalsky Labs Limited
     2  //
     3  // This program is free software: you can redistribute it and/or modify
     4  // it under the terms of the GNU Affero General Public License as
     5  // published by the Free Software Foundation, either version 3 of the
     6  // License, or (at your option) any later version.
     7  //
     8  // This program is distributed in the hope that it will be useful,
     9  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    10  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    11  // GNU Affero General Public License for more details.
    12  //
    13  // You should have received a copy of the GNU Affero General Public License
    14  // along with this program.  If not, see <http://www.gnu.org/licenses/>.
    15  
    16  package steps
    17  
    18  import (
    19  	"fmt"
    20  
    21  	"code.vegaprotocol.io/vega/core/events"
    22  	"code.vegaprotocol.io/vega/core/integration/stubs"
    23  
    24  	"github.com/cucumber/godog"
    25  )
    26  
    27  func TheFollowingEventsShouldBeEmitted(broker *stubs.BrokerStub, table *godog.Table) error {
    28  	for _, row := range parseEmittedEventsTable(table) {
    29  		eventType := row.MustEventType("type")
    30  
    31  		if len(broker.GetBatch(eventType)) == 0 {
    32  			return errEventNotEmitted(eventType)
    33  		}
    34  	}
    35  
    36  	return nil
    37  }
    38  
    39  func TheFollowingEventsShouldNotBeEmitted(broker *stubs.BrokerStub, table *godog.Table) error {
    40  	for _, row := range parseEmittedEventsTable(table) {
    41  		eventType := row.MustEventType("type")
    42  
    43  		if len(broker.GetBatch(eventType)) > 0 {
    44  			return errEventEmitted(eventType)
    45  		}
    46  	}
    47  
    48  	return nil
    49  }
    50  
    51  func parseEmittedEventsTable(table *godog.Table) []RowWrapper {
    52  	return StrictParseTable(table, []string{
    53  		"type",
    54  	}, []string{})
    55  }
    56  
    57  func errEventNotEmitted(t events.Type) error {
    58  	return fmt.Errorf("event of type %s has not been emitted", t)
    59  }
    60  
    61  func errEventEmitted(t events.Type) error {
    62  	return fmt.Errorf("event of type %s has been emitted", t)
    63  }