simpleSecretsStorage_test.go

package httpsignatures

import (
	"reflect"
	"testing"
)

const testSecretErrType = "*httpsignatures.ErrSecret"

func TestNewSimpleSecretsStorage(t *testing.T) {
	storageExample := map[string]Secret{
		"k1": {
			KeyID:      "k1",
			PublicKey:  "PublicKey1",
			PrivateKey: "PrivateKey1",
			Algorithm:  "md5",
		},
		"k2": {
			KeyID:      "k2",
			PublicKey:  "PublicKey2",
			PrivateKey: "PrivateKey2",
			Algorithm:  "sha1",
		},
	}
	type args struct {
		storage map[string]Secret
	}
	tests := []struct {
		name string
		args args
		want Secrets
	}{
		{
			name: "Valid NewSimpleSecretsStorage",
			args: args{
				storage: storageExample,
			},
			want: (func() *SimpleSecretsStorage {
				s := new(SimpleSecretsStorage)
				s.storage = storageExample
				return s
			})(),
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := NewSimpleSecretsStorage(tt.args.storage)
			if !reflect.DeepEqual(got, tt.want) {
				t.Errorf(tt.name+"\ngot  = %v,\nwant = %v", got, tt.want)
			}
		})
	}
}

func TestSimpleSecretsStorageGet(t *testing.T) {
	storageExample := map[string]Secret{
		"k1": {
			KeyID:      "k1",
			PublicKey:  "PublicKey1",
			PrivateKey: "PrivateKey1",
			Algorithm:  "md5",
		},
	}
	type args struct {
		keyID string
	}
	tests := []struct {
		name        string
		args        args
		want        Secret
		wantErrType string
		wantErrMsg  string
	}{
		{
			name: "Valid SimpleSecretsStorage Get",
			args: args{
				keyID: "k1",
			},
			want:        storageExample["k1"],
			wantErrType: testSecretErrType,
			wantErrMsg:  "",
		},
		{
			name: "Key Not Found",
			args: args{
				keyID: "k2",
			},
			want:        Secret{},
			wantErrType: testSecretErrType,
			wantErrMsg:  "ErrSecret: secret not found",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			s := NewSimpleSecretsStorage(storageExample)
			got, err := s.Get(tt.args.keyID)
			assert(t, got, err, tt.wantErrType, tt.name, tt.want, tt.wantErrMsg)
		})
	}
}