1
0
mirror of https://github.com/google/nomulus synced 2025-12-23 06:15:42 +00:00

Generate fake XSRF token in FakeConsoleApiParams for tests (#2421)

This commit is contained in:
gbrodman
2024-04-30 13:47:53 -04:00
committed by GitHub
parent b27218d799
commit d121f8f547

View File

@@ -15,12 +15,14 @@
package google.registry.testing;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import com.google.appengine.api.users.UserService;
import google.registry.request.auth.AuthResult;
import google.registry.request.auth.UserAuthInfo;
import google.registry.security.XsrfTokenManager;
import google.registry.ui.server.registrar.ConsoleApiParams;
import jakarta.servlet.http.Cookie;
import jakarta.servlet.http.HttpServletRequest;
import java.util.Optional;
import org.joda.time.DateTime;
@@ -36,11 +38,15 @@ public final class FakeConsoleApiParams {
new com.google.appengine.api.users.User(
"JohnDoe@theregistrar.com", "theregistrar.com"),
false)));
return ConsoleApiParams.create(
mock(HttpServletRequest.class),
new FakeResponse(),
authResult,
HttpServletRequest request = mock(HttpServletRequest.class);
XsrfTokenManager xsrfTokenManager =
new XsrfTokenManager(
new FakeClock(DateTime.parse("2020-02-02T01:23:45Z")), mock(UserService.class)));
new FakeClock(DateTime.parse("2020-02-02T01:23:45Z")), mock(UserService.class));
when(request.getCookies())
.thenReturn(
new Cookie[] {
new Cookie(XsrfTokenManager.X_CSRF_TOKEN, xsrfTokenManager.generateToken(""))
});
return ConsoleApiParams.create(request, new FakeResponse(), authResult, xsrfTokenManager);
}
}