From 6184a05c15e1de8803b3443f31093bf5529bdace Mon Sep 17 00:00:00 2001 From: Nanne Baars Date: Tue, 2 May 2017 23:37:37 +0200 Subject: [PATCH] Fix challenge 1 testcase --- .../owasp/webgoat/plugin/challenge1/Assignment1Test.java | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/webgoat-lessons/challenge/src/test/java/org/owasp/webgoat/plugin/challenge1/Assignment1Test.java b/webgoat-lessons/challenge/src/test/java/org/owasp/webgoat/plugin/challenge1/Assignment1Test.java index 5e123a7b2..e3f301bfb 100644 --- a/webgoat-lessons/challenge/src/test/java/org/owasp/webgoat/plugin/challenge1/Assignment1Test.java +++ b/webgoat-lessons/challenge/src/test/java/org/owasp/webgoat/plugin/challenge1/Assignment1Test.java @@ -11,6 +11,8 @@ import org.owasp.webgoat.plugin.SolutionConstants; import org.springframework.test.web.servlet.MockMvc; import org.springframework.test.web.servlet.request.MockMvcRequestBuilders; +import java.net.InetAddress; + import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath; import static org.springframework.test.web.servlet.setup.MockMvcBuilders.standaloneSetup; @@ -33,8 +35,10 @@ public class Assignment1Test extends AssignmentEndpointTest { @Test public void success() throws Exception { + InetAddress addr = InetAddress.getLocalHost(); + String host = addr.getHostAddress(); mockMvc.perform(MockMvcRequestBuilders.post("/challenge/1") - .header("X-Forwarded-For", "127.0.1.1") + .header("X-Forwarded-For", host) .param("username", "admin") .param("password", SolutionConstants.PASSWORD)) .andExpect(jsonPath("$.feedback", CoreMatchers.containsString("flag: " + Flag.FLAGS.get(1))))