Add support for multiple response USSD transactions in test script

diff --git a/test/initiate-ussd b/test/initiate-ussd
index ab0a8c4..198b427 100755 (executable)
--- 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)