unittest omaha_server.utils

2 files changed, 26 insertions, 8 deletions

diff --git a/omaha_server/omaha_server/tests/test_core.py b/omaha_server/omaha_server/tests/test_core.py
deleted file mode 100644
index 395455a..0000000
--- a/omaha_server/omaha_server/tests/test_core.py
+++ /dev/null
@@ -1,8 +0,0 @@
-# coding: utf8
-
-from django.test import TestCase
-
-
-class CoreTest(TestCase):
-    def test_true(self):
-        assert True
\ No newline at end of file
diff --git a/omaha_server/omaha_server/tests/test_utils.py b/omaha_server/omaha_server/tests/test_utils.py
new file mode 100644
index 0000000..2f156a4
--- /dev/null
+++ b/omaha_server/omaha_server/tests/test_utils.py
@@ -0,0 +1,26 @@
+# coding: utf8
+
+from django.test import TestCase
+from django.test import override_settings
+from mock import Mock
+
+from omaha_server.utils import show_toolbar
+
+
+class UtilsTest(TestCase):
+    def setUp(self):
+        self.request = Mock()
+
+    def test_show_toolbar_ajax(self):
+        self.request.is_ajax = lambda: True
+        self.assertFalse(show_toolbar(self.request))
+
+    @override_settings(DEBUG=True)
+    def test_show_toolbar_debug_true(self):
+        self.request.is_ajax = lambda: False
+        self.assertTrue(show_toolbar(self.request))
+
+    @override_settings(DEBUG=False)
+    def test_show_toolbar_debug_false(self):
+        self.request.is_ajax = lambda: False
+        self.assertFalse(show_toolbar(self.request))