Merge pull request #33 from christ66/JENKINS-24671

[JENKINS-24671] Add username and referrer to slow-requests
jenkinsci · May 18, 2015 · 6af72162ceba9216abf535aed90d0d3de17684aa · 6af7216
2 parents 040f5a1 + 7cc1eaf
commit 6af72162ceba9216abf535aed90d0d3de17684aa
Unified Split

Showing with 113 additions and 0 deletions.

+17 −0 src/main/java/com/cloudbees/jenkins/support/slowrequest/InflightRequest.java

+96 −0 src/test/java/com/cloudbees/jenkins/support/slowrequest/InflightRequestTest.java
diff --git a/src/main/java/com/cloudbees/jenkins/support/slowrequest/InflightRequest.java b/src/main/java/com/cloudbees/jenkins/support/slowrequest/InflightRequest.java
@@ -1,5 +1,9 @@
 package com.cloudbees.jenkins.support.slowrequest;
+  
+import com.google.common.net.HttpHeaders;
+  
+import jenkins.model.Jenkins;
+  
+import org.kohsuke.stapler.StaplerRequest;
+  
+
 import javax.servlet.http.HttpServletRequest;
 import java.io.File;
 import java.io.PrintWriter;
@@ -36,14 +40,27 @@
     */
    File record;
+  
+    /**
+  
+     * Username of user who made the http call.
+  
+     */
+  
+    final String userName;
+  
+
+  
+    /**
+  
+     * Referer link to track any redirect urls.
+  
+     */
+  
+    final String referer;
    InflightRequest(HttpServletRequest req) {
        String query = req.getQueryString();
        url = req.getRequestURL() + (query == null ? "" : "?" + query);
        startTime = System.currentTimeMillis();
+  
+        userName = Jenkins.getAuthentication().getName();
+  
+        referer = req.getHeader("Referer");
    }
    void writeHeader(PrintWriter w) {
+  
+        w.println("Username: " + userName);
+  
+        w.println("Referer: " + referer);
        w.println("Date: " + new Date());
        w.println("URL: " + url);
        w.println();

diff --git a/src/test/java/com/cloudbees/jenkins/support/slowrequest/InflightRequestTest.java b/src/test/java/com/cloudbees/jenkins/support/slowrequest/InflightRequestTest.java
@@ -0,0 +1,96 @@
+  
+package com.cloudbees.jenkins.support.slowrequest;
+  
+
+  
+import com.gargoylesoftware.htmlunit.*;
+  
+import com.gargoylesoftware.htmlunit.html.HtmlPage;
+  
+import com.gargoylesoftware.htmlunit.util.WebConnectionWrapper;
+  
+import hudson.model.RootAction;
+  
+import hudson.util.HttpResponses;
+  
+import org.apache.commons.io.FileUtils;
+  
+import org.apache.http.HttpRequest;
+  
+import org.junit.BeforeClass;
+  
+import org.junit.Rule;
+  
+import org.junit.Test;
+  
+import org.jvnet.hudson.test.Bug;
+  
+import org.jvnet.hudson.test.JenkinsRule;
+  
+import org.jvnet.hudson.test.TestExtension;
+  
+import org.kohsuke.stapler.HttpResponse;
+  
+import org.kohsuke.stapler.Stapler;
+  
+import org.kohsuke.stapler.StaplerRequest;
+  
+import org.kohsuke.stapler.StaplerResponse;
+  
+
+  
+import javax.servlet.ServletException;
+  
+import javax.servlet.http.HttpServletRequest;
+  
+import javax.servlet.http.HttpServletRequestWrapper;
+  
+import java.io.File;
+  
+import java.io.IOException;
+  
+import java.net.URL;
+  
+import java.sql.ClientInfoStatus;
+  
+import java.util.ResourceBundle;
+  
+
+  
+import static org.junit.Assert.assertEquals;
+  
+import static org.junit.Assert.assertNotNull;
+  
+
+  
+/**
+  
+ * @author schristou88
+  
+ */
+  
+public class InflightRequestTest {
+  
+    @Rule public JenkinsRule r = new JenkinsRule();
+  
+
+  
+    @Test
+  
+    @Bug(24671)
+  
+    public void verifyUsernameInflightRequest() throws Exception {
+  
+        r.jenkins.setSecurityRealm(r.createDummySecurityRealm());
+  
+        JenkinsRule.WebClient webClient = r.createWebClient().login("bob", "bob");
+  
+        MockSlowURLCall.seconds = 5;
+  
+        webClient.goTo("mockSlowURLCall/submit");
+  
+        InflightRequest request = MockSlowURLCall.request;
+  
+        assertNotNull(request);
+  
+        assertEquals("bob", request.userName);
+  
+        assertEquals(webClient.getContextPath() + "mockSlowURLCall/submit", request.url);
+  
+    }
+  
+
+  
+    @Test
+  
+    @Bug(24671)
+  
+    public void verifyRefererHeaderFromInflightRequest() throws Exception {
+  
+        JenkinsRule.WebClient webClient = r.createWebClient();
+  
+
+  
+        URL refererUrl = new URL(webClient.getContextPath() + "mockSlowURLCall/submitReferer");
+  
+
+  
+        ((HtmlPage) webClient.getPage(refererUrl)).getHtmlElementById("link").click();
+  
+
+  
+        InflightRequest request = MockSlowURLCall.request;
+  
+        assertEquals(refererUrl.toString(), request.referer);
+  
+    }
+  
+
+  
+    @TestExtension
+  
+    public static class MockSlowURLCall implements RootAction {
+  
+        public static int seconds;
+  
+        public static InflightRequest request;
+  
+
+  
+        public HttpResponse doSubmitReferer(StaplerRequest req, StaplerResponse rsp) throws ServletException, IOException {
+  
+            String literalHTML = "<html>" +
+  
+                                 "  <body>" +
+  
+                                 "    <a href='" + req.getContextPath() + "/mockSlowURLCall/submit"+ "' id='link'>link</a>" +
+  
+                                 "  </body>" +
+  
+                                 "</html>";
+  
+            return HttpResponses.html(literalHTML);
+  
+        }
+  
+
+  
+        public HttpResponse doSubmit(StaplerRequest req, StaplerResponse rsp) throws ServletException, IOException {
+  
+            request = new InflightRequest(req);
+  
+            return HttpResponses.redirectTo("..");
+  
+        }
+  
+
+  
+        public String getIconFileName() {
+  
+            return "";
+  
+        }
+  
+
+  
+        public String getDisplayName() {
+  
+            return "MockSlowURLCall";
+  
+        }
+  
+
+  
+        public String getUrlName() {
+  
+            return "mockSlowURLCall";
+  
+        }
+  
+    }
+  
+}