11"""Unit tests for the sitemap plugin."""
2+
23import datetime
34from unittest .mock import MagicMock , patch
45
@@ -85,7 +86,7 @@ def mock_component():
8586 image = "favicon.ico" ,
8687 on_load = None ,
8788 meta = [],
88- context = {}
89+ context = {},
8990 ),
9091 UnevaluatedPage (
9192 component = mock_component ,
@@ -95,7 +96,7 @@ def mock_component():
9596 image = "favicon.ico" ,
9697 on_load = None ,
9798 meta = [],
98- context = {}
99+ context = {},
99100 ),
100101 UnevaluatedPage (
101102 component = mock_component ,
@@ -105,7 +106,7 @@ def mock_component():
105106 image = "favicon.ico" ,
106107 on_load = None ,
107108 meta = [],
108- context = {"sitemap" : {"priority" : 0.7 , "changefreq" : "monthly" }}
109+ context = {"sitemap" : {"priority" : 0.7 , "changefreq" : "monthly" }},
109110 ),
110111 ]
111112 links = generate_links_for_sitemap (pages )
@@ -147,7 +148,7 @@ def mock_component():
147148 "lastmod" : now ,
148149 "priority" : 0.9 ,
149150 }
150- }
151+ },
151152 ),
152153 UnevaluatedPage (
153154 component = mock_component ,
@@ -157,7 +158,7 @@ def mock_component():
157158 image = "favicon.ico" ,
158159 on_load = None ,
159160 meta = [],
160- context = {}
161+ context = {},
161162 ), # No sitemap config
162163 UnevaluatedPage (
163164 component = mock_component ,
@@ -167,7 +168,7 @@ def mock_component():
167168 image = "favicon.ico" ,
168169 on_load = None ,
169170 meta = [],
170- context = {"sitemap" : {"changefreq" : "yearly" }}
171+ context = {"sitemap" : {"changefreq" : "yearly" }},
171172 ), # Has sitemap config but no loc
172173 ]
173174 links = generate_links_for_sitemap (pages )
@@ -204,7 +205,7 @@ def mock_component():
204205 image = "favicon.ico" ,
205206 on_load = None ,
206207 meta = [],
207- context = {"sitemap" : {"loc" : "/custom-404" , "priority" : 0.1 }}
208+ context = {"sitemap" : {"loc" : "/custom-404" , "priority" : 0.1 }},
208209 ),
209210 UnevaluatedPage (
210211 component = mock_component ,
@@ -214,7 +215,7 @@ def mock_component():
214215 image = "favicon.ico" ,
215216 on_load = None ,
216217 meta = [],
217- context = {"sitemap" : {"priority" : 0.2 }}
218+ context = {"sitemap" : {"priority" : 0.2 }},
218219 ), # Has sitemap config but no loc
219220 ]
220221 links = generate_links_for_sitemap (pages )
@@ -242,7 +243,7 @@ def mock_component():
242243 image = "favicon.ico" ,
243244 on_load = None ,
244245 meta = [],
245- context = {"sitemap" : {"loc" : "https://override.com/features_page" }}
246+ context = {"sitemap" : {"loc" : "https://override.com/features_page" }},
246247 ),
247248 UnevaluatedPage (
248249 component = mock_component ,
@@ -252,7 +253,7 @@ def mock_component():
252253 image = "favicon.ico" ,
253254 on_load = None ,
254255 meta = [],
255- context = {"sitemap" : {"loc" : "/custom_pricing" }}
256+ context = {"sitemap" : {"loc" : "/custom_pricing" }},
256257 ),
257258 ]
258259 links = generate_links_for_sitemap (pages )
@@ -278,7 +279,7 @@ def mock_component():
278279 image = "favicon.ico" ,
279280 on_load = None ,
280281 meta = [],
281- context = {"sitemap" : {"priority" : 1.5 }}
282+ context = {"sitemap" : {"priority" : 1.5 }},
282283 ),
283284 UnevaluatedPage (
284285 component = mock_component ,
@@ -288,7 +289,7 @@ def mock_component():
288289 image = "favicon.ico" ,
289290 on_load = None ,
290291 meta = [],
291- context = {"sitemap" : {"priority" : - 0.5 }}
292+ context = {"sitemap" : {"priority" : - 0.5 }},
292293 ),
293294 UnevaluatedPage (
294295 component = mock_component ,
@@ -298,14 +299,14 @@ def mock_component():
298299 image = "favicon.ico" ,
299300 on_load = None ,
300301 meta = [],
301- context = {"sitemap" : {"priority" : 0.5 }}
302+ context = {"sitemap" : {"priority" : 0.5 }},
302303 ),
303304 ]
304305 links = generate_links_for_sitemap (pages )
305306 expected_links = [
306307 {"loc" : "https://example.com/high_prio" , "priority" : 1.0 },
307308 {"loc" : "https://example.com/low_prio" , "priority" : 0.0 },
308- {"loc" : "https://example.com/valid_prio" , "priority" : 0.5 }
309+ {"loc" : "https://example.com/valid_prio" , "priority" : 0.5 },
309310 ]
310311 for expected_link in expected_links :
311312 assert expected_link in links
@@ -328,7 +329,7 @@ def mock_component():
328329 image = "favicon.ico" ,
329330 on_load = None ,
330331 meta = [],
331- context = {"sitemap" : {"loc" : "/home" }}
332+ context = {"sitemap" : {"loc" : "/home" }},
332333 ),
333334 UnevaluatedPage (
334335 component = mock_component ,
@@ -338,7 +339,7 @@ def mock_component():
338339 image = "favicon.ico" ,
339340 on_load = None ,
340341 meta = [],
341- context = {}
342+ context = {},
342343 ), # No loc, should use route
343344 UnevaluatedPage (
344345 component = mock_component ,
@@ -348,22 +349,20 @@ def mock_component():
348349 image = "favicon.ico" ,
349350 on_load = None ,
350351 meta = [],
351- context = {}
352+ context = {},
352353 ), # Special case for index
353354 ]
354355 links = generate_links_for_sitemap (pages )
355356 assert len (links ) == 3
356- expected_links = [
357- {"loc" : "/home" },
358- {"loc" : "/about" },
359- {"loc" : "/" }
360- ]
357+ expected_links = [{"loc" : "/home" }, {"loc" : "/about" }, {"loc" : "/" }]
361358 for expected_link in expected_links :
362359 assert expected_link in links
363360
364361
365362@patch ("reflex.config.get_config" )
366- def test_generate_links_for_sitemap_deploy_url_trailing_slash (mock_get_config : MagicMock ):
363+ def test_generate_links_for_sitemap_deploy_url_trailing_slash (
364+ mock_get_config : MagicMock ,
365+ ):
367366 """Test generate_links_for_sitemap with deploy_url having a trailing slash."""
368367 mock_get_config .return_value .deploy_url = "https://example.com/"
369368
@@ -379,7 +378,7 @@ def mock_component():
379378 image = "favicon.ico" ,
380379 on_load = None ,
381380 meta = [],
382- context = {}
381+ context = {},
383382 ),
384383 ]
385384 links = generate_links_for_sitemap (pages )
@@ -404,7 +403,7 @@ def mock_component():
404403 image = "favicon.ico" ,
405404 on_load = None ,
406405 meta = [],
407- context = {"sitemap" : {"loc" : "/another" }}
406+ context = {"sitemap" : {"loc" : "/another" }},
408407 ),
409408 ]
410409 links = generate_links_for_sitemap (pages )
@@ -429,7 +428,7 @@ def mock_component():
429428 image = "favicon.ico" ,
430429 on_load = None ,
431430 meta = [],
432- context = {"sitemap" : {"loc" : "http://othersite.com/page" }}
431+ context = {"sitemap" : {"loc" : "http://othersite.com/page" }},
433432 ),
434433 ]
435434 links = generate_links_for_sitemap (pages )
0 commit comments