Add support for multiple response USSD transactions in test script

1 files changed, 8 insertions, 15 deletions

diff --git a/test/initiate-ussd b/test/initiate-ussd
index ab0a8c46..198b4277 100755
--- a/test/initiate-ussd
+++ b/test/initiate-ussd
@@ -38,22 +38,15 @@ if state == "idle":
 
 print "State: %s" % (state)
 
-if state != "user-response":
-	ussd.Cancel()
-	sys.exit(0)
-
-response = raw_input("Enter response: ")
-
-result = ussd.Respond(response)
+while state == "user-response":
+	response = raw_input("Enter response: ")
 
-properties = ussd.GetProperties()
-state = properties["State"]
+	result = ussd.Respond(response, timeout=100)
 
-print result
+	properties = ussd.GetProperties()
+	state = properties["State"]
 
-if state == "idle":
-	sys.exit(0)
-
-print "State: %s" % (state)
+	print result
 
-ussd.Cancel()
+	if state != "idle":
+		print "State: %s" % (state)