mirror of
https://github.com/ai-robots-txt/ai.robots.txt.git
synced 2025-04-04 19:13:57 +00:00
17b826a6d3
For these simple tests Python's built-in unittest framework is more than enough. No additional dependencies are required. Added some more test cases with "special" characters to test the escaping code better.
58 lines
1.6 KiB
Python
Executable file
58 lines
1.6 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
"""To run these tests just execute this script."""
|
|
|
|
import json
|
|
import unittest
|
|
|
|
from robots import json_to_txt, json_to_table, json_to_htaccess
|
|
|
|
class RobotsUnittestExtensions:
|
|
def loadJson(self, pathname):
|
|
with open(pathname, "rt") as f:
|
|
return json.load(f)
|
|
|
|
def assertEqualsFile(self, f, s):
|
|
with open(f, "rt") as f:
|
|
f_contents = f.read()
|
|
|
|
return self.assertMultiLineEqual(f_contents, s)
|
|
|
|
|
|
class TestRobotsTXTGeneration(unittest.TestCase, RobotsUnittestExtensions):
|
|
maxDiff = 8192
|
|
|
|
def setUp(self):
|
|
self.robots_dict = self.loadJson("test_files/robots.json")
|
|
|
|
def test_robots_txt_generation(self):
|
|
robots_txt = json_to_txt(self.robots_dict)
|
|
self.assertEqualsFile("test_files/robots.txt", robots_txt)
|
|
|
|
|
|
class TestTableMetricsGeneration(unittest.TestCase, RobotsUnittestExtensions):
|
|
maxDiff = 32768
|
|
|
|
def setUp(self):
|
|
self.robots_dict = self.loadJson("test_files/robots.json")
|
|
|
|
def test_table_generation(self):
|
|
robots_table = json_to_table(self.robots_dict)
|
|
self.assertEqualsFile("test_files/table-of-bot-metrics.md", robots_table)
|
|
|
|
|
|
class TestHtaccessGeneration(unittest.TestCase, RobotsUnittestExtensions):
|
|
maxDiff = 8192
|
|
|
|
def setUp(self):
|
|
self.robots_dict = self.loadJson("test_files/robots.json")
|
|
|
|
def test_htaccess_generation(self):
|
|
robots_htaccess = json_to_htaccess(self.robots_dict)
|
|
self.assertEqualsFile("test_files/.htaccess", robots_htaccess)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import os
|
|
os.chdir(os.path.dirname(__file__))
|
|
|
|
unittest.main(verbosity=2)
|