refactor: use AssertJ for testing
Majority of our test cases use AssertJ
This commit is contained in:
parent
ac7a9c7863
commit
87edc7d1db
@ -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() {
|
||||||
|
@ -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();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -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() {
|
||||||
|
@ -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();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -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);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -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
|
||||||
|
Loading…
x
Reference in New Issue
Block a user