test_documentation.py

import unittest
import pytest

import httpx

from marqo import marqo_docs

class TestDocumentation(unittest.TestCase):
    def test_urls(self):
        # Retrieve all public functions in the module
        public_functions = [func for func in dir(marqo_docs)
                            if callable(getattr(marqo_docs, func)) and not func.startswith("_")]

        # Verify all URLs return a 200 response
        for func in public_functions:
            with self.subTest(func=func):
                url = getattr(marqo_docs, func)()
                print(f"url: {url}")
                response = httpx.get(url, follow_redirects=True)
                response.raise_for_status()
                self.assertFalse(
                    '404.html' in response.content.decode(), f"{func} URL returned a 404 response"
                )
                self.assertFalse(
                    '<title>Redirecting</title>' in response.content.decode(), f"{func} URL is a redirect"
                )