configcli_steps.go

/*
Copyright SecureKey Technologies Inc. All Rights Reserved.

SPDX-License-Identifier: Apache-2.0
*/

package bddtests

import (
	"fmt"

	"strings"

	"github.com/DATA-DOG/godog"
)

var queryResult string

// ConfigCLISteps config cli BDD test steps
type ConfigCLISteps struct {
	BDDContext *BDDContext
}

// NewConfigCLISteps new config cli steps
func NewConfigCLISteps(context *BDDContext) *ConfigCLISteps {
	return &ConfigCLISteps{BDDContext: context}
}

// UpdateConfig update config using config cli
func (c *ConfigCLISteps) UpdateConfig(configFile, mspID, orgID, channelID string) error {
	_, err := c.BDDContext.configCLI.ExecUpdate(channelID, mspID, orgID, configFile, "")
	if err != nil {
		return fmt.Errorf("failed to update config: %v", err)
	}
	return nil
}

// UpdateConfigWithExceptions updates config using config cli
func (c *ConfigCLISteps) UpdateConfigWithExceptions(configFile, mspID, orgID, channelID, blackListRegex string) error {
	var targets []string
	if blackListRegex != "" {
		logger.Infof("Config won't be updated on the following peers: [%s]", blackListRegex)
		var err error
		targets, err = c.getLocalTargets(orgID, blackListRegex)
		if err != nil {
			return err
		}
	}

	peers := ""
	if len(targets) > 0 {
		for i, target := range targets {
			peers += target
			if i+1 < len(targets) {
				peers += ","
			}
		}
		logger.Infof("Updating config on peers: [%s]", peers)
	}

	_, err := c.BDDContext.configCLI.ExecUpdate(channelID, mspID, orgID, configFile, peers)
	if err != nil {
		return fmt.Errorf("failed to update config: %v", err)
	}
	return nil
}

func (c *ConfigCLISteps) getLocalTargets(orgID string, blackListRegex string) ([]string, error) {
	return getLocalTargets(c.BDDContext, orgID, blackListRegex)
}

// exec executes action on config using config cli
func (c *ConfigCLISteps) exec(action, peerID, mspID, appName, appVersion, compName, compVersion, channelID string) error {
	var err error
	queryResult, err = c.BDDContext.configCLI.Exec(action, channelID, mspID, peerID, appName, appVersion, compName, compVersion)
	if err != nil {
		return fmt.Errorf("failed to %s config: %v", action, err)
	}
	return nil
}

func (c *ConfigCLISteps) containsInQueryResult(value string) error {
	if queryResult == "" {
		return fmt.Errorf("queryResult is empty")
	}
	logger.Infof("Query value %s and tested value %s", queryResult, value)
	if !strings.Contains(queryResult, value) {
		return fmt.Errorf("query value(%s) doesn't contain expected value(%s)", queryResult, value)
	}
	return nil
}

func (c *ConfigCLISteps) notContainsInQueryResult(value string) error {
	logger.Infof("Query value %s and tested value %s", queryResult, value)
	if strings.Contains(queryResult, value) {
		return fmt.Errorf("query value(%s) shoud not contain expected value(%s)", queryResult, value)
	}
	return nil
}

// RegisterSteps register steps
func (c *ConfigCLISteps) RegisterSteps(s *godog.Suite) {
	s.BeforeScenario(c.BDDContext.BeforeScenario)
	s.AfterScenario(c.BDDContext.AfterScenario)
	s.Step(`^client update config "([^"]*)" with mspid "([^"]*)" with orgid "([^"]*)" on the "([^"]*)" channel$`, c.UpdateConfig)
	s.Step(`^client update config "([^"]*)" with mspid "([^"]*)" with orgid "([^"]*)" on the "([^"]*)" channel except on "([^"]*)"$`, c.UpdateConfigWithExceptions)
	s.Step(`^client "([^"]*)" config by peer id "([^"]*)" with mspid "([^"]*)" with app name "([^"]*)" with app version "([^"]*)" with comp name "([^"]*)" with comp version "([^"]*)" on the "([^"]*)" channel$`, c.exec)
	s.Step(`^response from cli query to client contains value "([^"]*)"$`, c.containsInQueryResult)
	s.Step(`^response from cli query to client not contains value "([^"]*)"$`, c.notContainsInQueryResult)

}