tests.py

#! /usr/bin/python3
"""Run tests."""

import unittest
import os
from staticfy.staticfy import staticfy


class DummyParser:
    """
    Empty class that simulates the parser in the main program.

    This was done so multiple arguments can be passed as an object.
    """

    framework = None
    namespace = None
    static_endpoint = None
    exc_tags = None
    add_tags = None


class StaticfyTest(unittest.TestCase):
    """Test Case."""

    @classmethod
    def setUpClass(cls):
        """Setup file to be used for tests."""
        cls.filename = "test.html"
        data = (
            """<link rel='stylesheet' href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/4.6.3/css/font-awesome.css" />\n"""  # noqa
            """<img src="/static/images/staticfy.jpg" />\n"""
            """<img data-url="images/staticfy.jpg" />\n"""
            """<link rel="stylesheet" href='../css/style.css' />\n"""
            """<script src="/js/script.js">alert('hello world')</script>\n"""
        )

        with open(cls.filename, "w+") as f:
            f.write(data)

    def setUp(self):
        """Create a fresh DummyParser for each test case."""
        self.args = DummyParser()

    def test_normal_staticfy(self):
        """Flask => {{ url_for('static', filename='css/style.css') }}."""
        result = staticfy(self.filename, args=self.args)

        expected_result = (
            """<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/4.6.3/css/font-awesome.css" />\n"""  # noqa
            """<img src="{{ url_for('static', filename='images/staticfy.jpg') }}" />\n""" # noqa
            """<img data-url="images/staticfy.jpg" />\n"""
            """<link rel="stylesheet" href="{{ url_for('static', filename='css/style.css') }}" />\n""" # noqa
            """<script src="{{ url_for('static', filename='js/script.js') }}">alert("hello world")</script>\n""" # noqa
        )

        self.assertEqual(result, expected_result)

    def test_static_endpoint(self):
        """Testing Static endpoint."""
        self.args.static_endpoint = "my_static"
        result = staticfy(self.filename, args=self.args)

        expected_result = (
            """<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/4.6.3/css/font-awesome.css" />\n""" # noqa
            """<img src="{{ url_for('my_static', filename='images/staticfy.jpg') }}" />\n""" # noqa
            """<img data-url="images/staticfy.jpg" />\n"""
            """<link rel="stylesheet" href="{{ url_for('my_static', filename='css/style.css') }}" />\n""" # noqa
            """<script src="{{ url_for('my_static', filename='js/script.js') }}">alert("hello world")</script>\n""" # noqa
        )
        self.assertEqual(result, expected_result)

    def test_additional_tags(self):
        """Testing additional tags."""
        self.args.add_tags = {"img": "data-url"}
        result = staticfy(self.filename, args=self.args)

        expected_result = (
            """<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/4.6.3/css/font-awesome.css" />\n""" # noqa
            """<img src="{{ url_for('static', filename='images/staticfy.jpg') }}" />\n""" # noqa
            """<img data-url="{{ url_for('static', filename='images/staticfy.jpg') }}" />\n""" # noqa
            """<link rel="stylesheet" href="{{ url_for('static', filename='css/style.css') }}" />\n""" # noqa
            """<script src="{{ url_for('static', filename='js/script.js') }}">alert("hello world")</script>\n""" # noqa
        )
        self.assertEqual(result, expected_result)

    def test_exclusive_tags(self):
        """Testing exclusive tags."""
        self.args.exc_tags = {"link": "href", "img": "src"}
        res = staticfy(self.filename, args=self.args)

        expected_result = (
            """<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/4.6.3/css/font-awesome.css" />\n""" # noqa
            """<img src="/static/images/staticfy.jpg" />\n"""
            """<img data-url="images/staticfy.jpg" />\n"""
            """<link rel="stylesheet" href="../css/style.css" />\n"""
            """<script src="{{ url_for('static', filename='js/script.js') }}">alert("hello world")</script>\n""" # noqa
        )
        self.assertEqual(res, expected_result)

    def test_cleanup_html(self):
        """Testing HTML cleanup."""
        result = staticfy(self.filename, args=self.args)

        expected_result = (
            """<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/4.6.3/css/font-awesome.css" />\n""" # noqa
            """<img src="{{ url_for('static', filename='images/staticfy.jpg') }}" />\n""" # noqa
            """<img data-url="images/staticfy.jpg" />\n"""
            """<link rel="stylesheet" href="{{ url_for('static', filename='css/style.css') }}" />\n""" # noqa
            """<script src="{{ url_for('static', filename='js/script.js') }}">alert("hello world")</script>\n""" # noqa
        )

        self.assertEqual(result, expected_result)

    def test_django_project(self):
        """Testing Django => {% static 'css/style.css' %}."""
        self.args.framework = "django"
        result = staticfy(self.filename, args=self.args)

        expected_result = (
            """<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/4.6.3/css/font-awesome.css" />\n""" # noqa
            """<img src="{% static 'images/staticfy.jpg' %}" />\n"""
            """<img data-url="images/staticfy.jpg" />\n"""
            """<link rel="stylesheet" href="{% static 'css/style.css' %}" />\n""" # noqa
            """<script src="{% static 'js/script.js' %}">alert("hello world")</script>\n""" # noqa
        )
        self.assertEqual(result, expected_result)

    def test_laravel_project(self):
        """Testing Laravel => {{ URL::asset('css/bootstrap.min.css') }}."""
        self.args.framework = "laravel"
        result = staticfy(self.filename, args=self.args)

        expected_result = (
            """<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/4.6.3/css/font-awesome.css" />\n""" # noqa
            """<img src="{{ URL::asset('images/staticfy.jpg') }}" />\n"""
            """<img data-url="images/staticfy.jpg" />\n"""
            """<link rel="stylesheet" href="{{ URL::asset('css/style.css') }}" />\n""" # noqa
            """<script src="{{ URL::asset('js/script.js') }}">alert("hello world")</script>\n""" # noqa
        )
        self.assertEqual(result, expected_result)

    def test_namespace(self):
        """Testing namespace."""
        self.args.framework = "django"
        self.args.namespace = "admin"
        result = staticfy(self.filename, args=self.args)

        expected_result = (
            """<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/4.6.3/css/font-awesome.css" />\n""" # noqa
            """<img src="{% static 'admin/images/staticfy.jpg' %}" />\n"""
            """<img data-url="images/staticfy.jpg" />\n"""
            """<link rel="stylesheet" href="{% static 'admin/css/style.css' %}" />\n""" # noqa
            """<script src="{% static 'admin/js/script.js' %}">alert("hello world")</script>\n""" # noqa
        )
        self.assertEqual(result, expected_result)

    def test_replace_relative_links(self):
        """Testing replace_relative_links."""
        self.args.framework = "django"
        result = staticfy(self.filename, args=self.args)

        expected_result = (
            """<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/4.6.3/css/font-awesome.css" />\n""" # noqa
            """<img src="{% static 'images/staticfy.jpg' %}" />\n"""
            """<img data-url="images/staticfy.jpg" />\n"""
            """<link rel="stylesheet" href="{% static 'css/style.css' %}" />\n""" # noqa
            """<script src="{% static 'js/script.js' %}">alert("hello world")</script>\n""" # noqa
        )
        self.assertEqual(result, expected_result)

    def test_filenotfound_exception(self):
        self.assertRaises(
            IOError, staticfy, "Invalid html file", args=self.args
        )

    @classmethod
    def tearDownClass(cls):
        """Remove test.html file."""
        os.remove(cls.filename)


if __name__ == "__main__":
    unittest.main()