Improve coverage

nephila · Oct 31, 2015 · 194e9aa · 194e9aa
1 parent bf0563e
commit 194e9aa
Showing 1 changed file with 21 additions and 9 deletions.
diff --git a/tests/test_views.py b/tests/test_views.py
@@ -340,6 +340,7 @@ def test_feed(self):
 
     def test_sitemap(self):
         posts = self.get_posts()
+        self.get_pages()
         posts[0].tags.add('tag 1', 'tag 2', 'tag 3', 'tag 4')
         posts[0].save()
         posts[1].tags.add('tag 6', 'tag 2', 'tag 5', 'tag 8')
@@ -350,37 +351,48 @@ def test_sitemap(self):
         sitemap = BlogSitemap()
         self.assertEqual(len(sitemap.items()), 6)
         for item in sitemap.items():
-            self.assertTrue(sitemap.lastmod(item).date(), now().today())
-            self.assertTrue(
+            self.assertEqual(sitemap.lastmod(item).date(), now().date())
+            self.assertEqual(
                 sitemap.priority(item), get_setting('SITEMAP_PRIORITY_DEFAULT')
             )
-            self.assertTrue(
-                sitemap.priority(item), get_setting('SITEMAP_CHANGEFREQ_DEFAULT')
+            self.assertEqual(
+                sitemap.changefreq(item), get_setting('SITEMAP_CHANGEFREQ_DEFAULT')
             )
+            with smart_override(item.get_current_language()):
+                self.assertEqual(
+                    sitemap.location(item), item.get_absolute_url()
+                )
 
     def test_sitemap_config(self):
         posts = self.get_posts()
         self.app_config_1.app_data.config.sitemap_changefreq = 'daily'
         self.app_config_1.app_data.config.sitemap_priority = '0.2'
+        self.app_config_1.save()
 
         sitemap = BlogSitemap()
         self.assertEqual(len(sitemap.items()), 4)
         for item in sitemap.items():
-            self.assertTrue(sitemap.lastmod(item).date(), now().today())
+            self.assertEqual(sitemap.lastmod(item).date(), now().date())
             if item.app_config == self.app_config_1:
-                self.assertTrue(
+                self.assertEqual(
                     sitemap.priority(item), '0.2'
                 )
-                self.assertTrue(
+                self.assertEqual(
                     sitemap.changefreq(item), 'daily'
                 )
             else:
-                self.assertTrue(
+                self.assertEqual(
                     sitemap.priority(item), get_setting('SITEMAP_PRIORITY_DEFAULT')
                 )
-                self.assertTrue(
+                self.assertEqual(
                     sitemap.changefreq(item), get_setting('SITEMAP_CHANGEFREQ_DEFAULT')
                 )
+        self.assertEqual(
+            sitemap.priority(None), get_setting('SITEMAP_PRIORITY_DEFAULT')
+        )
+        self.assertEqual(
+            sitemap.changefreq(None), get_setting('SITEMAP_CHANGEFREQ_DEFAULT')
+        )
 
     def test_templates(self):
         posts = self.get_posts()