@@ -8,6 +8,7 @@ package admin
8
8
9
9
import (
10
10
"context"
11
+ "strings"
11
12
"testing"
12
13
13
14
"github.com/hyperledger/fabric/common/flogging"
@@ -121,16 +122,47 @@ func TestLoggingCalls(t *testing.T) {
121
122
assert .Nil (t , err , "Error should have been nil" )
122
123
}
123
124
124
- for _ , llr := range []* pb.LogLevelRequest {{LogModule : "test" , LogLevel : "debug" }, nil } {
125
- mv .On ("validate" ).Return (wrapLogLevelRequest (llr ), nil ).Once ()
125
+ type levelTestCase struct {
126
+ req * pb.LogLevelRequest
127
+ expectedErr string
128
+ }
129
+
130
+ levelTests := []levelTestCase {
131
+ {
132
+ req : & pb.LogLevelRequest {
133
+ LogModule : "test" ,
134
+ LogLevel : "borken" ,
135
+ },
136
+ expectedErr : "rpc error: code = InvalidArgument desc = error setting log spec to 'info:test=borken': invalid logging specification 'info:test=borken': bad segment 'test=borken'" ,
137
+ },
138
+ {
139
+ req : & pb.LogLevelRequest {
140
+ LogModule : "test" ,
141
+ LogLevel : "debug" ,
142
+ },
143
+ },
144
+ {
145
+ req : nil ,
146
+ },
147
+ }
148
+
149
+ for _ , tc := range levelTests {
150
+ mv .On ("validate" ).Return (wrapLogLevelRequest (tc .req ), nil ).Once ()
126
151
logResponse , err := adminServer .SetModuleLogLevel (context .Background (), nil )
127
- if llr == nil {
152
+
153
+ if tc .req == nil {
128
154
assert .Nil (t , logResponse )
129
155
assert .Equal (t , "request is nil" , err .Error ())
130
156
continue
131
157
}
158
+
159
+ if tc .expectedErr != "" {
160
+ assert .Equal (t , tc .expectedErr , err .Error ())
161
+ continue
162
+ }
163
+
132
164
assert .NotNil (t , logResponse , "logResponse should have been set" )
133
- assert .Equal (t , "DEBUG" , logResponse .LogLevel , "logger level should have been set to debug" )
165
+ assert .Equal (t , tc . req . LogLevel , strings . ToLower ( logResponse .LogLevel ) )
134
166
assert .Nil (t , err , "Error should have been nil" )
135
167
}
136
168
@@ -148,32 +180,34 @@ func TestLoggingCalls(t *testing.T) {
148
180
_ , err = adminServer .GetLogSpec (context .Background (), nil )
149
181
assert .Nil (t , err , "Error should have been nil" )
150
182
151
- type testCase struct {
183
+ type specTestCase struct {
152
184
req * pb.LogSpecRequest
153
185
expectedErr string
154
186
}
155
187
156
- testCases := []testCase {
188
+ specTests := []specTestCase {
189
+ {
190
+ req : nil ,
191
+ },
157
192
{
158
193
req : & pb.LogSpecRequest {LogSpec : "info" },
159
194
},
160
195
{
161
196
req : & pb.LogSpecRequest {LogSpec : "borken" },
162
197
expectedErr : "invalid logging specification 'borken': bad segment 'borken'" ,
163
198
},
164
- {
165
- req : nil ,
166
- },
167
199
}
168
200
169
- for _ , tc := range testCases {
201
+ for _ , tc := range specTests {
170
202
mv .On ("validate" ).Return (wrapLogSpecRequest (tc .req ), nil ).Once ()
171
203
resp , err := adminServer .SetLogSpec (context .Background (), nil )
204
+
172
205
if tc .req == nil {
173
206
assert .Nil (t , resp )
174
207
assert .Equal (t , "request is nil" , err .Error ())
175
208
continue
176
209
}
210
+
177
211
assert .Nil (t , err , "Error should have been nil" )
178
212
assert .Equal (t , tc .expectedErr , resp .Error )
179
213
0 commit comments