1 files changed, 71 insertions, 0 deletions
diff --git a/ipa_webui/tests/test_controllers.py b/ipa_webui/tests/test_controllers.py
new file mode 100644
index 00000000..71e5c0e2
--- /dev/null
+++ b/ipa_webui/tests/test_controllers.py
@@ -0,0 +1,71 @@
+# Authors: Jason Gerard DeRose <jderose@redhat.com>
+#
+# Copyright (C) 2008  Red Hat
+# see file 'COPYING' for use and warranty information
+#
+# 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 the Free Software Foundation; version 2 only
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+
+"""
+Test the `ipa_webui.controller` module.
+"""
+
+from ipa_webui import controller
+
+
+class test_Controller(object):
+    """
+    Test the `controller.Controller` class.
+    """
+
+    def test_init(self):
+        """
+        Test the `ipa_webui.controller.Controller.__init__()` method.
+        """
+        cmd = 'The command.'
+        template = 'The template.'
+        o = controller.Controller(cmd, template)
+        assert o.cmd is cmd
+        assert o.template is template
+
+    def test_serialize(self):
+        """
+        Test the `ipa_webui.controller.Controller.serialize` method.
+        """
+        class Template(object):
+            def __init__(self):
+                self.calls = 0
+                self.kw = {}
+
+            def serialize(self, **kw):
+                self.calls += 1
+                self.kw = kw
+                return dict(kw)
+
+        d = dict(output='xhtml-strict', format='pretty+nice')
+        t = Template()
+        o = controller.Controller(None, t)
+        assert o.serialize() == d
+        assert t.calls == 1
+
+    def test_call(self):
+        """
+        Test the `ipa_webui.controller.Controller.__call__` method.
+        """
+        class Template(object):
+            def serialize(self, **kw):
+                return 'Your login is %s.' % kw['result']
+        def cmd(**kw):
+            return kw['first'][0] + kw['last']
+        o = controller.Controller(cmd, Template())
+        assert o(first='John', last='Doe') == 'Your login is JDoe.'