Move setup to top of tests

SasanLabs · Sep 14, 2023 · 9a75a1d · 9a75a1d
1 parent b2e8e05
commit 9a75a1d
Showing 1 changed file with 6 additions and 11 deletions.
diff --git a/...sasanlabs/service/vulnerability/sqlInjection/UnionBasedSQLInjectionVulnerabilityTest.java b/...sasanlabs/service/vulnerability/sqlInjection/UnionBasedSQLInjectionVulnerabilityTest.java
@@ -29,6 +29,12 @@ void setUp() throws IOException {
         doReturn(null)
                 .when(template)
                 .query(anyString(), (ResultSetExtractor<? extends Object>) any());
+        doReturn(null)
+                .when(template)
+                .query(
+                        anyString(),
+                        (PreparedStatementSetter) any(),
+                        (ResultSetExtractor<? extends Object>) any());
 
         unionBasedSQLInjectionVulnerability = new UnionBasedSQLInjectionVulnerability(template);
     }
@@ -77,17 +83,6 @@ void getCarInformationLevel3_ExpectParamEscaped() throws IOException {
 
     @Test
     void getCarInformationLevel4_ExpecParamEscaped() throws IOException {
-        // Setup
-        template = Mockito.spy(new JdbcTemplate());
-        doReturn(null)
-                .when(template)
-                .query(
-                        anyString(),
-                        (PreparedStatementSetter) any(),
-                        (ResultSetExtractor<? extends Object>) any());
-
-        unionBasedSQLInjectionVulnerability = new UnionBasedSQLInjectionVulnerability(template);
-
         // Act
         final Map<String, String> params =
                 Collections.singletonMap("id", "1' UNION SELECT * FROM cars; --");