From dc8914f4e1131c3d2e7cd1a7d1bbabae61473c0e Mon Sep 17 00:00:00 2001
From: "rogan.dawes" <rogan.dawes@4033779f-a91e-0410-96ef-6bf7bf53c507>
Date: Wed, 11 Jul 2007 12:53:00 +0000
Subject: [PATCH] Allow for simple restarting of a lesson.

git-svn-id: http://webgoat.googlecode.com/svn/trunk@163 4033779f-a91e-0410-96ef-6bf7bf53c507
---
 .../org/owasp/webgoat/session/RandomLessonTracker.java       | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/ webgoat/main/project/JavaSource/org/owasp/webgoat/session/RandomLessonTracker.java b/ webgoat/main/project/JavaSource/org/owasp/webgoat/session/RandomLessonTracker.java
index 84ac3b128..7306e503c 100755
--- a/ webgoat/main/project/JavaSource/org/owasp/webgoat/session/RandomLessonTracker.java	
+++ b/ webgoat/main/project/JavaSource/org/owasp/webgoat/session/RandomLessonTracker.java	
@@ -50,7 +50,10 @@ public class RandomLessonTracker extends LessonTracker {
 	
 	@Override
 	public void setCompleted(boolean complete) {
-		throw new UnsupportedOperationException("Use individual stage completion instead");
+		if (complete == true)
+			throw new UnsupportedOperationException("Use individual stage completion instead");
+		for (int i=0;i<stages.length; i++)
+			setStageComplete(stages[i], false);
 	}
 	
     protected void setProperties(Properties props, Screen screen) {