Final changes for splitting SQL WebGoat and lessons

webwolf/src/main/java/org/owasp/webwolf/MvcConfiguration.java

@@ -64,12 +64,4 @@ public class MvcConfiguration implements WebMvcConfigurer {
             file.mkdirs();
         }
     }
-
-    @Bean
-    @Primary
-    public DataSource dataSource(@Value("${spring.datasource.url}") String url) {
-        DriverManagerDataSource driverManagerDataSource = new DriverManagerDataSource(url);
-        driverManagerDataSource.setDriverClassName("org.hsqldb.jdbc.JDBCDriver");
-        return driverManagerDataSource;
-    }
 }

webwolf/src/main/java/org/owasp/webwolf/WebWolf.java

@@ -23,10 +23,14 @@
 package org.owasp.webwolf;
 
 import org.owasp.webwolf.requests.WebWolfTraceRepository;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.actuate.trace.http.HttpTraceRepository;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
 import org.springframework.context.annotation.Bean;
+import org.springframework.jdbc.datasource.DriverManagerDataSource;
+
+import javax.sql.DataSource;
 
 @SpringBootApplication
 public class WebWolf {
@@ -40,4 +44,11 @@ public class WebWolf {
         System.setProperty("spring.config.name", "application-webwolf");
         SpringApplication.run(WebWolf.class, args);
     }
+
+    @Bean
+    public DataSource dataSource(@Value("${spring.datasource.url}") String url, @Value("${spring.datasource.driver-class-name}") String driverClassName) {
+        DriverManagerDataSource driverManagerDataSource = new DriverManagerDataSource(url);
+        driverManagerDataSource.setDriverClassName(driverClassName);
+        return driverManagerDataSource;
+    }
 }

webwolf/src/main/resources/application-webwolf.properties

@@ -8,6 +8,7 @@ server.servlet.session.cookie.name=WEBWOLFSESSION
 server.servlet.session.timeout=6000
 
 spring.datasource.url=jdbc:hsqldb:hsql://${WEBGOAT_HOST:127.0.0.1}:${WEBGOAT_HSQLPORT:9001}/webgoat
+spring.datasource.driver-class-name=org.hsqldb.jdbc.JDBCDriver
 spring.jpa.properties.hibernate.default_schema=CONTAINER
 spring.jpa.properties.hibernate.dialect=org.hibernate.dialect.HSQLDialect
 spring.jpa.hibernate.ddl-auto=update

webwolf/src/test/java/org/owasp/webwolf/mailbox/MailboxControllerTest.java

@@ -35,6 +35,7 @@ import org.springframework.boot.test.autoconfigure.web.servlet.WebMvcTest;
 import org.springframework.boot.test.mock.mockito.MockBean;
 import org.springframework.http.MediaType;
 import org.springframework.security.test.context.support.WithMockUser;
+import org.springframework.test.context.ActiveProfiles;
 import org.springframework.test.context.junit4.SpringRunner;
 import org.springframework.test.web.servlet.MockMvc;
 
@@ -49,6 +50,7 @@ import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.
 
 @RunWith(SpringRunner.class)
 @WebMvcTest(MailboxController.class)
+@ActiveProfiles({"test", "webwolf"})
 public class MailboxControllerTest {
 
     @Autowired