refactor: use AssertJ for testing

Majority of our test cases use AssertJ
This commit is contained in:
Nanne Baars 2023-12-06 08:51:27 +01:00 committed by Nanne Baars
parent ac7a9c7863
commit 87edc7d1db
6 changed files with 18 additions and 26 deletions

View File

@ -23,8 +23,7 @@
package org.owasp.webgoat.lessons.challenges.challenge7; package org.owasp.webgoat.lessons.challenges.challenge7;
import static org.junit.jupiter.api.Assertions.assertEquals; import static org.assertj.core.api.Assertions.assertThat;
import java.util.stream.Stream; import java.util.stream.Stream;
import org.junit.jupiter.api.DisplayName; import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.params.ParameterizedTest; import org.junit.jupiter.params.ParameterizedTest;
@ -37,7 +36,7 @@ public class MD5Test {
@DisplayName("MD5 test") @DisplayName("MD5 test")
@MethodSource("providedForMD5Values") @MethodSource("providedForMD5Values")
void testMD5(String in, String out) { void testMD5(String in, String out) {
assertEquals(MD5.getHashString(in.getBytes()), out); assertThat(out).isEqualTo(MD5.getHashString(in.getBytes()));
} }
private static Stream<Arguments> providedForMD5Values() { private static Stream<Arguments> providedForMD5Values() {

View File

@ -1,7 +1,7 @@
package org.owasp.webgoat.lessons.cryptography; package org.owasp.webgoat.lessons.cryptography;
import static org.junit.jupiter.api.Assertions.assertTrue; import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.jupiter.api.Assertions.fail; import static org.assertj.core.api.Assertions.fail;
import java.security.KeyPair; import java.security.KeyPair;
import java.security.PrivateKey; import java.security.PrivateKey;
@ -23,11 +23,9 @@ public class CryptoUtilTest {
String modulus = DatatypeConverter.printHexBinary(rsaPubKey.getModulus().toByteArray()); String modulus = DatatypeConverter.printHexBinary(rsaPubKey.getModulus().toByteArray());
String signature = CryptoUtil.signMessage(modulus, privateKey); String signature = CryptoUtil.signMessage(modulus, privateKey);
log.debug("public exponent {}", rsaPubKey.getPublicExponent()); log.debug("public exponent {}", rsaPubKey.getPublicExponent());
assertTrue(CryptoUtil.verifyAssignment(modulus, signature, keyPair.getPublic())); assertThat(CryptoUtil.verifyAssignment(modulus, signature, keyPair.getPublic())).isTrue();
} catch (Exception e) { } catch (Exception e) {
log.error("signing failed", e); fail("Signing failed");
;
fail();
} }
} }
} }

View File

@ -22,11 +22,9 @@
package org.owasp.webgoat.lessons.spoofcookie.encoders; package org.owasp.webgoat.lessons.spoofcookie.encoders;
import static org.assertj.core.api.Assertions.assertThat;
import static org.hamcrest.CoreMatchers.is; import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.MatcherAssert.assertThat; import static org.hamcrest.MatcherAssert.assertThat;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertTrue;
import java.util.stream.Stream; import java.util.stream.Stream;
import org.junit.jupiter.api.DisplayName; import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
@ -48,7 +46,7 @@ class EncDecTest {
void testEncode(String decoded, String encoded) { void testEncode(String decoded, String encoded) {
String result = EncDec.encode(decoded); String result = EncDec.encode(decoded);
assertTrue(result.endsWith(encoded)); assertThat(result.endsWith(encoded)).isTrue();
} }
@ParameterizedTest @ParameterizedTest
@ -63,13 +61,13 @@ class EncDecTest {
@Test @Test
@DisplayName("null encode test") @DisplayName("null encode test")
void testNullEncode() { void testNullEncode() {
assertNull(EncDec.encode(null)); assertThat(EncDec.encode(null)).isNull();
} }
@Test @Test
@DisplayName("null decode test") @DisplayName("null decode test")
void testNullDecode() { void testNullDecode() {
assertNull(EncDec.decode(null)); assertThat(EncDec.decode(null)).isNull();
} }
private static Stream<Arguments> providedForEncValues() { private static Stream<Arguments> providedForEncValues() {

View File

@ -22,10 +22,8 @@
package org.owasp.webgoat.lessons.vulnerablecomponents; package org.owasp.webgoat.lessons.vulnerablecomponents;
import static org.junit.jupiter.api.Assertions.assertNotNull; import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.jupiter.api.Assertions.assertThrows; import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
import com.thoughtworks.xstream.XStream; import com.thoughtworks.xstream.XStream;
import com.thoughtworks.xstream.io.StreamException; import com.thoughtworks.xstream.io.StreamException;
import org.junit.jupiter.api.Disabled; import org.junit.jupiter.api.Disabled;
@ -53,7 +51,7 @@ public class VulnerableComponentsLessonTest {
xstream.setClassLoader(Contact.class.getClassLoader()); xstream.setClassLoader(Contact.class.getClassLoader());
xstream.alias("contact", ContactImpl.class); xstream.alias("contact", ContactImpl.class);
xstream.ignoreUnknownElements(); xstream.ignoreUnknownElements();
assertNotNull(xstream.fromXML(contact)); assertThat(xstream.fromXML(contact)).isNotNull();
} }
@Test @Test
@ -67,7 +65,7 @@ public class VulnerableComponentsLessonTest {
assertThrows( assertThrows(
RuntimeException.class, RuntimeException.class,
() -> ((Contact) xstream.fromXML(strangeContact)).getFirstName()); () -> ((Contact) xstream.fromXML(strangeContact)).getFirstName());
assertTrue(e.getCause().getMessage().contains("calc.exe")); assertThat(e.getCause().getMessage().contains("calc.exe")).isTrue();
} }
@Test @Test
@ -79,6 +77,6 @@ public class VulnerableComponentsLessonTest {
Exception e = Exception e =
assertThrows( assertThrows(
StreamException.class, () -> ((Contact) xstream.fromXML("bullssjfs")).getFirstName()); StreamException.class, () -> ((Contact) xstream.fromXML("bullssjfs")).getFirstName());
assertTrue(e.getCause().getMessage().contains("START_DOCUMENT")); assertThat(e.getCause().getMessage().contains("START_DOCUMENT")).isTrue();
} }
} }

View File

@ -22,8 +22,7 @@
package org.owasp.webgoat.webwolf.mailbox; package org.owasp.webgoat.webwolf.mailbox;
import static org.junit.jupiter.api.Assertions.assertEquals; import static org.assertj.core.api.Assertions.assertThat;
import java.time.LocalDateTime; import java.time.LocalDateTime;
import java.util.List; import java.util.List;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
@ -58,6 +57,6 @@ public class MailboxRepositoryTest {
List<Email> emails = mailboxRepository.findByRecipientOrderByTimeDesc("someone@webwolf.org"); List<Email> emails = mailboxRepository.findByRecipientOrderByTimeDesc("someone@webwolf.org");
assertEquals(emails.size(), 1); assertThat(emails.size()).isEqualTo(1);
} }
} }

View File

@ -22,7 +22,7 @@
package org.owasp.webgoat.webwolf.user; package org.owasp.webgoat.webwolf.user;
import static org.junit.jupiter.api.Assertions.assertThrows; import static org.assertj.core.api.AssertionsForClassTypes.assertThatExceptionOfType;
import static org.mockito.ArgumentMatchers.any; import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.times; import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify; import static org.mockito.Mockito.verify;
@ -62,7 +62,7 @@ public class UserServiceTest {
when(mockUserRepository.findByUsername(username)).thenReturn(null); when(mockUserRepository.findByUsername(username)).thenReturn(null);
assertThrows(UsernameNotFoundException.class, () -> sut.loadUserByUsername(username)); assertThatExceptionOfType(UsernameNotFoundException.class).isThrownBy(() -> sut.loadUserByUsername(username));
} }
@Test @Test