testutils

var SetupCMM = func(keyIDs []string, opts ...func(options *config.LoadOptions) error) materials.CryptoMaterialsManager {
	keyProvider, err := providers.NewKmsKeyProviderWithOpts(
		keyIDs,
		providers.WithAwsLoadOptions(opts...),
		providers.WithDiscovery(true),
	)
	if err != nil {
		log.Error().Err(err).Msg("setupCMM")
		return nil

	}

	cmm, _ := materials.NewDefault(keyProvider)

	return cmm
}

var SetupClient = func(maxEdk int) *client.Client {
	cfg, err := clientconfig.NewConfigWithOpts(
		clientconfig.WithCommitmentPolicy(suite.CommitmentPolicyRequireEncryptRequireDecrypt),
		clientconfig.WithMaxEncryptedDataKeys(maxEdk),
	)
	if err != nil {
		panic(err)
	}

	c := client.NewClientWithConfig(cfg)

	return c
}

var SetupDecryptCmd = func(keyIDs []string, ec map[string]string, frame int, edk int, alg string) *CliCmd {
	return NewDecryptCmd(keyIDs, ec, frame, edk)
}

var SetupEncryptCmd = func(keyIDs []string, ec map[string]string, frame int, edk int, alg string) *CliCmd {
	return NewEncryptCmd(keyIDs, ec, frame, edk, alg)
}