test: Handle DisplayText with async_callbacks option

1 files changed, 16 insertions, 3 deletions

diff --git a/test/test-stk-menu b/test/test-stk-menu
index f4553d12..215b0bc6 100755
--- a/test/test-stk-menu
+++ b/test/test-stk-menu
@@ -54,13 +54,21 @@ class StkAgent(dbus.service.Object):
 			return int(select);
 
 	@dbus.service.method("org.ofono.SimToolkitAgent",
-					in_signature="syb", out_signature="")
-	def DisplayText(self, title, icon, urgent):
+					in_signature="syb", out_signature="",
+					async_callbacks=("reply_func",
+							"error_func"))
+	def DisplayText(self, title, icon, urgent, reply_func, error_func):
 		print "DisplayText (%s)" % (title)
 		print "Icon: (%d)" % (icon)
 		print "Urgent: (%d)" % (urgent)
 		key = raw_input("Press return to clear ('t' terminates, "
-				"'b' goes back, 'n' busy):")
+						"'b' goes back, 'n' busy, "
+						"'w' return and wait):")
+
+		if key == 'w':
+			seconds = 60
+		else:
+			seconds = 0
 
 		if key == 'b':
 			raise GoBack("User wishes to go back")
@@ -69,6 +77,11 @@ class StkAgent(dbus.service.Object):
 		elif key == 'n':
 			raise Busy("User wishes to simulate busy screen")
 
+		if (seconds > 0):
+			print "Waiting for %d seconds" % (seconds)
+
+		gobject.timeout_add_seconds(seconds,  lambda: reply_func())
+
 	@dbus.service.method("org.ofono.SimToolkitAgent",
 				in_signature="sysyyb", out_signature="s")
 	def RequestInput(self, title, icon, default, min_chars, max_chars,