1818
1919import static com .google .common .truth .Truth .assertThat ;
2020
21- import com .beust .jcommander .JCommander ;
2221import io .grpc .s2a .handshaker .S2AIdentity ;
2322import java .util .Optional ;
2423import org .junit .Before ;
3029public final class SingleTokenAccessTokenManagerTest {
3130 private static final S2AIdentity IDENTITY = S2AIdentity .fromSpiffeId ("spiffe_id" );
3231 private static final String TOKEN = "token" ;
33- private static final String [] SET_TOKEN = {"--s2a_access_token" , TOKEN };
34- private static final SingleTokenFetcher .Flags FLAGS = new SingleTokenFetcher .Flags ();
3532
3633 @ Before
3734 public void setUp () {
38- FLAGS . reset ( );
35+ SingleTokenFetcher . setAccessToken ( null );
3936 }
4037
4138 @ Test
4239 public void getDefaultToken_success () throws Exception {
43- JCommander . newBuilder (). addObject ( FLAGS ). build (). parse ( SET_TOKEN );
40+ SingleTokenFetcher . setAccessToken ( TOKEN );
4441 Optional <AccessTokenManager > manager = AccessTokenManager .create ();
4542 assertThat (manager ).isPresent ();
4643 assertThat (manager .get ().getDefaultToken ()).isEqualTo (TOKEN );
4744 }
4845
4946 @ Test
5047 public void getToken_success () throws Exception {
51- JCommander . newBuilder (). addObject ( FLAGS ). build (). parse ( SET_TOKEN );
48+ SingleTokenFetcher . setAccessToken ( TOKEN );
5249 Optional <AccessTokenManager > manager = AccessTokenManager .create ();
5350 assertThat (manager ).isPresent ();
5451 assertThat (manager .get ().getToken (IDENTITY )).isEqualTo (TOKEN );
@@ -61,7 +58,7 @@ public void getToken_noEnvironmentVariable() throws Exception {
6158
6259 @ Test
6360 public void create_success () throws Exception {
64- JCommander . newBuilder (). addObject ( FLAGS ). build (). parse ( SET_TOKEN );
61+ SingleTokenFetcher . setAccessToken ( TOKEN );
6562 Optional <AccessTokenManager > manager = AccessTokenManager .create ();
6663 assertThat (manager ).isPresent ();
6764 assertThat (manager .get ().getToken (IDENTITY )).isEqualTo (TOKEN );
0 commit comments