github.com/tenywen/fabric@v1.0.0-beta.0.20170620030522-a5b1ed380643/peer/common/common_test.go

/*
Copyright IBM Corp. All Rights Reserved.

SPDX-License-Identifier: Apache-2.0
*/

package common_test

import (
	"fmt"
	"os"
	"testing"

	"github.com/hyperledger/fabric/common/util"
	"github.com/hyperledger/fabric/core/config"
	"github.com/hyperledger/fabric/msp"
	"github.com/hyperledger/fabric/peer/common"
	pb "github.com/hyperledger/fabric/protos/peer"
	"github.com/stretchr/testify/assert"
)

func TestInitConfig(t *testing.T) {
	type args struct {
		cmdRoot string
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{
			name:    "Empty command root",
			args:    args{cmdRoot: ""},
			wantErr: true,
		},
		{
			name:    "Bad command root",
			args:    args{cmdRoot: "cre"},
			wantErr: true,
		},
		{
			name:    "Good command root",
			args:    args{cmdRoot: "core"},
			wantErr: false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := common.InitConfig(tt.args.cmdRoot); (err != nil) != tt.wantErr {
				t.Errorf("InitConfig() error = %v, wantErr %v", err, tt.wantErr)
			}
		})
	}
}

func TestINitCryptoMissingDir(t *testing.T) {
	dir := os.TempDir() + "/" + util.GenerateUUID()
	err := common.InitCrypto(dir, "DEFAULT")
	assert.Error(t, err, "Should be able to initialize crypto with non-existing directory")
	assert.Contains(t, err.Error(), fmt.Sprintf("missing %s folder", dir))
}

func TestInitCrypto(t *testing.T) {

	mspConfigPath, err := config.GetDevMspDir()
	localMspId := "DEFAULT"
	err = common.InitCrypto(mspConfigPath, localMspId)
	assert.NoError(t, err, "Unexpected error [%s] calling InitCrypto()", err)
	err = common.InitCrypto("/etc/foobaz", localMspId)
	assert.Error(t, err, "Expected error [%s] calling InitCrypto()", err)
	localMspId = ""
	err = common.InitCrypto(mspConfigPath, localMspId)
	assert.Error(t, err, "Expected error [%s] calling InitCrypto()", err)
}

func TestGetEndorserClient(t *testing.T) {
	tests := []struct {
		name    string
		want    pb.EndorserClient
		wantErr bool
	}{
		{
			name:    "Should not return EndorserClient, there is no peer running",
			want:    nil,
			wantErr: true,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			_, err := common.GetEndorserClient()
			if (err != nil) != tt.wantErr {
				t.Errorf("GetEndorserClient() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
		})
	}
}

func TestSetLogLevelFromViper(t *testing.T) {
	type args struct {
		module string
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{
			name:    "Empty module name",
			args:    args{module: ""},
			wantErr: true,
		},
		{
			name:    "Invalid module name",
			args:    args{module: "policy"},
			wantErr: true,
		},
		{
			name:    "Valid module name",
			args:    args{module: "cauthdsl"},
			wantErr: false,
		},
		{
			name:    "Valid module name",
			args:    args{module: "peer"},
			wantErr: false,
		},
		{
			name:    "Valid module name",
			args:    args{module: "gossip"},
			wantErr: false,
		},
		{
			name:    "Valid module name",
			args:    args{module: "grpc"},
			wantErr: false,
		},
		{
			name:    "Valid module name",
			args:    args{module: "msp"},
			wantErr: false,
		},
		{
			name:    "Valid module name",
			args:    args{module: "ledger"},
			wantErr: false,
		},
		{
			name:    "Valid module name",
			args:    args{module: "policies"},
			wantErr: false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := common.SetLogLevelFromViper(tt.args.module); (err != nil) != tt.wantErr {
				t.Errorf("SetLogLevelFromViper() args = %v error = %v, wantErr %v", tt.args, err, tt.wantErr)
			}
		})
	}
}

func TestCheckLogLevel(t *testing.T) {
	type args struct {
		level string
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{
			name:    "Empty module name",
			args:    args{level: ""},
			wantErr: true,
		},
		{
			name:    "Valie module name",
			args:    args{level: "warning"},
			wantErr: false,
		},
		{
			name:    "Valie module name",
			args:    args{level: "foobaz"},
			wantErr: true,
		},
		{
			name:    "Valie module name",
			args:    args{level: "error"},
			wantErr: false,
		},
		{
			name:    "Valie module name",
			args:    args{level: "info"},
			wantErr: false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := common.CheckLogLevel(tt.args.level); (err != nil) != tt.wantErr {
				t.Errorf("CheckLogLevel() args = %v error = %v, wantErr %v", tt.args, err, tt.wantErr)
			}
		})
	}
}

func TestGetDefaultSigner(t *testing.T) {
	tests := []struct {
		name    string
		want    msp.SigningIdentity
		wantErr bool
	}{
		{
			name:    "Should return DefaultSigningIdentity",
			want:    nil,
			wantErr: false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			_, err := common.GetDefaultSigner()
			if (err != nil) != tt.wantErr {
				t.Errorf("GetDefaultSigner() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
		})
	}
}