1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
/*
* SonarQube
* Copyright (C) 2009-2025 SonarSource SA
* mailto:info AT sonarsource DOT com
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 3 of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program; if not, write to the Free Software Foundation,
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
package org.sonar.auth.saml;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Base64;
import java.util.HashMap;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.junit.jupiter.MockitoExtension;
import org.sonar.api.server.http.HttpRequest;
import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
@ExtendWith(MockitoExtension.class)
public class SamlAuthStatusPageGeneratorTest {
@InjectMocks
private SamlAuthStatusPageGenerator samlAuthStatusPageGenerator;
@Test
public void getSamlAuthStatusHtml_whenCalled_shouldGeneratePageWithData() {
SamlAuthenticationStatus samlAuthenticationStatus = mock(SamlAuthenticationStatus.class);
HttpRequest request = mock(HttpRequest.class);
when(samlAuthenticationStatus.getStatus()).thenReturn("success");
when(samlAuthenticationStatus.getErrors()).thenReturn(new ArrayList<>());
when(samlAuthenticationStatus.getWarnings()).thenReturn(new ArrayList<>());
when(samlAuthenticationStatus.getAvailableAttributes()).thenReturn(new HashMap<>());
when(samlAuthenticationStatus.getMappedAttributes()).thenReturn(new HashMap<>());
when(samlAuthenticationStatus.isEncryptionEnabled()).thenReturn(false);
when(samlAuthenticationStatus.isSignatureEnabled()).thenReturn(false);
when(request.getContextPath()).thenReturn("context");
String decodedDataResponse = getDecodedDataResponse(samlAuthStatusPageGenerator.getSamlAuthStatusHtml(request, samlAuthenticationStatus));
assertThat(decodedDataResponse).contains(
"\"encryptionEnabled\":false",
"\"signatureEnabled\":false",
"\"errors\":[]",
"\"warnings\":[]",
"\"status\":\"success\"",
"\"availableAttributes\":{}",
"\"mappedAttributes\":{}");
}
private static String getDecodedDataResponse(String completeHtmlTemplate) {
String pattern = "data-response=\"([^\"]+)\"";
Pattern regex = Pattern.compile(pattern);
Matcher matcher = regex.matcher(completeHtmlTemplate);
if (matcher.find()) {
String dataResponseValue = matcher.group(1);
byte[] decoded = Base64.getDecoder().decode(dataResponseValue);
return new String(decoded, StandardCharsets.UTF_8);
}
return "";
}
}
|
