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:
@@ -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);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user