Added more tests

pywinauto/unittests/test_win32controls.py

@@ -27,6 +27,236 @@ from pywinauto.controls.win32_controls import *
 
 import unittest
 
+# following imports are not required for the tests
+# but are useful for debugging
+import time
+import pdb
+import pprint
+
+
+
+class ButtonTestCases(unittest.TestCase):
+    "Unit tests for the ComboBoxWrapper class"
+
+    def setUp(self):
+        """Start the application set some data and ensure the application
+        is in the state we want it."""
+
+        # start the application
+        from pywinauto.application import Application
+        self.app = Application()
+
+        self.app.start_("calc.exe")
+        self.calc = self.app.SciCalc
+        self.calc.MenuSelect("View->Scientific")
+
+    def tearDown(self):
+        "Close the application after tests"
+
+        self.calc.TypeKeys("%{F4}")
+
+
+    def testGetProperties(self):
+        "Test getting the properties for the control"
+        props  = self.calc._6.GetProperties()
+
+        self.assertEquals(
+            "Button", props['FriendlyClassName'])
+
+        self.assertEquals(
+            self.calc._6.Texts(), ['6'])
+
+        self.assertEquals(
+            self.calc._6.Texts(), props['Texts'])
+
+        for prop_name in props:
+            self.assertEquals(getattr(self.calc._6, prop_name)(), props[prop_name])
+
+    def testClick(self):
+        "Test getting the properties for the control"
+        self.calc._6.Click()
+        self.calc._5.Click()
+        self.calc['+'].Click()
+        self.calc._4.Click()
+        self.calc._3.Click()
+        self.calc['='].Click()
+        self.assertEquals(self.calc.Edit.Texts()[1], "108. ")
+
+
+    def testFriendlyClass(self):
+        self.assertEquals(self.calc._8.FriendlyClassName(), "Button")
+        self.assertEquals(self.calc.Dec.FriendlyClassName(), "RadioButton")
+        self.assertEquals(self.calc.Hyp.FriendlyClassName(), "CheckBox")
+
+        from pywinauto.findwindows import find_windows
+        from pywinauto.controls import WrapHandle
+
+        children = self.calc.Children()
+        no_text_buttons = [c for c in children if not c.Text() and c.Class() == "Button"]
+
+        first_group = no_text_buttons[0]
+
+        self.assertEquals(first_group.FriendlyClassName(), "GroupBox")
+
+
+    def testGetCheckState_unchecked(self):
+        "unchecked"
+        self.assertEquals(self.calc.Inv.GetCheckState(), 0)
+
+
+    def testGetCheckState_checked(self):
+        "checked"
+        self.calc.Inv.Check()
+        self.assertEquals(self.calc.Inv.GetCheckState(), 1)
+
+
+#    def testGetCheckState_indeterminate(self):
+#        "indeterminate"
+#        self.calc.Inv.SetCheckIndeterminate()
+#        self.assertEquals(self.calc.Inv.GetCheckState(), 0)
+
+    def testIsSelected(self):
+
+        self.assertEquals(self.calc.Hex.GetCheckState(), 0)
+
+        self.calc.Hex.Click()
+
+        self.assertEquals(self.calc.Hex.GetCheckState(), 1)
+
+
+
+
+
+class ComboBoxTestCases(unittest.TestCase):
+    "Unit tests for the ComboBoxWrapper class"
+
+    def setUp(self):
+        """Start the application set some data and ensure the application
+        is in the state we want it."""
+
+        # start the application
+        from pywinauto.application import Application
+        self.app = Application()
+
+        self.app.start_("Notepad.exe")
+
+
+        self.app.UntitledNotepad.MenuSelect("Format->Font")
+
+        self.ctrl = self.app.Font.ComboBox2.ctrl_()
+
+
+    def tearDown(self):
+        "Close the application after tests"
+
+        self.app.Font.Cancel.CloseClick()
+
+        # close the application
+        self.app.UntitledNotepad.MenuSelect("File->Exit")
+
+        if self.app.Notepad.No.Exists():
+            self.app.Notepad.No.Click()
+
+
+    def testGetProperties(self):
+        "Test getting the properties for the control"
+        props  = self.ctrl.GetProperties()
+
+        self.assertEquals(
+            "ComboBox", props['FriendlyClassName'])
+
+        self.assertEquals(
+            self.ctrl.Texts(), props['Texts'])
+
+        for prop_name in props:
+            self.assertEquals(getattr(self.ctrl, prop_name)(), props[prop_name])
+
+
+    def testItemCount(self):
+        "Test that ItemCount returns the correct number of items"
+        self.assertEquals(self.ctrl.ItemCount(), 4)
+
+
+    def testDroppedRect(self):
+        "Test that the dropped rect is correct"
+        rect = self.ctrl.DroppedRect()
+        #import pdb;pdb.set_trace()
+        self.assertEquals(rect.left, 0)
+        self.assertEquals(rect.top, 0)
+        self.assertEquals(rect.right, self.ctrl.ClientRect().right)
+        self.assertEquals(rect.bottom, self.ctrl.Rectangle().height() - 3)
+
+
+    def testSelectedIndex(self):
+        "That teh control returns the correct index for the selected item"
+        self.ctrl.Select(2)
+        self.assertEquals(self.ctrl.SelectedIndex(), 2)
+        self.assertEquals(self.ctrl.Texts()[3], self.app.Font.Edit2.Texts()[1])
+
+
+    def testSelect_negative(self):
+        "Test that the Select method correctly handles negative indices"
+        self.ctrl.Select(-1)
+        self.assertEquals(self.ctrl.SelectedIndex(), 3)
+
+
+    def testSelect_toohigh(self):
+        "Test that the Select correctly raises if the item is too high"
+        self.assertRaises(IndexError, self.ctrl.Select, 21)
+
+
+#    def testItemData(self):
+#        pass
+#
+#    def testTexts(self):
+#        pass
+#
+
+
+
+
+class ListBoxTestCases(unittest.TestCase):
+    "Unit tests for the TreeViewWrapper class"
+
+    def setUp(self):
+        """Start the application set some data and ensure the application
+        is in the state we want it."""
+
+        # start the application
+        from pywinauto.application import Application
+        self.app = Application()
+
+        self.app.start_(r"c:\Program Files\Windows NT\Accessories\wordpad.exe")
+        self.app.DocumentWordPad.MenuSelect("Insert->Date and time...")
+
+        self.dlg = self.app.DateAndTime
+        self.ctrl = self.dlg.ListBox.ctrl_()
+
+    def tearDown(self):
+        "Close the application after tests"
+
+        self.dlg.Cancel.Click()
+
+        # close the application
+        self.app.DocumentWordPad.MenuSelect("File->Exit")
+
+    def testGetProperties(self):
+        "Test getting the properties for the control"
+        props  = self.ctrl.GetProperties()
+
+        self.assertEquals(
+            "ListBox", props['FriendlyClassName'])
+
+        self.assertEquals(
+            self.ctrl.Texts(), props['Texts'])
+
+        for prop_name in props:
+            self.assertEquals(getattr(self.ctrl, prop_name)(), props[prop_name])
+
+
+
+
+
 class EditTestCases(unittest.TestCase):
     "Unit tests for the TreeViewWrapper class"
 
@@ -141,46 +371,6 @@ class EditTestCases(unittest.TestCase):
 
 
 
-class ListBoxTestCases(unittest.TestCase):
-    "Unit tests for the TreeViewWrapper class"
-
-    def setUp(self):
-        """Start the application set some data and ensure the application
-        is in the state we want it."""
-
-        # start the application
-        from pywinauto.application import Application
-        self.app = Application()
-
-        self.app.start_(r"c:\Program Files\Windows NT\Accessories\wordpad.exe")
-        self.app.DocumentWordPad.MenuSelect("Insert->Date and time...")
-
-        self.dlg = self.app.DateAndTime
-        self.ctrl = self.dlg.ListBox.ctrl_()
-
-    def tearDown(self):
-        "Close the application after tests"
-
-        self.dlg.Cancel.Click()
-
-        # close the application
-        self.app.DocumentWordPad.MenuSelect("File->Exit")
-
-    def testGetProperties(self):
-        "Test getting the properties for the control"
-        props  = self.ctrl.GetProperties()
-
-        self.assertEquals(
-            "ListBox", props['FriendlyClassName'])
-
-        self.assertEquals(
-            self.ctrl.Texts(), props['Texts'])
-
-        for prop_name in props:
-            self.assertEquals(getattr(self.ctrl, prop_name)(), props[prop_name])
-
-
-