Skip to content

Commit 56d06f7

Browse files
committed
[FAB-4335] Remove checks on mock/util function
External unit tests that call `GenerateMockPublicPrivateKeyPairPEM` shouldn't have to check for the error returned by this function. This should be checked once by the `TestGenerateMockPublicPrivateKeyPairPEM` test. This changeset addresses this. Change-Id: Ia70f707f4943647ef614533686b99105213e855d Signed-off-by: Kostas Christidis <[email protected]>
1 parent a690a05 commit 56d06f7

File tree

4 files changed

+35
-45
lines changed

4 files changed

+35
-45
lines changed

common/viperutil/config_test.go

+2-8
Original file line numberDiff line numberDiff line change
@@ -217,10 +217,7 @@ func TestPEMBlocksFromFile(t *testing.T) {
217217
numberOfCertificates := 3
218218
var pems []byte
219219
for i := 0; i < numberOfCertificates; i++ {
220-
publicKeyCert, _, err := util.GenerateMockPublicPrivateKeyPairPEM(true)
221-
if err != nil {
222-
t.Fatalf("Enable to generate a signer certificate: %v", err)
223-
}
220+
publicKeyCert, _, _ := util.GenerateMockPublicPrivateKeyPairPEM(true)
224221
pems = append(pems, publicKeyCert...)
225222
}
226223

@@ -259,10 +256,7 @@ func TestPEMBlocksFromFileEnv(t *testing.T) {
259256
numberOfCertificates := 3
260257
var pems []byte
261258
for i := 0; i < numberOfCertificates; i++ {
262-
publicKeyCert, _, err := util.GenerateMockPublicPrivateKeyPairPEM(true)
263-
if err != nil {
264-
t.Fatalf("Enable to generate a signer certificate: %v", err)
265-
}
259+
publicKeyCert, _, _ := util.GenerateMockPublicPrivateKeyPairPEM(true)
266260
pems = append(pems, publicKeyCert...)
267261
}
268262

orderer/kafka/util_test.go

+6-24
Original file line numberDiff line numberDiff line change
@@ -112,14 +112,8 @@ func TestNewBrokerConfig(t *testing.T) {
112112
}
113113

114114
func TestTLSConfigEnabled(t *testing.T) {
115-
publicKey, privateKey, err := util.GenerateMockPublicPrivateKeyPairPEM(false)
116-
if err != nil {
117-
t.Fatalf("Enable to generate a public/private key pair: %v", err)
118-
}
119-
caPublicKey, _, err := util.GenerateMockPublicPrivateKeyPairPEM(true)
120-
if err != nil {
121-
t.Fatalf("Enable to generate a signer certificate: %v", err)
122-
}
115+
publicKey, privateKey, _ := util.GenerateMockPublicPrivateKeyPairPEM(false)
116+
caPublicKey, _, _ := util.GenerateMockPublicPrivateKeyPairPEM(true)
123117

124118
config := newBrokerConfig(testConf.Kafka.Version, 0, config.TLS{
125119
Enabled: true,
@@ -137,14 +131,8 @@ func TestTLSConfigEnabled(t *testing.T) {
137131
}
138132

139133
func TestTLSConfigDisabled(t *testing.T) {
140-
publicKey, privateKey, err := util.GenerateMockPublicPrivateKeyPairPEM(false)
141-
if err != nil {
142-
t.Fatalf("Enable to generate a public/private key pair: %v", err)
143-
}
144-
caPublicKey, _, err := util.GenerateMockPublicPrivateKeyPairPEM(true)
145-
if err != nil {
146-
t.Fatalf("Enable to generate a signer certificate: %v", err)
147-
}
134+
publicKey, privateKey, _ := util.GenerateMockPublicPrivateKeyPairPEM(false)
135+
caPublicKey, _, _ := util.GenerateMockPublicPrivateKeyPairPEM(true)
148136

149137
config := newBrokerConfig(testConf.Kafka.Version, 0, config.TLS{
150138
Enabled: false,
@@ -159,14 +147,8 @@ func TestTLSConfigDisabled(t *testing.T) {
159147
}
160148

161149
func TestTLSConfigBadCert(t *testing.T) {
162-
publicKey, privateKey, err := util.GenerateMockPublicPrivateKeyPairPEM(false)
163-
if err != nil {
164-
t.Fatalf("Enable to generate a public/private key pair: %v", err)
165-
}
166-
caPublicKey, _, err := util.GenerateMockPublicPrivateKeyPairPEM(true)
167-
if err != nil {
168-
t.Fatalf("Enable to generate a signer certificate: %v", err)
169-
}
150+
publicKey, privateKey, _ := util.GenerateMockPublicPrivateKeyPairPEM(false)
151+
caPublicKey, _, _ := util.GenerateMockPublicPrivateKeyPairPEM(true)
170152

171153
t.Run("BadPrivateKey", func(t *testing.T) {
172154
assert.Panics(t, func() {

orderer/mocks/util/util.go

+4-13
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,7 @@
11
/*
2-
Copyright IBM Corp. 2017 All Rights Reserved.
2+
Copyright IBM Corp. All Rights Reserved.
33
4-
Licensed under the Apache License, Version 2.0 (the "License");
5-
you may not use this file except in compliance with the License.
6-
You may obtain a copy of the License at
7-
8-
http://www.apache.org/licenses/LICENSE-2.0
9-
10-
Unless required by applicable law or agreed to in writing, software
11-
distributed under the License is distributed on an "AS IS" BASIS,
12-
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13-
See the License for the specific language governing permissions and
14-
limitations under the License.
4+
SPDX-License-Identifier: Apache-2.0
155
*/
166

177
package util
@@ -25,7 +15,8 @@ import (
2515
"math/big"
2616
)
2717

28-
// GenerateMockPublicPrivateKeyPairPEM returns public/private key pair encoded as PEM strings
18+
// GenerateMockPublicPrivateKeyPairPEM returns public/private key pair encoded
19+
// as PEM strings.
2920
func GenerateMockPublicPrivateKeyPairPEM(isCA bool) (string, string, error) {
3021
privateKey, err := rsa.GenerateKey(rand.Reader, 1024)
3122
if err != nil {

orderer/mocks/util/util_test.go

+23
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
/*
2+
Copyright IBM Corp. All Rights Reserved.
3+
4+
SPDX-License-Identifier: Apache-2.0
5+
*/
6+
7+
package util
8+
9+
import (
10+
"testing"
11+
12+
"github.com/stretchr/testify/assert"
13+
)
14+
15+
func TestGenerateMockPublicPrivateKeyPairPEM(t *testing.T) {
16+
_, _, err := GenerateMockPublicPrivateKeyPairPEM(false)
17+
assert.NoError(t, err, "Unable to generate a public/private key pair: %v", err)
18+
}
19+
20+
func TestGenerateMockPublicPrivateKeyPairPEMWhenCASet(t *testing.T) {
21+
_, _, err := GenerateMockPublicPrivateKeyPairPEM(true)
22+
assert.NoError(t, err, "Unable to generate a signer certificate: %v", err)
23+
}

0 commit comments

Comments
 (0)