From 6d5c55191f504357760267849c2524b3d8d64711 Mon Sep 17 00:00:00 2001
From: Vincent Vinet <vince.vinet@gmail.com>
Date: Wed, 25 Mar 2009 18:35:17 +0000
Subject: Split action tests into own file, and add some testing

---
(limited to 'src/sugar/tutorius/tests/actiontests.py')

diff --git a/src/sugar/tutorius/tests/actiontests.py b/src/sugar/tutorius/tests/actiontests.py
index 2a60a65..7c508a8 100644
--- a/src/sugar/tutorius/tests/actiontests.py
+++ b/src/sugar/tutorius/tests/actiontests.py
@@ -1,4 +1,6 @@
 # Copyright (C) 2009, Tutorius.org
+# Copyright (C) 2009, Michael Janelle-Montcalm <michael.jmontcalm@gmail.com>
+# Copyright (C) 2009, Vincent Vinet <vince.vinet@gmail.com>
 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -20,7 +22,83 @@ The behavior of the actions must be tested here.
 """
 
 import unittest
-from sugar.tutorius.actions import *
+import gtk
+
+from sugar.tutorius.actions import Action, OnceWrapper, DisableWidgetAction
+from sugar.tutorius.services import ObjectStore
+
+class CountAction(Action):
+    """
+    This action counts how many times it's do and undo methods get called
+    """
+    def __init__(self):
+        self.do_count = 0
+        self.undo_count = 0
+
+    def do(self):
+        self.do_count += 1
+
+    def undo(self):
+        self.undo_count += 1
+
+
+class BaseActionTests(unittest.TestCase):
+    def test_do_unimplemented(self):
+        act = Action()
+        try:
+            act.do()
+            assert False, "do() should trigger a NotImplemented"
+        except NotImplementedError:
+            assert True, "do() should trigger a NotImplemented"
+
+    def test_undo(self):
+        act = Action()
+        act.undo()
+        assert True, "undo() should never fail on the base action"
+
+
+class OnceWrapperTests(unittest.TestCase):
+    def test_onceaction_toggle(self):
+        """
+        Validate that the OnceWrapper wrapper works properly using the
+        CountAction
+        """
+        act = CountAction()
+        wrap = OnceWrapper(act)
+
+        assert act.do_count == 0, "do() should not have been called in __init__()"
+        assert act.undo_count == 0, "undo() should not have been called in __init__()"
+
+        wrap.undo()
+
+        assert act.undo_count == 0, "undo() should not be called if do() has not been called"
+
+        wrap.do()
+        assert act.do_count == 1, "do() should have been called once"
+
+        wrap.do()
+        assert act.do_count == 1, "do() should have been called only once"
+
+        wrap.undo()
+        assert act.undo_count == 1, "undo() should have been called once"
+
+        wrap.undo()
+        assert act.undo_count == 1, "undo() should have been called only once"
+
+class DisableWidgetActionTests(unittest.TestCase):
+    def test_disable(self):
+        btn = gtk.Button()
+        ObjectStore().activity = btn
+        btn.set_sensitive(True)
+        
+        assert btn.props.sensitive is True, "Callback should have been called"
+
+        act = DisableWidgetAction("0")
+        assert btn.props.sensitive is True, "Callback should have been called again"
+        act.do()
+        assert btn.props.sensitive is False, "Callback should not have been called again"
+        act.undo()
+        assert btn.props.sensitive is True, "Callback should have been called again"
 
 class PropertyAction(Action):
     def __init__(self, na):
@@ -42,4 +120,5 @@ class PropsTest(unittest.TestCase):
         assert prop.get_properties() == ['a'], "Action does not contain property 'a'"
     
 if __name__ == "__main__":
-    unittest.main()
\ No newline at end of file
+    unittest.main()
+
--
cgit v0.9.1