Update TestRollout.java

Author: lukasmatusiewicz

src/test/java/org/privacyidea/TestRollout.java

@@ -24,15 +24,15 @@
 import org.mockserver.model.HttpRequest;
 import org.mockserver.model.HttpResponse;
 
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertNotNull;
-import static org.junit.Assert.assertNull;
-import static org.junit.Assert.assertTrue;
+import java.util.ArrayList;
+
+import static org.junit.Assert.*;
 
 public class TestRollout
 {
     private PrivacyIDEA privacyIDEA;
     private ClientAndServer mockServer;
+    String authToken = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VybmFtZSI6ImFkbWluIiwicmVhbG0iOiIiLCJub25jZSI6IjNjMTZmNGUxODg2NmVmMjI1NmM3OWIwOGM1ZTkzODUzYTViMTgyMTMiLCJyb2xlIjoiYWRtaW4iLCJhdXRodHlwZSI6InBhc3N3b3JkIiwiZXhwIjoxNzM4MDY2NDc5LCJyaWdodHMiOlsidHJpZ2dlcmNoYWxsZW5nZSIsInJlc29sdmVyd3JpdGUiLCJlbnJvbGxXRUJBVVRITiIsImF1ZGl0bG9nIiwic3RhdGlzdGljc19kZWxldGUiLCJwb2xpY3lkZWxldGUiLCJlbnJvbGxWQVNDTyIsInJhZGl1c3NlcnZlcl9yZWFkIiwiZW5yb2xsU1NIS0VZIiwic210cHNlcnZlcl93cml0ZSIsImVucm9sbFFVRVNUSU9OIiwiZW5yb2xsVEFOIiwibG9zdHRva2VuIiwic3lzdGVtX2RvY3VtZW50YXRpb24iLCJzZXQiLCJlbmFibGUiLCJlbnJvbGxSRU1PVEUiLCJjb25maWdyZWFkIiwiY29weXRva2VudXNlciIsImVucm9sbFlVQklDTyIsImNvcHl0b2tlbnBpbiIsIm1yZXNvbHZlcndyaXRlIiwiaW1wb3J0dG9rZW5zIiwic2V0cGluIiwiZW5yb2xsUFVTSCIsImVucm9sbEVNQUlMIiwiY29uZmlnZGVsZXRlIiwibXJlc29sdmVycmVhZCIsInRva2VucmVhbG1zIiwiY2Fjb25uZWN0b3J3cml0ZSIsInBlcmlvZGljdGFza193cml0ZSIsImVucm9sbFBBUEVSIiwidG9rZW5saXN0Iiwic2V0X2hzbV9wYXNzd29yZCIsInJlc29sdmVyZGVsZXRlIiwiYXVkaXRsb2dfZG93bmxvYWQiLCJyZXNldCIsInNtc2dhdGV3YXlfcmVhZCIsImVucm9sbFRPVFAiLCJwb2xpY3lyZWFkIiwiZW5yb2xsWVVCSUtFWSIsInJlc29sdmVycmVhZCIsIm1hbmFnZV9tYWNoaW5lX3Rva2VucyIsImFzc2lnbiIsImdldHJhbmRvbSIsImFkZHVzZXIiLCJmZXRjaF9hdXRoZW50aWNhdGlvbl9pdGVtcyIsImVucm9sbFBXIiwibWFjaGluZWxpc3QiLCJjbGllbnR0eXBlIiwiZW5yb2xsQ0VSVElGSUNBVEUiLCJjb25maWd3cml0ZSIsImVucm9sbFNQQVNTIiwibWFuYWdlc3Vic2NyaXB0aW9uIiwiZW5yb2xsREFQTFVHIiwiZ2V0Y2hhbGxlbmdlcyIsInVzZXJsaXN0IiwiZW5yb2xsUkVHSVNUUkFUSU9OIiwiZW5yb2xsUkFESVVTIiwicmFkaXVzc2VydmVyX3dyaXRlIiwiZGVsZXRlIiwic3RhdGlzdGljc19yZWFkIiwidXBkYXRldXNlciIsImVucm9sbFRJUVIiLCJzbXRwc2VydmVyX3JlYWQiLCJlbnJvbGxwaW4iLCJ1bmFzc2lnbiIsImNhY29ubmVjdG9yZGVsZXRlIiwiZW5yb2xsTU9UUCIsImdldHNlcmlhbCIsInJlc3luYyIsImVucm9sbDRFWUVTIiwiZW5yb2xsT0NSQSIsInNtc2dhdGV3YXlfd3JpdGUiLCJlbnJvbGxIT1RQIiwiZXZlbnRoYW5kbGluZ19yZWFkIiwiZXZlbnRoYW5kbGluZ193cml0ZSIsInByaXZhY3lpZGVhc2VydmVyX3dyaXRlIiwibXJlc29sdmVyZGVsZXRlIiwicHJpdmFjeWlkZWFzZXJ2ZXJfcmVhZCIsImVucm9sbFUyRiIsInJldm9rZSIsInBlcmlvZGljdGFza19yZWFkIiwicG9saWN5d3JpdGUiLCJkaXNhYmxlIiwiZGVsZXRldXNlciIsImVucm9sbFNNUyIsInNldHRva2VuaW5mbyJdfQ.bgxeEFPcwTY9V8jxLHtDQPGlmfxewc7HSV29Hutd3H8";
 
     @Before
     public void setup()
@@ -43,7 +43,6 @@ public void setup()
     @Test
     public void testSuccess()
     {
-        String authToken = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VybmFtZSI6ImFkbWluIiwicmVhbG0iOiIiLCJub25jZSI6IjNjMTZmNGUxODg2NmVmMjI1NmM3OWIwOGM1ZTkzODUzYTViMTgyMTMiLCJyb2xlIjoiYWRtaW4iLCJhdXRodHlwZSI6InBhc3N3b3JkIiwiZXhwIjoxNzM4MDY2NDc5LCJyaWdodHMiOlsidHJpZ2dlcmNoYWxsZW5nZSIsInJlc29sdmVyd3JpdGUiLCJlbnJvbGxXRUJBVVRITiIsImF1ZGl0bG9nIiwic3RhdGlzdGljc19kZWxldGUiLCJwb2xpY3lkZWxldGUiLCJlbnJvbGxWQVNDTyIsInJhZGl1c3NlcnZlcl9yZWFkIiwiZW5yb2xsU1NIS0VZIiwic210cHNlcnZlcl93cml0ZSIsImVucm9sbFFVRVNUSU9OIiwiZW5yb2xsVEFOIiwibG9zdHRva2VuIiwic3lzdGVtX2RvY3VtZW50YXRpb24iLCJzZXQiLCJlbmFibGUiLCJlbnJvbGxSRU1PVEUiLCJjb25maWdyZWFkIiwiY29weXRva2VudXNlciIsImVucm9sbFlVQklDTyIsImNvcHl0b2tlbnBpbiIsIm1yZXNvbHZlcndyaXRlIiwiaW1wb3J0dG9rZW5zIiwic2V0cGluIiwiZW5yb2xsUFVTSCIsImVucm9sbEVNQUlMIiwiY29uZmlnZGVsZXRlIiwibXJlc29sdmVycmVhZCIsInRva2VucmVhbG1zIiwiY2Fjb25uZWN0b3J3cml0ZSIsInBlcmlvZGljdGFza193cml0ZSIsImVucm9sbFBBUEVSIiwidG9rZW5saXN0Iiwic2V0X2hzbV9wYXNzd29yZCIsInJlc29sdmVyZGVsZXRlIiwiYXVkaXRsb2dfZG93bmxvYWQiLCJyZXNldCIsInNtc2dhdGV3YXlfcmVhZCIsImVucm9sbFRPVFAiLCJwb2xpY3lyZWFkIiwiZW5yb2xsWVVCSUtFWSIsInJlc29sdmVycmVhZCIsIm1hbmFnZV9tYWNoaW5lX3Rva2VucyIsImFzc2lnbiIsImdldHJhbmRvbSIsImFkZHVzZXIiLCJmZXRjaF9hdXRoZW50aWNhdGlvbl9pdGVtcyIsImVucm9sbFBXIiwibWFjaGluZWxpc3QiLCJjbGllbnR0eXBlIiwiZW5yb2xsQ0VSVElGSUNBVEUiLCJjb25maWd3cml0ZSIsImVucm9sbFNQQVNTIiwibWFuYWdlc3Vic2NyaXB0aW9uIiwiZW5yb2xsREFQTFVHIiwiZ2V0Y2hhbGxlbmdlcyIsInVzZXJsaXN0IiwiZW5yb2xsUkVHSVNUUkFUSU9OIiwiZW5yb2xsUkFESVVTIiwicmFkaXVzc2VydmVyX3dyaXRlIiwiZGVsZXRlIiwic3RhdGlzdGljc19yZWFkIiwidXBkYXRldXNlciIsImVucm9sbFRJUVIiLCJzbXRwc2VydmVyX3JlYWQiLCJlbnJvbGxwaW4iLCJ1bmFzc2lnbiIsImNhY29ubmVjdG9yZGVsZXRlIiwiZW5yb2xsTU9UUCIsImdldHNlcmlhbCIsInJlc3luYyIsImVucm9sbDRFWUVTIiwiZW5yb2xsT0NSQSIsInNtc2dhdGV3YXlfd3JpdGUiLCJlbnJvbGxIT1RQIiwiZXZlbnRoYW5kbGluZ19yZWFkIiwiZXZlbnRoYW5kbGluZ193cml0ZSIsInByaXZhY3lpZGVhc2VydmVyX3dyaXRlIiwibXJlc29sdmVyZGVsZXRlIiwicHJpdmFjeWlkZWFzZXJ2ZXJfcmVhZCIsImVucm9sbFUyRiIsInJldm9rZSIsInBlcmlvZGljdGFza19yZWFkIiwicG9saWN5d3JpdGUiLCJkaXNhYmxlIiwiZGVsZXRldXNlciIsImVucm9sbFNNUyIsInNldHRva2VuaW5mbyJdfQ.bgxeEFPcwTY9V8jxLHtDQPGlmfxewc7HSV29Hutd3H8";
         String img = "data:image/png;base64,iVBdgfgsdfgRK5CYII=";
 
         mockServer.when(HttpRequest.request().withPath(PIConstants.ENDPOINT_AUTH).withMethod("POST").withBody(""))
@@ -80,6 +79,9 @@ public void testSuccess()
 
         assertNotNull(rolloutInfo.serial);
         assertTrue(rolloutInfo.rolloutState.isEmpty());
+
+        rolloutInfo = privacyIDEA.tokenInit("games", "hotp", "12345");
+        assertEquals(img, rolloutInfo.googleurl.img);
     }
 
     @Test
@@ -90,9 +92,18 @@ public void testNoServiceAccount()
                                  .logger(new PILogImplementation())
                                  .build();
 
+        ArrayList<String> endpoints = new ArrayList<>();
+        endpoints.add("token/init");
+        endpoints.add("token/rollout");
+        privacyIDEA.logExcludedEndpoints(endpoints);
+
         RolloutInfo rolloutInfo = privacyIDEA.tokenRollout("games", "hotp");
 
         assertNull(rolloutInfo);
+
+        rolloutInfo = privacyIDEA.tokenInit("games", "hotp", "12345");
+
+        assertNull(rolloutInfo);
     }
 
     @Test