From a1375b671819f0302be98e89ec90a9e084750480 Mon Sep 17 00:00:00 2001
From: Nanne Baars <nbaars@xebia.com>
Date: Thu, 15 Sep 2016 23:17:05 +0200
Subject: [PATCH] Fixed failing testcase

---
 .../webgoat/session/LabelDebuggerTest.java      | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/webgoat-container/src/test/java/org/owasp/webgoat/session/LabelDebuggerTest.java b/webgoat-container/src/test/java/org/owasp/webgoat/session/LabelDebuggerTest.java
index 48daf1431..b9bc79aae 100644
--- a/webgoat-container/src/test/java/org/owasp/webgoat/session/LabelDebuggerTest.java
+++ b/webgoat-container/src/test/java/org/owasp/webgoat/session/LabelDebuggerTest.java
@@ -1,42 +1,45 @@
 package org.owasp.webgoat.session;
 
-import org.junit.Assert;
 import org.junit.Test;
 
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
 public class LabelDebuggerTest {
 
     @Test
     public void testSetEnabledTrue() throws Exception {
         LabelDebugger ld = new LabelDebugger();
         ld.setEnabled(true);
-        Assert.assertTrue(ld.isEnabled());
+        assertTrue(ld.isEnabled());
     }
 
     @Test
     public void testSetEnabledFalse() throws Exception {
         LabelDebugger ld = new LabelDebugger();
         ld.setEnabled(false);
-        Assert.assertFalse(ld.isEnabled());
+        assertFalse(ld.isEnabled());
     }
 
-    @Test(expected = Exception.class)
-    public void testSetEnabledNullThrowsException() throws Exception {
+    @Test
+    public void testSetEnabledNullThrowsException() {
         LabelDebugger ld = new LabelDebugger();
         ld.setEnabled(true);
+        assertTrue(ld.isEnabled());
     }
 
     @Test
     public void testEnableIsTrue() {
         LabelDebugger ld = new LabelDebugger();
         ld.enable();
-        Assert.assertTrue(ld.isEnabled());
+        assertTrue(ld.isEnabled());
     }
 
     @Test
     public void testDisableIsFalse() {
         LabelDebugger ld = new LabelDebugger();
         ld.disable();
-        Assert.assertFalse(ld.isEnabled());
+        assertFalse(ld.isEnabled());
     }