def test_includes(): """ Test of webpage.includes() """ webpage = Webpage(TEST_WEBSITE) webpage.html = '<h1>Some html</h1>' webpage.html += "<!-- MAGNETIZER_INCLUDE _include1.html -->" webpage.html += "<!-- MAGNETIZER_INCLUDE _include2.html -->" webpage.html += '<div>More html...</div>' webpage.html += "<!-- MAGNETIZER_INCLUDE _include3.html -->" webpage.html += "<!-- MAGNETIZER_INCLUDE _include1.html -->" correct_includes = ['_include1.html', '_include2.html', '_include3.html'] includes = webpage.includes() # Set should contain each include from the html for correct_include in correct_includes: assert correct_include in includes assert len(includes) == len(correct_includes)
def test_webpage_write(): """ Test of webpage.write() """ result = "This is a test!" webpage = Webpage(TEST_WEBSITE) webpage.html = result webpage.filename = 'my-post.html' webpage.write() # File should have the correct contents with open( TEST_WEBSITE.config.value('output_path') + webpage.filename, 'r') as myfile: assert myfile.read() == result # Page should be included in sitemap assert 'https://example.com/my-post.html' in TEST_WEBSITE.sitemap.pages TEST_WEBSITE.wipe()