epistemophiliac/fabric-samples
1
0
Fork 0
mirror of https://github.com/hyperledger/fabric-samples.git synced 2026-06-23 01:55:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 12
fabric-samples/asset-transfer-private-data/chaincode-go/chaincode/asset_transfer_test.go

468 lines
17 KiB
Go
Raw Blame History

/*
Copyright IBM Corp. All Rights Reserved.
SPDX-License-Identifier: Apache-2.0
*/
package chaincode_test
import (
"encoding/json"
"testing"
"github.com/hyperledger/fabric-chaincode-go/pkg/cid"
"github.com/hyperledger/fabric-chaincode-go/shim"
"github.com/hyperledger/fabric-contract-api-go/contractapi"
"github.com/hyperledger/fabric-samples/asset-transfer-private-data/chaincode-go/chaincode"
"github.com/hyperledger/fabric-samples/asset-transfer-private-data/chaincode-go/chaincode/mocks"
"github.com/stretchr/testify/require"
)
/*
These unit tests use mocks to simulate chaincode-api & fabric interactions
The mocks are generated using counterfeiter directives in the comments (starting with "go:generate counterfeiter")
All files in mocks/* are generated by running following, in the directory with your directive:
`go generate`
*/
//go:generate counterfeiter -o mocks/transaction.go -fake-name TransactionContext . transactionContext
type transactionContext interface {
contractapi.TransactionContextInterface
}
//go:generate counterfeiter -o mocks/chaincodestub.go -fake-name ChaincodeStub . chaincodeStub
type chaincodeStub interface {
shim.ChaincodeStubInterface
}
//go:generate counterfeiter -o mocks/statequeryiterator.go -fake-name StateQueryIterator . stateQueryIterator
type stateQueryIterator interface {
shim.StateQueryIteratorInterface
}
//go:generate counterfeiter -o mocks/clientIdentity.go -fake-name ClientIdentity . clientIdentity
type clientIdentity interface {
cid.ClientIdentity
}
// const assetCollectionName = "assetCollection"
// const transferAgreementObjectType = "transferAgreement"
// const myOrg1Msp = "Org1Testmsp"
// const myOrg1Clientid = "myOrg1Userid"
// const myOrg1PrivCollection = "Org1TestmspPrivateCollection"
// const myOrg2Msp = "Org2Testmsp"
// const myOrg2Clientid = "myOrg2Userid"
// const myOrg2PrivCollection = "Org2TestmspPrivateCollection"
// type assetTransientInput struct {
// Type string `json:"objectType"`
// ID string `json:"assetID"`
// Color string `json:"color"`
// Size int `json:"size"`
// AppraisedValue int `json:"appraisedValue"`
// }
// type assetTransferTransientInput struct {
// ID string `json:"assetID"`
// BuyerMSP string `json:"buyerMSP"`
// }
// func TestCreateAssetBadInput(t *testing.T) {
// transactionContext, chaincodeStub := prepMocksAsOrg1()
// assetTransferCC := chaincode.SmartContract{}
// // No transient map
// err := assetTransferCC.CreateAsset(transactionContext)
// require.EqualError(t, err, "asset not found in the transient map input")
// // transient map with incomplete asset data
// assetPropMap := map[string][]byte{
// "asset_properties": []byte("ill formatted property"),
// }
// chaincodeStub.GetTransientReturns(assetPropMap, nil)
// err = assetTransferCC.CreateAsset(transactionContext)
// require.Error(t, err, "Expected error: transient map with incomplete asset data")
// require.Contains(t, err.Error(), "failed to unmarshal JSON")
// testAsset := &assetTransientInput{
// Type: "testfulasset",
// }
// setReturnAssetPropsInTransientMap(t, chaincodeStub, testAsset)
// err = assetTransferCC.CreateAsset(transactionContext)
// require.EqualError(t, err, "assetID field must be a non-empty string")
// testAsset = &assetTransientInput{
// ID: "id1",
// Color: "gray",
// }
// setReturnAssetPropsInTransientMap(t, chaincodeStub, testAsset)
// err = assetTransferCC.CreateAsset(transactionContext)
// require.EqualError(t, err, "objectType field must be a non-empty string")
// // case when asset exists, GetPrivateData returns a valid data from ledger
// testAsset = &assetTransientInput{
// ID: "id1",
// Type: "testfulasset",
// Color: "gray",
// Size: 7,
// AppraisedValue: 500,
// }
// setReturnAssetPropsInTransientMap(t, chaincodeStub, testAsset)
// chaincodeStub.GetPrivateDataReturns([]byte{}, nil)
// err = assetTransferCC.CreateAsset(transactionContext)
// require.EqualError(t, err, "this asset already exists: id1")
// }
// func TestCreateAssetSuccessful(t *testing.T) {
// transactionContext, chaincodeStub := prepMocksAsOrg1()
// assetTransferCC := chaincode.SmartContract{}
// testAsset := &assetTransientInput{
// ID: "id1",
// Type: "testfulasset",
// Color: "gray",
// Size: 7,
// AppraisedValue: 500,
// }
// setReturnAssetPropsInTransientMap(t, chaincodeStub, testAsset)
// err := assetTransferCC.CreateAsset(transactionContext)
// require.NoError(t, err)
// //Validate PutPrivateData calls
// calledCollection, calledId, _ := chaincodeStub.PutPrivateDataArgsForCall(0)
// require.Equal(t, assetCollectionName, calledCollection)
// require.Equal(t, "id1", calledId)
// expectedPrivateDetails := &chaincode.AssetPrivateDetails{
// ID: "id1",
// AppraisedValue: 500,
// }
// assetBytes, err := json.Marshal(expectedPrivateDetails)
// calledCollection, calledId, calledAssetBytes := chaincodeStub.PutPrivateDataArgsForCall(1)
// require.Equal(t, myOrg1PrivCollection, calledCollection)
// require.Equal(t, "id1", calledId)
// require.Equal(t, assetBytes, calledAssetBytes)
// }
// func TestAgreeToTransferBadInput(t *testing.T) {
// transactionContext, chaincodeStub := prepMocksAsOrg1()
// assetTransferCC := chaincode.SmartContract{}
// assetPrivDetail := &chaincode.AssetPrivateDetails{
// ID: "id1",
// //no AppraisedValue
// }
// setReturnAssetPrivateDetailsInTransientMap(t, chaincodeStub, assetPrivDetail)
// origAsset := chaincode.Asset{
// ID: "id1",
// Type: "testfulasset",
// Color: "gray",
// Size: 7,
// Owner: myOrg1Clientid,
// }
// setReturnPrivateDataInStub(t, chaincodeStub, &origAsset)
// err := assetTransferCC.AgreeToTransfer(transactionContext)
// require.EqualError(t, err, "appraisedValue field must be a positive integer")
// assetPrivDetail = &chaincode.AssetPrivateDetails{
// //no ID
// AppraisedValue: 500,
// }
// setReturnAssetPrivateDetailsInTransientMap(t, chaincodeStub, assetPrivDetail)
// err = assetTransferCC.AgreeToTransfer(transactionContext)
// require.EqualError(t, err, "assetID field must be a non-empty string")
// assetPrivDetail = &chaincode.AssetPrivateDetails{
// ID: "id1",
// AppraisedValue: 500,
// }
// setReturnAssetPrivateDetailsInTransientMap(t, chaincodeStub, assetPrivDetail)
// //asset does not exist
// setReturnPrivateDataInStub(t, chaincodeStub, nil)
// err = assetTransferCC.AgreeToTransfer(transactionContext)
// require.EqualError(t, err, "id1 does not exist")
// }
// func TestAgreeToTransferSuccessful(t *testing.T) {
// transactionContext, chaincodeStub := prepMocksAsOrg1()
// assetTransferCC := chaincode.SmartContract{}
// assetPrivDetail := &chaincode.AssetPrivateDetails{
// ID: "id1",
// AppraisedValue: 500,
// }
// setReturnAssetPrivateDetailsInTransientMap(t, chaincodeStub, assetPrivDetail)
// origAsset := chaincode.Asset{
// ID: "id1",
// Type: "testfulasset",
// Color: "gray",
// Size: 7,
// Owner: myOrg1Clientid,
// }
// setReturnPrivateDataInStub(t, chaincodeStub, &origAsset)
// chaincodeStub.CreateCompositeKeyReturns(transferAgreementObjectType+"id1", nil)
// err := assetTransferCC.AgreeToTransfer(transactionContext)
// require.NoError(t, err)
// expectedDataBytes, err := json.Marshal(assetPrivDetail)
// calledCollection, calledId, calledWithDataBytes := chaincodeStub.PutPrivateDataArgsForCall(0)
// require.Equal(t, myOrg1PrivCollection, calledCollection)
// require.Equal(t, "id1", calledId)
// require.Equal(t, expectedDataBytes, calledWithDataBytes)
// calledCollection, calledId, calledWithDataBytes = chaincodeStub.PutPrivateDataArgsForCall(1)
// require.Equal(t, assetCollectionName, calledCollection)
// require.Equal(t, transferAgreementObjectType+"id1", calledId)
// require.Equal(t, []byte(myOrg1Clientid), calledWithDataBytes)
// }
// func TestTransferAssetBadInput(t *testing.T) {
// transactionContext, chaincodeStub := prepMocksAsOrg1()
// assetTransferCC := chaincode.SmartContract{}
// assetNewOwner := &assetTransferTransientInput{
// ID: "id1",
// BuyerMSP: "",
// }
// setReturnAssetOwnerInTransientMap(t, chaincodeStub, assetNewOwner)
// setReturnPrivateDataInStub(t, chaincodeStub, &chaincode.Asset{})
// err := assetTransferCC.TransferAsset(transactionContext)
// require.EqualError(t, err, "buyerMSP field must be a non-empty string")
// assetNewOwner = &assetTransferTransientInput{
// ID: "id1",
// BuyerMSP: myOrg2Msp,
// }
// setReturnAssetOwnerInTransientMap(t, chaincodeStub, assetNewOwner)
// //asset does not exist
// setReturnPrivateDataInStub(t, chaincodeStub, nil)
// err = assetTransferCC.TransferAsset(transactionContext)
// require.EqualError(t, err, "id1 does not exist")
// }
// func TestTransferAssetSuccessful(t *testing.T) {
// transactionContext, chaincodeStub := prepMocksAsOrg1()
// assetTransferCC := chaincode.SmartContract{}
// assetNewOwner := &assetTransferTransientInput{
// ID: "id1",
// BuyerMSP: myOrg2Msp,
// }
// setReturnAssetOwnerInTransientMap(t, chaincodeStub, assetNewOwner)
// origAsset := chaincode.Asset{
// ID: "id1",
// Type: "testfulasset",
// Color: "gray",
// Size: 7,
// Owner: myOrg1Clientid,
// }
// setReturnPrivateDataInStub(t, chaincodeStub, &origAsset)
// //to ensure we pass data hash verification
// chaincodeStub.GetPrivateDataHashReturns([]byte("datahash"), nil)
// //to ensure that ReadTransferAgreement call returns org2 client ID
// chaincodeStub.GetPrivateDataReturnsOnCall(1, []byte(myOrg2Clientid), nil)
// chaincodeStub.CreateCompositeKeyReturns(transferAgreementObjectType+"id1", nil)
// err := assetTransferCC.TransferAsset(transactionContext)
// require.NoError(t, err)
// //Validate PutPrivateData calls
// expectedNewAsset := origAsset
// expectedNewAsset.Owner = myOrg2Clientid
// expectedNewAssetBytes, err := json.Marshal(expectedNewAsset)
// require.NoError(t, err)
// calledCollection, calledId, calledWithAssetBytes := chaincodeStub.PutPrivateDataArgsForCall(0)
// require.Equal(t, assetCollectionName, calledCollection)
// require.Equal(t, "id1", calledId)
// require.Equal(t, expectedNewAssetBytes, calledWithAssetBytes)
// calledCollection, calledId = chaincodeStub.DelPrivateDataArgsForCall(0)
// require.Equal(t, myOrg1PrivCollection, calledCollection)
// require.Equal(t, "id1", calledId)
// calledCollection, calledId = chaincodeStub.DelPrivateDataArgsForCall(1)
// require.Equal(t, assetCollectionName, calledCollection)
// require.Equal(t, transferAgreementObjectType+"id1", calledId)
// }
// func TestTransferAssetByNonOwner(t *testing.T) {
// transactionContext, chaincodeStub := prepMocksAsOrg1()
// assetTransferCC := chaincode.SmartContract{}
// assetNewOwner := &assetTransferTransientInput{
// ID: "id1",
// BuyerMSP: myOrg1Msp,
// }
// setReturnAssetOwnerInTransientMap(t, chaincodeStub, assetNewOwner)
// //Try to transfer asset owned by Org2
// org2Asset := chaincode.Asset{
// ID: "id1",
// Type: "testfulasset",
// Color: "gray",
// Size: 7,
// Owner: myOrg2Clientid,
// }
// setReturnPrivateDataInStub(t, chaincodeStub, &org2Asset)
// err := assetTransferCC.TransferAsset(transactionContext)
// require.EqualError(t, err, "failed transfer verification: error: submitting client identity does not own asset")
// }
// func TestTransferAssetWithoutAnAgreement(t *testing.T) {
// transactionContext, chaincodeStub := prepMocksAsOrg1()
// assetTransferCC := chaincode.SmartContract{}
// assetNewOwner := &assetTransferTransientInput{
// ID: "id1",
// BuyerMSP: myOrg1Msp,
// }
// setReturnAssetOwnerInTransientMap(t, chaincodeStub, assetNewOwner)
// orgAsset := chaincode.Asset{
// ID: "id1",
// Type: "testfulasset",
// Color: "gray",
// Size: 7,
// Owner: myOrg1Clientid,
// }
// setReturnPrivateDataInStub(t, chaincodeStub, &orgAsset)
// //to ensure we pass data hash verification
// chaincodeStub.GetPrivateDataHashReturns([]byte("datahash"), nil)
// chaincodeStub.CreateCompositeKeyReturns(transferAgreementObjectType+"id1", nil)
// //ReadTransferAgreement call returns no buyer client ID
// chaincodeStub.GetPrivateDataReturnsOnCall(1, []byte{}, nil)
// err := assetTransferCC.TransferAsset(transactionContext)
// require.EqualError(t, err, "BuyerID not found in TransferAgreement for id1")
// }
// func TestTransferAssetNonMatchingAppraisalValue(t *testing.T) {
// transactionContext, chaincodeStub := prepMocksAsOrg1()
// assetTransferCC := chaincode.SmartContract{}
// assetNewOwner := &assetTransferTransientInput{
// ID: "id1",
// BuyerMSP: myOrg2Msp,
// }
// setReturnAssetOwnerInTransientMap(t, chaincodeStub, assetNewOwner)
// orgAsset := chaincode.Asset{
// ID: "id1",
// Type: "testfulasset",
// Color: "gray",
// Size: 7,
// Owner: myOrg1Clientid,
// }
// setReturnPrivateDataInStub(t, chaincodeStub, &orgAsset)
// chaincodeStub.CreateCompositeKeyReturns(transferAgreementObjectType+"id1", nil)
// //data hash different in each collection
// chaincodeStub.GetPrivateDataHashReturnsOnCall(0, []byte("datahash1"), nil)
// chaincodeStub.GetPrivateDataHashReturnsOnCall(1, []byte("datahash2"), nil)
// err := assetTransferCC.TransferAsset(transactionContext)
// require.Error(t, err, "Expected failed hash verification")
// require.Contains(t, err.Error(), "failed transfer verification: hash for appraised value")
// }
// func prepMocksAsOrg1() (*mocks.TransactionContext, *mocks.ChaincodeStub) {
// return prepMocks(myOrg1Msp, myOrg1Clientid)
// }
// func prepMocksAsOrg2() (*mocks.TransactionContext, *mocks.ChaincodeStub) {
// return prepMocks(myOrg2Msp, myOrg2Clientid)
// }
// func prepMocks(orgMSP, clientId string) (*mocks.TransactionContext, *mocks.ChaincodeStub) {
// chaincodeStub := &mocks.ChaincodeStub{}
// transactionContext := &mocks.TransactionContext{}
// transactionContext.GetStubReturns(chaincodeStub)
// clientIdentity := &mocks.ClientIdentity{}
// clientIdentity.GetMSPIDReturns(orgMSP, nil)
// clientIdentity.GetIDReturns(clientId, nil)
// //set matching msp ID using peer shim env variable
// os.Setenv("CORE_PEER_LOCALMSPID", orgMSP)
// transactionContext.GetClientIdentityReturns(clientIdentity)
// return transactionContext, chaincodeStub
// }
// func setReturnAssetPrivateDetailsInTransientMap(t *testing.T, chaincodeStub *mocks.ChaincodeStub, assetPrivDetail *chaincode.AssetPrivateDetails) []byte {
// assetOwnerBytes := []byte{}
// if assetPrivDetail != nil {
// var err error
// assetOwnerBytes, err = json.Marshal(assetPrivDetail)
// require.NoError(t, err)
// }
// assetPropMap := map[string][]byte{
// "asset_value": assetOwnerBytes,
// }
// chaincodeStub.GetTransientReturns(assetPropMap, nil)
// return assetOwnerBytes
// }
// func setReturnAssetOwnerInTransientMap(t *testing.T, chaincodeStub *mocks.ChaincodeStub, assetOwner *assetTransferTransientInput) []byte {
// assetOwnerBytes := []byte{}
// if assetOwner != nil {
// var err error
// assetOwnerBytes, err = json.Marshal(assetOwner)
// require.NoError(t, err)
// }
// assetPropMap := map[string][]byte{
// "asset_owner": assetOwnerBytes,
// }
// chaincodeStub.GetTransientReturns(assetPropMap, nil)
// return assetOwnerBytes
// }
// func setReturnAssetPropsInTransientMap(t *testing.T, chaincodeStub *mocks.ChaincodeStub, testAsset *assetTransientInput) []byte {
// assetBytes := []byte{}
// if testAsset != nil {
// var err error
// assetBytes, err = json.Marshal(testAsset)
// require.NoError(t, err)
// }
// assetPropMap := map[string][]byte{
// "asset_properties": assetBytes,
// }
// chaincodeStub.GetTransientReturns(assetPropMap, nil)
// return assetBytes
// }
// func setReturnPrivateDataInStub(t *testing.T, chaincodeStub *mocks.ChaincodeStub, testAsset *chaincode.Asset) []byte {
// if testAsset == nil {
// chaincodeStub.GetPrivateDataReturns(nil, nil)
// return nil
// } else {
// var err error
// assetBytes, err := json.Marshal(testAsset)
// require.NoError(t, err)
// chaincodeStub.GetPrivateDataReturns(assetBytes, nil)
// return assetBytes
// }
// }
// func setReturnAssetPrivateDetailsInStub(t *testing.T, chaincodeStub *mocks.ChaincodeStub, testAsset *chaincode.AssetPrivateDetails) []byte {
// if testAsset == nil {
// chaincodeStub.GetPrivateDataReturns(nil, nil)
// return nil
// } else {
// var err error
// assetBytes, err := json.Marshal(testAsset)
// require.NoError(t, err)
// chaincodeStub.GetPrivateDataReturns(assetBytes, nil)
// return assetBytes
// }
// }
func TestCreateUserID(t *testing.T) {
chaincodeStub := &mocks.ChaincodeStub{}
transactionContext := &mocks.TransactionContext{}
transactionContext.GetStubReturns(chaincodeStub)
contract := chaincode.SmartContract{}
err := contract.CreateUserID(transactionContext, "testID", "testOrg")
require.NoError(t, err)
}
func TestCreateUserIDDuplicate(t *testing.T) {
chaincodeStub := &mocks.ChaincodeStub{}
transactionContext := &mocks.TransactionContext{}
transactionContext.GetStubReturns(chaincodeStub)
existingUser := &chaincode.User{UUID: "uuid1", APIUserId: []string{"testID"}, Org: "testOrg"}
bytes, err := json.Marshal(existingUser)
require.NoError(t, err)
chaincodeStub.GetStateReturns(bytes, nil)
contract := chaincode.SmartContract{}
err = contract.CreateUserID(transactionContext, "testID", "testOrg")
require.EqualError(t, err, "the user with APIId testID already exists")
}
Reference in a new issue View git blame Copy permalink