mirror of
https://github.com/hyperledger/fabric-samples.git
synced 2026-06-23 01:55:10 +00:00
23dc4e698a
Many users raise questions on how to test chaincode. It used to be much easier with the old shim, as they could directly use the old mock stub in the shim. Now that it no longer exists the fabcar example can provide an example of how to test chaincode. Also worth noting is our current directory structure of our Go chaincodes prevents the creation of mocks due to import cycles. This change also pushes the chaincode logic down into a `contract` package. Signed-off-by: Brett Logan <brett.t.logan@ibm.com>
133 lines
4 KiB
Go
133 lines
4 KiB
Go
package contract_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/hyperledger/fabric-protos-go/ledger/queryresult"
|
|
"github.com/hyperledger/fabric-samples/chaincode/fabcar/go/contract"
|
|
"github.com/hyperledger/fabric-samples/chaincode/fabcar/go/contract/mocks"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestInitLedger(t *testing.T) {
|
|
chaincodeStub := &mocks.ChaincodeStub{}
|
|
transactionContext := &mocks.TransactionContext{}
|
|
transactionContext.GetStubReturns(chaincodeStub)
|
|
|
|
fabcar := &contract.SmartContract{}
|
|
err := fabcar.InitLedger(transactionContext)
|
|
require.NoError(t, err)
|
|
|
|
chaincodeStub.PutStateReturns(fmt.Errorf("failed inserting key"))
|
|
err = fabcar.InitLedger(transactionContext)
|
|
require.EqualError(t, err, "failed to put to world state. failed inserting key")
|
|
}
|
|
|
|
func TestCreateCar(t *testing.T) {
|
|
chaincodeStub := &mocks.ChaincodeStub{}
|
|
transactionContext := &mocks.TransactionContext{}
|
|
transactionContext.GetStubReturns(chaincodeStub)
|
|
|
|
fabcar := &contract.SmartContract{}
|
|
err := fabcar.CreateCar(transactionContext, "CAR1", "Ford", "F150", "red", "Tim")
|
|
require.NoError(t, err)
|
|
|
|
chaincodeStub.PutStateReturns(fmt.Errorf("failed inserting key"))
|
|
err = fabcar.CreateCar(transactionContext, "CAR1", "Ford", "F150", "red", "Tim")
|
|
require.EqualError(t, err, "failed inserting key")
|
|
|
|
}
|
|
|
|
func TestQueryCar(t *testing.T) {
|
|
car := &contract.Car{
|
|
Make: "Ford",
|
|
Model: "F150",
|
|
Colour: "red",
|
|
Owner: "Tim",
|
|
}
|
|
bytes, err := json.Marshal(car)
|
|
require.NoError(t, err)
|
|
|
|
chaincodeStub := &mocks.ChaincodeStub{}
|
|
chaincodeStub.GetStateReturns(bytes, nil)
|
|
transactionContext := &mocks.TransactionContext{}
|
|
transactionContext.GetStubReturns(chaincodeStub)
|
|
|
|
fabcar := &contract.SmartContract{}
|
|
car, err = fabcar.QueryCar(transactionContext, "CAR1")
|
|
require.NoError(t, err)
|
|
require.NotNil(t, car)
|
|
|
|
chaincodeStub.GetStateReturns(nil, nil)
|
|
car, err = fabcar.QueryCar(transactionContext, "CAR1")
|
|
require.EqualError(t, err, "CAR1 does not exist")
|
|
require.Nil(t, car)
|
|
|
|
chaincodeStub.GetStateReturns([]byte{}, fmt.Errorf("failed retrieving key"))
|
|
car, err = fabcar.QueryCar(transactionContext, "CAR1")
|
|
require.Nil(t, car)
|
|
require.EqualError(t, err, "failed to read from world state: failed retrieving key")
|
|
}
|
|
|
|
func TestQueryAllCars(t *testing.T) {
|
|
car := &contract.Car{
|
|
Make: "Ford",
|
|
Model: "F150",
|
|
Colour: "red",
|
|
Owner: "Tim",
|
|
}
|
|
bytes, err := json.Marshal(car)
|
|
require.NoError(t, err)
|
|
|
|
iterator := &mocks.StateQueryIterator{}
|
|
iterator.HasNextReturnsOnCall(0, true)
|
|
iterator.HasNextReturnsOnCall(1, false)
|
|
iterator.NextReturns(&queryresult.KV{Value: bytes}, nil)
|
|
|
|
chaincodeStub := &mocks.ChaincodeStub{}
|
|
chaincodeStub.GetStateByRangeReturns(iterator, nil)
|
|
transactionContext := &mocks.TransactionContext{}
|
|
transactionContext.GetStubReturns(chaincodeStub)
|
|
|
|
fabcar := &contract.SmartContract{}
|
|
cars, err := fabcar.QueryAllCars(transactionContext)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, cars)
|
|
|
|
iterator.HasNextReturns(true)
|
|
iterator.NextReturns(nil, fmt.Errorf("failed retrieving next item"))
|
|
cars, err = fabcar.QueryAllCars(transactionContext)
|
|
require.EqualError(t, err, "failed retrieving next item")
|
|
require.Nil(t, cars)
|
|
|
|
chaincodeStub.GetStateByRangeReturns(nil, fmt.Errorf("failed retrieving all cars"))
|
|
cars, err = fabcar.QueryAllCars(transactionContext)
|
|
require.EqualError(t, err, "failed retrieving all cars")
|
|
require.Nil(t, cars)
|
|
}
|
|
|
|
func TestChangeCarOwner(t *testing.T) {
|
|
car := &contract.Car{
|
|
Make: "Ford",
|
|
Model: "F150",
|
|
Colour: "red",
|
|
Owner: "Tim",
|
|
}
|
|
bytes, err := json.Marshal(car)
|
|
require.NoError(t, err)
|
|
|
|
chaincodeStub := &mocks.ChaincodeStub{}
|
|
chaincodeStub.GetStateReturns(bytes, nil)
|
|
transactionContext := &mocks.TransactionContext{}
|
|
transactionContext.GetStubReturns(chaincodeStub)
|
|
|
|
fabcar := &contract.SmartContract{}
|
|
err = fabcar.ChangeCarOwner(transactionContext, "CAR1", "Ben")
|
|
require.NoError(t, err)
|
|
|
|
chaincodeStub.GetStateReturns(nil, nil)
|
|
err = fabcar.ChangeCarOwner(transactionContext, "CAR1", "Ben")
|
|
require.EqualError(t, err, "CAR1 does not exist")
|
|
}
|