go_study/fabric-main/cmd/ledgerutil/main_test.go

80 lines
1.6 KiB
Go

/*
Copyright IBM Corp. All Rights Reserved.
SPDX-License-Identifier: Apache-2.0
*/
package main
import (
"os/exec"
"testing"
"time"
"github.com/onsi/gomega"
"github.com/onsi/gomega/gexec"
)
func TestArguments(t *testing.T) {
testCases := map[string]struct {
exitCode int
args []string
}{
"ledger": {
exitCode: 0,
args: []string{},
},
"ledger-help": {
exitCode: 0,
args: []string{"--help"},
},
"compare-help": {
exitCode: 0,
args: []string{"compare", "--help"},
},
"compare": {
exitCode: 1,
args: []string{"compare"},
},
"one-snapshot": {
exitCode: 1,
args: []string{"compare", "snapshotDir1"},
},
"invalid-snapshot-dirs": {
exitCode: 1,
args: []string{"compare", "/non-existent/snapshot1", "/non-existent/snapshot2"},
},
"identifytxs-help": {
exitCode: 0,
args: []string{"identifytxs", "--help"},
},
"identifytxs": {
exitCode: 1,
args: []string{"identifytxs"},
},
"verify-help": {
exitCode: 0,
args: []string{"verify", "--help"},
},
"verify": {
exitCode: 1,
args: []string{"verify"},
},
}
// Build ledger binary
gt := gomega.NewWithT(t)
ledgerutil, err := gexec.Build("github.com/hyperledger/fabric/cmd/ledgerutil")
gt.Expect(err).NotTo(gomega.HaveOccurred())
defer gexec.CleanupBuildArtifacts()
for testName, testCase := range testCases {
t.Run(testName, func(t *testing.T) {
cmd := exec.Command(ledgerutil, testCase.args...)
session, err := gexec.Start(cmd, nil, nil)
gt.Expect(err).NotTo(gomega.HaveOccurred())
gt.Eventually(session, 5*time.Second).Should(gexec.Exit(testCase.exitCode))
})
}
}