2 files changed, 42 insertions, 0 deletions
diff --git a/examples/pygments_api/models.py b/examples/pygments_api/models.py
new file mode 100644
index 00000000..402a813e
--- /dev/null
+++ b/examples/pygments_api/models.py
@@ -0,0 +1 @@
+#We need models.py otherwise the test framework complains (http://code.djangoproject.com/ticket/7198)
diff --git a/examples/pygments_api/tests.py b/examples/pygments_api/tests.py
new file mode 100644
index 00000000..f70eff0e
--- /dev/null
+++ b/examples/pygments_api/tests.py
@@ -0,0 +1,41 @@
+from django.test import TestCase
+from djangorestframework.compat import RequestFactory
+from pygments_api import views
+import os, tempfile, shutil
+
+class TestPygmentsExample(TestCase):
+
+    def setUp(self):
+        self.factory = RequestFactory()
+        self.temp_dir = tempfile.mkdtemp()
+        views.HIGHLIGHTED_CODE_DIR = self.temp_dir
+        
+    def tearDown(self):
+        try:
+            shutil.rmtree(self.temp_dir)
+        except:
+            pass
+        
+    def test_get_to_root(self):
+        '''Just do a get on the base url'''
+        request = self.factory.get('/pygments')
+        view = views.PygmentsRoot.as_view()
+        response = view(request)
+        self.assertEqual(response.status_code, 200)
+
+    def test_snippets_datetime_sorted(self):
+        '''Pygments examples should be datetime sorted'''
+        locations = []
+        for snippet in 'abcdefghijk':
+            form_data = {'code': '%s' % snippet, 'style':'friendly', 'lexer':'python'}
+            request = self.factory.post('/pygments', data=form_data)
+            view = views.PygmentsRoot.as_view()
+            response = view(request)
+            locations.append(response.items()[2][1])
+        request = self.factory.get('/pygments')
+        view = views.PygmentsRoot.as_view()
+        response = view(request)
+        self.assertEquals(locations, response.content)
+        
+        
+        
+\ No newline at end of file