From 867143bc0bd78958de411ddc5105441d8b54bc24 Mon Sep 17 00:00:00 2001
From: Ren RenJuan <jdaugherty@noojuan.thoughtcrime.biz>
Date: Fri, 10 Jan 2014 20:19:20 +0000
Subject: [PATCH] *

---
 ACTK1_0/session/SessionTest.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/ACTK1_0/session/SessionTest.cpp b/ACTK1_0/session/SessionTest.cpp
index 6611cda..eaef065 100644
--- a/ACTK1_0/session/SessionTest.cpp
+++ b/ACTK1_0/session/SessionTest.cpp
@@ -15,10 +15,14 @@ using namespace std;
 static std::string TEST_SE =
 	"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?><epp xmlns=\"urn:ietf:params:xml:ns:epp-1.0\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xsi:schemaLocation=\"urn:ietf:params:xml:ns:epp-1.0 epp-1.0.xsd\"><hello/></epp>";
 
+extern static int tkScenario;
 
 void doWork()
 {
+	if (!tkScenario)
 	init("etc/toolkit2.conf");
+	else
+		scenario(tkScenario);
 	{
 		TestEnvironment props;
 		auto_ptr<Session> sess(SessionFactory::newInstance(&props));
@@ -75,6 +79,8 @@ void doWork()
 
 int main(int argc, char* argv[])
 {
+	if (argc > 1)
+		tkScenario = atoi(argv[1]);
 	TEST_run(doWork);
 	return TEST_errorCount();
 }