From 46652566746c639942a0414f55eb099647f7eb85 Mon Sep 17 00:00:00 2001
From: "rogan.dawes" <rogan.dawes@4033779f-a91e-0410-96ef-6bf7bf53c507>
Date: Wed, 11 Jul 2007 12:54:15 +0000
Subject: [PATCH] Construct a message automatically when we complete a stage

git-svn-id: http://webgoat.googlecode.com/svn/trunk@168 4033779f-a91e-0410-96ef-6bf7bf53c507
---
 .../owasp/webgoat/lessons/RandomLessonAdapter.java    | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/ webgoat/main/project/JavaSource/org/owasp/webgoat/lessons/RandomLessonAdapter.java b/ webgoat/main/project/JavaSource/org/owasp/webgoat/lessons/RandomLessonAdapter.java
index 28592ce03..7bdf5f540 100755
--- a/ webgoat/main/project/JavaSource/org/owasp/webgoat/lessons/RandomLessonAdapter.java	
+++ b/ webgoat/main/project/JavaSource/org/owasp/webgoat/lessons/RandomLessonAdapter.java	
@@ -17,7 +17,16 @@ public abstract class RandomLessonAdapter extends LessonAdapter {
 	}
 	
 	public void setStageComplete(WebSession s, String stage) {
-		getLessonTracker(s).setStageComplete(stage, true);
+		RandomLessonTracker lt = getLessonTracker(s);
+		lt.setStageComplete(stage, true);
+		if (lt.getCompleted()) {
+			s.setMessage("Congratulations, you have completed this lesson");
+		} else {
+			String message = "You have completed stage " + stage + ".";
+			if (! stage.equals(lt.getStage()))
+				message = message + " Welcome to " + lt.getStage();
+			s.setMessage(message);
+		}
 	}
 	
 	@Override