Updated tests for CommandSet.

BellaTrix25336 · Sep 1, 2019 · 1b5da46 · 1b5da46
1 parent b9b957e
commit 1b5da46
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/src/core/tests/test_command.py b/src/core/tests/test_command.py
@@ -126,14 +126,17 @@ def _changed(self):
     def test_cmdset_init(self):
         test_widget = toga.Widget(factory=toga_dummy.factory)
         cs = toga.CommandSet(test_widget)
-        self.assertEqual(cs.widget, test_widget)
-        self.assertEqual(cs._values, set())
+        self.assertEqual(cs._commands, set())
         self.assertEqual(cs.on_change, None)
 
     def test_cmdset_add(self):
         self.changed = False
         test_widget = toga.Widget(factory=toga_dummy.factory)
-        cs = toga.CommandSet(test_widget, on_change=self._changed)
+        cs = toga.CommandSet(
+            factory=toga_dummy.factory,
+            widget=test_widget,
+            on_change=self._changed
+        )
         grp = toga.Group('Test group', order=10)
         cmd = toga.Command(
             lambda x: print('Hello World'),
@@ -147,11 +150,16 @@ def test_cmdset_add(self):
             factory=toga_dummy.factory
         )
         cs.add(cmd)
+
         self.assertTrue(self.changed)
+        self.assertIsNotNone(cmd._impl)
 
     def test_cmdset_iter(self):
         test_widget = toga.Widget(factory=toga_dummy.factory)
-        cs = toga.CommandSet(test_widget)
+        cs = toga.CommandSet(
+            factory=toga_dummy.factory,
+            widget=test_widget
+        )
         grp = toga.Group('Test group', order=10)
         cmd = toga.Command(
             lambda x: print('Hello World'),