Write a session for ui tests and use it for test-icons

1 files changed, 13 insertions, 7 deletions

diff --git a/tests/test-icons.py b/tests/test-icons.py
index b0d6485..3a2b47d 100755
--- a/tests/test-icons.py
+++ b/tests/test-icons.py
@@ -1,10 +1,16 @@
 #!/usr/bin/python
-import random
-
 import pygtk
 pygtk.require('2.0')
-import gtk
 
+from sugar.session.UITestSession import UITestSession
+
+session = UITestSession()
+session.start()
+
+import sys
+import random
+
+import gtk
 import goocanvas
 
 from sugar.canvas import IconColor
@@ -22,19 +28,19 @@ window.add(canvas)
 canvas_model = goocanvas.CanvasModelSimple()
 root = canvas_model.get_root_item()
 
-item = goocanvas.Rect(x=0, y=0, width=800, height=600,
+item = goocanvas.Rect(x=0, y=0, width=1200, height=900,
                       line_width=0.0, fill_color="#4f4f4f")
 root.add_child(item)
 
 icon_names = [ 'stock-buddy', 'activity-groupchat', 'activity-web']
 
 k = 0
-while k < 11:
+while k < 17:
 	i = 0
-	while i < 15:
+	while i < 23:
 		color = IconColor.IconColor()
 		icon_name_n = int(random.random() * len(icon_names))
-		icon = IconItem(x=i * 50 + 10, y=k * 50 + 10,
+		icon = IconItem(x=i * 50 + 20, y=k * 50 + 10,
 						size=46, color=color,
 						icon_name=icon_names[icon_name_n])
 		root.add_child(icon)