go_study/fabric-main/ccaas_builder/cmd/build/main_test.go

323 lines
10 KiB
Go

/*
Copyright IBM Corp. All Rights Reserved.
SPDX-License-Identifier: Apache-2.0
*/
package main
import (
"io/ioutil"
"os"
"os/exec"
"path"
"testing"
"time"
. "github.com/onsi/gomega"
"github.com/onsi/gomega/gbytes"
"github.com/onsi/gomega/gexec"
)
func TestArguements(t *testing.T) {
testCases := map[string]struct {
exitCode int
args []string
}{
"toofew1": {
exitCode: 1,
args: []string{"metadatadir"},
},
"toofew2": {
exitCode: 1,
args: []string{"wibble", "wibble"},
},
"twomany": {
exitCode: 1,
args: []string{"wibble", "wibble", "wibble", "wibble"},
},
}
// Build ledger binary
gt := NewWithT(t)
buildCmd, err := gexec.Build("github.com/hyperledger/fabric/ccaas_builder/cmd/build")
gt.Expect(err).NotTo(HaveOccurred())
defer gexec.CleanupBuildArtifacts()
for testName, testCase := range testCases {
t.Run(testName, func(t *testing.T) {
cmd := exec.Command(buildCmd, testCase.args...)
session, err := gexec.Start(cmd, nil, nil)
gt.Expect(err).NotTo(HaveOccurred())
gt.Eventually(session, 5*time.Second).Should(gexec.Exit(testCase.exitCode))
})
}
}
func TestGoodPath(t *testing.T) {
gt := NewWithT(t)
releaseCmd, err := gexec.Build("github.com/hyperledger/fabric/ccaas_builder/cmd/build")
gt.Expect(err).NotTo(HaveOccurred())
defer gexec.CleanupBuildArtifacts()
testPath := t.TempDir()
// create a basic structure of a chaincode
os.MkdirAll(path.Join(testPath, "in-builder-dir", "META-INF"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
os.MkdirAll(path.Join(testPath, "in-metadata-dir"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
os.MkdirAll(path.Join(testPath, "out-release-dir"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
connectionJson := path.Join(testPath, "in-builder-dir", "connection.json")
file, err := os.Create(connectionJson)
gt.Expect(err).NotTo(HaveOccurred())
file.WriteString(`{
"address": "audit-trail-ccaas:9999",
"dial_timeout": "10s",
"tls_required": false
}`)
metadataJson := path.Join(testPath, "in-metadata-dir", "metadata.json")
fileMetadata, err := os.Create(metadataJson)
gt.Expect(err).NotTo(HaveOccurred())
fileMetadata.WriteString(`{
"type":"ccaas"
}`)
indexJson := path.Join(testPath, "in-builder-dir", "META-INF", "indexes.json")
fileIndex, err := os.Create(indexJson)
gt.Expect(err).NotTo(HaveOccurred())
fileIndex.WriteString(`{
"couchindexes": "gohere"
}`)
args := []string{path.Join(testPath, "in-builder-dir"), path.Join(testPath, "in-metadata-dir"), path.Join(testPath, "out-release-dir")}
cmd := exec.Command(releaseCmd, args...)
session, err := gexec.Start(cmd, nil, nil)
gt.Expect(err).NotTo(HaveOccurred())
gt.Eventually(session, 5*time.Second).Should(gexec.Exit(0))
// check that the files have been copied
chkfile := path.Join(testPath, "out-release-dir", "connection.json")
_, err = os.Stat(chkfile)
gt.Expect(err).NotTo(HaveOccurred())
chkfile = path.Join(testPath, "out-release-dir", "META-INF", "indexes.json")
_, err = os.Stat(chkfile)
gt.Expect(err).NotTo(HaveOccurred())
}
func TestTemplating(t *testing.T) {
gt := NewWithT(t)
releaseCmd, err := gexec.Build("github.com/hyperledger/fabric/ccaas_builder/cmd/build")
gt.Expect(err).NotTo(HaveOccurred())
defer gexec.CleanupBuildArtifacts()
testPath := t.TempDir()
// create a basic structure of the chaincode to use
os.MkdirAll(path.Join(testPath, "in-builder-dir", "META-INF"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
os.MkdirAll(path.Join(testPath, "in-metadata-dir"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
os.MkdirAll(path.Join(testPath, "out-release-dir"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
connectionJson := path.Join(testPath, "in-builder-dir", "connection.json")
file, err := os.Create(connectionJson)
gt.Expect(err).NotTo(HaveOccurred())
file.WriteString(`{
"address": "{{.address}}:9999",
"dial_timeout": "10s",
"tls_required": true,
"root_cert":"{{.root_cert}}",
"client_key":"{{.client_key}}",
"client_cert":"{{.client_cert}}",
"client_auth_required":true
}`)
metadataJson := path.Join(testPath, "in-metadata-dir", "metadata.json")
fileMetadata, err := os.Create(metadataJson)
gt.Expect(err).NotTo(HaveOccurred())
fileMetadata.WriteString(`{
"type":"ccaas"
}`)
// call 'build' with the environment set for the templating to be tested
args := []string{path.Join(testPath, "in-builder-dir"), path.Join(testPath, "in-metadata-dir"), path.Join(testPath, "out-release-dir")}
cmd := exec.Command(releaseCmd, args...)
env := os.Environ()
env = append(env, `CHAINCODE_AS_A_SERVICE_BUILDER_CONFIG={"address":"URLAddress","root_cert":"a root cert","client_key":"a client key","client_cert":"a client cert"}`)
cmd.Env = env
session, err := gexec.Start(cmd, nil, nil)
gt.Expect(err).NotTo(HaveOccurred())
gt.Eventually(session, 5*time.Second).Should(gexec.Exit(0))
// check that the files have been copied
chkfile := path.Join(testPath, "out-release-dir", "connection.json")
_, err = os.Stat(chkfile)
gt.Expect(err).NotTo(HaveOccurred())
// check that the file has the exepected contents
connectionFileContents, err := ioutil.ReadFile(chkfile)
gt.Expect(err).NotTo(HaveOccurred())
expectedJson := `{
"address": "URLAddress:9999",
"dial_timeout": "10s",
"tls_required": true,
"root_cert":"a root cert",
"client_key":"a client key",
"client_cert":"a client cert",
"client_auth_required":true
}`
gt.Expect(string(connectionFileContents)).To(MatchJSON(expectedJson))
}
func TestTemplatingFailure(t *testing.T) {
gt := NewWithT(t)
releaseCmd, err := gexec.Build("github.com/hyperledger/fabric/ccaas_builder/cmd/build")
gt.Expect(err).NotTo(HaveOccurred())
defer gexec.CleanupBuildArtifacts()
testPath := t.TempDir()
// create a basic structure of the chaincode to use
os.MkdirAll(path.Join(testPath, "in-builder-dir", "META-INF"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
os.MkdirAll(path.Join(testPath, "in-metadata-dir"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
os.MkdirAll(path.Join(testPath, "out-release-dir"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
connectionJson := path.Join(testPath, "in-builder-dir", "connection.json")
file, err := os.Create(connectionJson)
gt.Expect(err).NotTo(HaveOccurred())
file.WriteString(`{
"address": "{{.address}}:9999",
"dial_timeout": "10s",
"tls_required": true,
"root_cert":"{{.root_cert}}",
"client_key":"{{.client_key}}",
"client_cert":"{{.client_cert}}",
"client_auth_required":true
}`)
metadataJson := path.Join(testPath, "in-metadata-dir", "metadata.json")
fileMetadata, err := os.Create(metadataJson)
gt.Expect(err).NotTo(HaveOccurred())
fileMetadata.WriteString(`{
"type":"ccaas"
}`)
// call 'build' with the environment set for the templating to be tested
args := []string{path.Join(testPath, "in-builder-dir"), path.Join(testPath, "in-metadata-dir"), path.Join(testPath, "out-release-dir")}
cmd := exec.Command(releaseCmd, args...)
env := os.Environ()
env = append(env, `CHAINCODE_AS_A_SERVICE_BUILDER_CONFIG={"address":"URLAddress","root_cert":"a root cert"}`)
cmd.Env = env
session, err := gexec.Start(cmd, nil, nil)
gt.Expect(err).NotTo(HaveOccurred())
gt.Eventually(session, 5*time.Second).Should(gexec.Exit(1))
gt.Expect(session.Err).To(gbytes.Say("Failed to parse the ClientKey field template"))
}
func TestMissingConnection(t *testing.T) {
gt := NewWithT(t)
releaseCmd, err := gexec.Build("github.com/hyperledger/fabric/ccaas_builder/cmd/build")
gt.Expect(err).NotTo(HaveOccurred())
defer gexec.CleanupBuildArtifacts()
testPath := t.TempDir()
// create a basic structure of a chaincode
os.MkdirAll(path.Join(testPath, "in-builder-dir", "META-INF"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
os.MkdirAll(path.Join(testPath, "in-metadata-dir"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
os.MkdirAll(path.Join(testPath, "out-release-dir"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
metadataJson := path.Join(testPath, "in-metadata-dir", "metadata.json")
fileMetadata, err := os.Create(metadataJson)
gt.Expect(err).NotTo(HaveOccurred())
fileMetadata.WriteString(`{
"type":"ccaas"
}`)
indexJson := path.Join(testPath, "in-builder-dir", "META-INF", "indexes.json")
fileIndex, err := os.Create(indexJson)
gt.Expect(err).NotTo(HaveOccurred())
fileIndex.WriteString(`{
"couchindexes": "gohere"
}`)
args := []string{path.Join(testPath, "in-builder-dir"), path.Join(testPath, "in-metadata-dir"), path.Join(testPath, "out-release-dir")}
cmd := exec.Command(releaseCmd, args...)
session, err := gexec.Start(cmd, nil, nil)
gt.Expect(err).NotTo(HaveOccurred())
gt.Eventually(session, 5*time.Second).Should(gexec.Exit(1))
gt.Expect(session.Err).To(gbytes.Say("connection.json not found"))
}
func TestMissingMetadata(t *testing.T) {
gt := NewWithT(t)
releaseCmd, err := gexec.Build("github.com/hyperledger/fabric/ccaas_builder/cmd/build")
gt.Expect(err).NotTo(HaveOccurred())
defer gexec.CleanupBuildArtifacts()
testPath := t.TempDir()
// create a basic structure of a chaincode
os.MkdirAll(path.Join(testPath, "in-builder-dir", "META-INF"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
os.MkdirAll(path.Join(testPath, "in-metadata-dir"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
os.MkdirAll(path.Join(testPath, "out-release-dir"), 0755)
gt.Expect(err).NotTo(HaveOccurred())
connectionJson := path.Join(testPath, "in-builder-dir", "connection.json")
file, err := os.Create(connectionJson)
gt.Expect(err).NotTo(HaveOccurred())
file.WriteString(`{
"address": "audit-trail-ccaas:9999",
"dial_timeout": "10s",
"tls_required": false
}`)
indexJson := path.Join(testPath, "in-builder-dir", "META-INF", "indexes.json")
fileIndex, err := os.Create(indexJson)
gt.Expect(err).NotTo(HaveOccurred())
fileIndex.WriteString(`{
"couchindexes": "gohere"
}`)
args := []string{path.Join(testPath, "in-builder-dir"), path.Join(testPath, "in-metadata-dir"), path.Join(testPath, "out-release-dir")}
cmd := exec.Command(releaseCmd, args...)
session, err := gexec.Start(cmd, nil, nil)
gt.Expect(err).NotTo(HaveOccurred())
gt.Eventually(session, 5*time.Second).Should(gexec.Exit(1))
gt.Expect(session.Err).To(gbytes.Say("metadata.json not found"))
}