github.com/defanghe/fabric@v2.1.1+incompatible/internal/peer/gossip/sa_test.go (about)

     1  /*
     2  Copyright IBM Corp. 2017 All Rights Reserved.
     3  
     4  SPDX-License-Identifier: Apache-2.0
     5  
     6  */
     7  
     8  package gossip
     9  
    10  import (
    11  	"testing"
    12  
    13  	"github.com/hyperledger/fabric/internal/peer/gossip/mocks"
    14  	"github.com/hyperledger/fabric/msp"
    15  	"github.com/stretchr/testify/assert"
    16  	"github.com/stretchr/testify/mock"
    17  )
    18  
    19  func TestMspSecurityAdvisor_OrgByPeerIdentity(t *testing.T) {
    20  	dm := &mocks.DeserializersManager{
    21  		LocalDeserializer: &mocks.IdentityDeserializer{Identity: []byte("Alice"), Msg: []byte("msg1"), Mock: mock.Mock{}},
    22  		ChannelDeserializers: map[string]msp.IdentityDeserializer{
    23  			"A": &mocks.IdentityDeserializer{Identity: []byte("Bob"), Msg: []byte("msg2"), Mock: mock.Mock{}},
    24  		},
    25  	}
    26  
    27  	advisor := NewSecurityAdvisor(dm)
    28  	assert.NotNil(t, advisor.OrgByPeerIdentity([]byte("Alice")))
    29  	assert.NotNil(t, advisor.OrgByPeerIdentity([]byte("Bob")))
    30  	assert.Nil(t, advisor.OrgByPeerIdentity([]byte("Charlie")))
    31  	assert.Nil(t, advisor.OrgByPeerIdentity(nil))
    32  }