16
16
17
17
package io .grpc .examples .authentication ;
18
18
19
- import io .grpc .*;
19
+ import static org .junit .Assert .assertEquals ;
20
+ import static org .mockito .AdditionalAnswers .delegatesTo ;
21
+ import static org .mockito .Mockito .mock ;
22
+ import static org .mockito .Mockito .verify ;
23
+
24
+ import io .grpc .Metadata ;
25
+ import io .grpc .ServerCall ;
26
+ import io .grpc .ServerCallHandler ;
27
+ import io .grpc .ServerInterceptors ;
20
28
import io .grpc .examples .helloworld .GreeterGrpc ;
21
29
import io .grpc .examples .helloworld .HelloReply ;
22
30
import io .grpc .examples .helloworld .HelloRequest ;
23
31
import io .grpc .inprocess .InProcessChannelBuilder ;
24
32
import io .grpc .inprocess .InProcessServerBuilder ;
25
33
import io .grpc .ServerCall .Listener ;
26
34
import io .grpc .ServerInterceptor ;
35
+ import io .grpc .stub .StreamObserver ;
27
36
import io .grpc .testing .GrpcCleanupRule ;
37
+ import java .io .IOException ;
28
38
import org .junit .Before ;
29
39
import org .junit .Rule ;
30
40
import org .junit .Test ;
33
43
import org .mockito .ArgumentCaptor ;
34
44
import org .mockito .Matchers ;
35
45
36
- import java .io .IOException ;
37
-
38
- import static org .junit .Assert .assertEquals ;
39
- import static org .mockito .AdditionalAnswers .delegatesTo ;
40
- import static org .mockito .Mockito .mock ;
41
- import static org .mockito .Mockito .verify ;
42
-
43
46
/**
44
47
* Unit tests for {@link AuthClient} testing the default and non-default tokens
45
48
*
@@ -55,13 +58,13 @@ public class AuthClientTest {
55
58
public final GrpcCleanupRule grpcCleanup = new GrpcCleanupRule ();
56
59
57
60
private final ServerInterceptor mockServerInterceptor = mock (ServerInterceptor .class , delegatesTo (
58
- new ServerInterceptor () {
59
- @ Override
60
- public <ReqT , RespT > Listener <ReqT > interceptCall (
61
- ServerCall <ReqT , RespT > call , Metadata headers , ServerCallHandler <ReqT , RespT > next ) {
62
- return next .startCall (call , headers );
63
- }
64
- }));
61
+ new ServerInterceptor () {
62
+ @ Override
63
+ public <ReqT , RespT > Listener <ReqT > interceptCall (
64
+ ServerCall <ReqT , RespT > call , Metadata headers , ServerCallHandler <ReqT , RespT > next ) {
65
+ return next .startCall (call , headers );
66
+ }
67
+ }));
65
68
66
69
private AuthClient client ;
67
70
@@ -73,64 +76,63 @@ public void setUp() throws IOException {
73
76
74
77
// Create a server, add service, start, and register for automatic graceful shutdown.
75
78
grpcCleanup .register (InProcessServerBuilder .forName (serverName ).directExecutor ()
76
- .addService (ServerInterceptors .intercept (new GreeterGrpc .GreeterImplBase () {
79
+ .addService (ServerInterceptors .intercept (
80
+ new GreeterGrpc .GreeterImplBase () {
77
81
78
82
@ Override
79
- public void sayHello (io .grpc .examples .helloworld .HelloRequest request ,
80
- io .grpc .stub .StreamObserver <io .grpc .examples .helloworld .HelloReply > responseObserver ) {
81
- HelloReply reply = HelloReply .newBuilder ().setMessage ("AuthClientTest user=" + request .getName ()).build ();
83
+ public void sayHello (
84
+ HelloRequest request , StreamObserver <HelloReply > responseObserver ) {
85
+ HelloReply reply = HelloReply .newBuilder ()
86
+ .setMessage ("AuthClientTest user=" + request .getName ()).build ();
82
87
responseObserver .onNext (reply );
83
88
responseObserver .onCompleted ();
84
89
}
85
- }, mockServerInterceptor ))
86
- .build ().start ());
90
+ },
91
+ mockServerInterceptor ))
92
+ .build ().start ());
87
93
88
94
// Create an AuthClient using the in-process channel;
89
95
client = new AuthClient (InProcessChannelBuilder .forName (serverName ).directExecutor ());
90
96
}
91
97
92
98
/**
93
- * Test default JWT token used
94
- *
95
- * @throws Exception
99
+ * Test default JWT token used.
96
100
*/
97
101
@ Test
98
- public void defaultTokenDeliveredToServer () throws Exception {
102
+ public void defaultTokenDeliveredToServer () {
99
103
ArgumentCaptor <Metadata > metadataCaptor = ArgumentCaptor .forClass (Metadata .class );
100
104
ArgumentCaptor <HelloRequest > requestCaptor = ArgumentCaptor .forClass (HelloRequest .class );
101
105
102
106
String retVal = client .greet ("default token test" );
103
107
104
108
verify (mockServerInterceptor ).interceptCall (
105
- Matchers .<ServerCall <HelloRequest , HelloReply >>any (),
106
- metadataCaptor .capture (),
107
- Matchers .<ServerCallHandler <HelloRequest , HelloReply >>any ());
109
+ Matchers .<ServerCall <HelloRequest , HelloReply >>any (),
110
+ metadataCaptor .capture (),
111
+ Matchers .<ServerCallHandler <HelloRequest , HelloReply >>any ());
108
112
assertEquals (
109
- "my-default-token" ,
110
- metadataCaptor .getValue ().get (Constant .JWT_METADATA_KEY ));
113
+ "my-default-token" ,
114
+ metadataCaptor .getValue ().get (Constant .JWT_METADATA_KEY ));
111
115
assertEquals ("AuthClientTest user=default token test" , retVal );
112
116
}
113
117
114
118
/**
115
- * Test non-default JWT token used
116
- *
117
- * @throws Exception
119
+ * Test non-default JWT token used.
118
120
*/
119
121
@ Test
120
- public void nonDefaultTokenDeliveredToServer () throws Exception {
122
+ public void nonDefaultTokenDeliveredToServer () {
121
123
ArgumentCaptor <Metadata > metadataCaptor = ArgumentCaptor .forClass (Metadata .class );
122
124
ArgumentCaptor <HelloRequest > requestCaptor = ArgumentCaptor .forClass (HelloRequest .class );
123
125
124
126
client .setTokenValue ("non-default-token" );
125
127
String retVal = client .greet ("non default token test" );
126
128
127
129
verify (mockServerInterceptor ).interceptCall (
128
- Matchers .<ServerCall <HelloRequest , HelloReply >>any (),
129
- metadataCaptor .capture (),
130
- Matchers .<ServerCallHandler <HelloRequest , HelloReply >>any ());
130
+ Matchers .<ServerCall <HelloRequest , HelloReply >>any (),
131
+ metadataCaptor .capture (),
132
+ Matchers .<ServerCallHandler <HelloRequest , HelloReply >>any ());
131
133
assertEquals (
132
- "non-default-token" ,
133
- metadataCaptor .getValue ().get (Constant .JWT_METADATA_KEY ));
134
+ "non-default-token" ,
135
+ metadataCaptor .getValue ().get (Constant .JWT_METADATA_KEY ));
134
136
assertEquals ("AuthClientTest user=non default token test" , retVal );
135
137
}
136
138
}
0 commit comments