2 files changed, 38 insertions, 21 deletions

diff --git a/test/create-context b/test/create-internet-context
index 6777ba79..3d548d02 100755
--- a/test/create-context
+++ b/test/create-internet-context
@@ -18,19 +18,30 @@ for path, properties in modems:
 					'org.ofono.ConnectionManager')
 
 	contexts = connman.GetContexts()
+	path = "";
 
-	if (len(contexts) == 0):
+	for i, properties in contexts:
+		if properties["Type"] == "internet":
+			path = i
+			break
+
+	if path == "":
 		path = connman.AddContext("internet")
+		print "Created new context %s" % (path)
 	else:
-		path = contexts[0][0]
+		print "Found context %s" % (path)
 
 	context = dbus.Interface(bus.get_object('org.ofono', path),
 					'org.ofono.ConnectionContext')
 
-	try:
+	if len(sys.argv) > 1:
 		context.SetProperty("AccessPointName", sys.argv[1])
-	except IndexError:
-		print "Usage: %s <apn_name>" % sys.argv[0]
-		exit(1)
+		print "Setting APN to %s" % (sys.argv[1])
+
+	if len(sys.argv) > 2:
+		context.SetProperty("Username", sys.argv[2])
+		print "Setting username to %s" % (sys.argv[2])
 
-	print "Setting APN of %s to %s" % (path, sys.argv[1])
+	if len(sys.argv) > 3:
+		context.SetProperty("Password", sys.argv[3])
+		print "Setting password to %s" % (sys.argv[3])
diff --git a/test/set-context b/test/create-mms-context
index 3d15764a..861ca7cb 100755
--- a/test/set-context
+++ b/test/create-mms-context
@@ -18,24 +18,30 @@ for path, properties in modems:
 					'org.ofono.ConnectionManager')
 
 	contexts = connman.GetContexts()
+	path = "";
 
-	if len(contexts) < 1:
-		print "No context available"
-		exit(1)
+	for i, properties in contexts:
+		if properties["Type"] == "mms":
+			path = i
+			break
+
+	if path == "":
+		path = connman.AddContext("mms")
+		print "Created new context %s" % (path)
 	else:
-		path = contexts[0][0]
+		print "Found context %s" % (path)
 
 	context = dbus.Interface(bus.get_object('org.ofono', path),
 					'org.ofono.ConnectionContext')
 
-	try:
+	if len(sys.argv) > 1:
 		context.SetProperty("AccessPointName", sys.argv[1])
-		if len(sys.argv) > 2:
-			context.SetProperty("Username", sys.argv[2])
-		if len(sys.argv) > 3:
-			context.SetProperty("Password", sys.argv[3])
-	except IndexError:
-		print "Usage: %s <apn_name> [username] [password]" % sys.argv[0]
-		exit(1)
-
-	print "Setting APN of %s to %s" % (path, sys.argv[1])
+		print "Setting APN to %s" % (sys.argv[1])
+
+	if len(sys.argv) > 2:
+		context.SetProperty("Username", sys.argv[2])
+		print "Setting username to %s" % (sys.argv[2])
+
+	if len(sys.argv) > 3:
+		context.SetProperty("Password", sys.argv[3])
+		print "Setting password to %s" % (sys.argv[3])