testutils

var SetupCMM = func(keyIDs []string, opts ...func(options *config.LoadOptions) error) model.CryptoMaterialsManager {
	var kp model.MasterKeyProvider
	if len(keyIDs) > 0 {
		keyProvider, err := kmsprovider.NewWithOpts(
			keyIDs,
			kmsprovider.WithAwsLoadOptions(opts...),
		)
		if err != nil {
			log.Error().Err(err).Msg("setupCMM")
			return nil
		}
		kp = keyProvider
	} else {
		keyProvider, err := kmsprovider.NewWithOpts(
			nil,
			kmsprovider.WithAwsLoadOptions(opts...),
			kmsprovider.WithDiscovery(),
		)
		if err != nil {
			log.Error().Err(err).Msg("setupCMM")
			return nil
		}
		kp = keyProvider
	}

	cmm, _ := materials.NewDefault(kp)

	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)
}