add: тесты

tests/test_safe_html_standalone.py

@@ -0,0 +1,236 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Тесты для функции safe_html_spec_symbols() из oknardia/web/add_func.py
+
+Проверяет:
+1. Удаление содержимого исключённых тегов (script, style, code, kbd, pre, var, samp)
+2. Удаление обычных HTML-тегов
+3. Замену HTML-мнемоник на Unicode (именованные, десятичные, шестнадцатеричные)
+4. Очистку лишних пробелов
+"""
+
+import sys
+import os
+
+# Добавим путь к проекту для импорта (подъём на одну папку выше, т.к. тесты в папке tests/)
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'oknardia'))
+
+from web.add_func import safe_html_spec_symbols
+
+
+def test_remove_script_tags():
+    """Тест 1: Удаление содержимого тегов <script>"""
+    html = 'Текст <script>alert("hack");</script> после'
+    result = safe_html_spec_symbols(html)
+    assert 'alert' not in result, f"Script-содержимое не удалено: {result}"
+    assert 'Текст' in result and 'после' in result, f"Обычный текст потеряется: {result}"
+    print("✓ Тест 1 (удаление <script>): пройден")
+
+
+def test_remove_style_tags():
+    """Тест 2: Удаление содержимого тегов <style>"""
+    html = 'Блок <style>#id { color: red; }</style> текста'
+    result = safe_html_spec_symbols(html)
+    assert 'color' not in result, f"Style-содержимое не удалено: {result}"
+    assert 'Блок' in result and 'текста' in result, f"Обычный текст потеряется: {result}"
+    print("✓ Тест 2 (удаление <style>): пройден")
+
+
+def test_remove_code_tags():
+    """Тест 3: Удаление содержимого тегов <code>, <kbd>, <pre>"""
+    html = 'Команда <code>def foo():</code> в тексте. <kbd>Ctrl+C</kbd> текст.'
+    result = safe_html_spec_symbols(html)
+    assert 'def foo' not in result, f"Code-содержимое не удалено: {result}"
+    assert 'Ctrl' not in result, f"Kbd-содержимое не удалено: {result}"
+    assert 'Команда' in result and 'тексте' in result, f"Обычный текст потеряется: {result}"
+    print("✓ Тест 3 (удаление <code>, <kbd>): пройден")
+
+
+def test_remove_object_tags():
+    """Тест 3a: Удаление содержимого тегов <object>, <embed>"""
+    html = 'Текст <object data="malicious.swf"></object> и <embed src="bad.swf"/> после'
+    result = safe_html_spec_symbols(html)
+    assert 'malicious.swf' not in result, f"Object не удалён: {result}"
+    assert 'bad.swf' not in result, f"Embed не удалён: {result}"
+    assert 'Текст' in result and 'после' in result, f"Обычный текст потеряется: {result}"
+    print("✓ Тест 3a (удаление <object>, <embed>): пройден")
+
+
+def test_remove_form_tags():
+    """Тест 3b: Удаление содержимого тегов <form>, <input>, <textarea>"""
+    html = 'Текст <form><input type="password"/><textarea>secret</textarea></form> после'
+    result = safe_html_spec_symbols(html)
+    assert 'secret' not in result and 'password' not in result, f"Form содержимое не удалено: {result}"
+    assert 'password' not in result, f"Input атрибут не удалён: {result}"
+    assert 'Текст' in result and 'после' in result, f"Обычный текст потеряется: {result}"
+    print("✓ Тест 3b (удаление <form>, <input>, <textarea>): пройден")
+
+
+def test_remove_svg_canvas():
+    """Тест 3c: Удаление содержимого тегов <svg>, <canvas>"""
+    html = 'Текст <svg><script>alert("xss")</script></svg> и <canvas id="c"></canvas> после'
+    result = safe_html_spec_symbols(html)
+    assert 'xss' not in result and 'script' not in result, f"SVG содержимое не удалено: {result}"
+    assert 'Текст' in result and 'после' in result, f"Обычный текст потеряется: {result}"
+    print("✓ Тест 3c (удаление <svg>, <canvas>): пройден")
+
+
+def test_remove_html_tags():
+    """Тест 4: Удаление обычных HTML-тегов"""
+    html = '<p>Параграф <b>с полужирным</b> <i>и курсивом</i></p> <span>спан</span>'
+    result = safe_html_spec_symbols(html)
+    assert '<' not in result and '>' not in result, f"HTML-теги не удалены: {result}"
+    assert 'Параграф' in result and 'полужирным' in result and 'курсивом' in result, \
+        f"Текст из тегов потеряется: {result}"
+    print("✓ Тест 4 (удаление HTML-тегов): пройден")
+
+
+def test_named_entities():
+    """Тест 5: Замена именованных HTML-мнемоник"""
+    html = '&nbsp;&lt; &gt; &quot; &apos; &amp; &euro; &copy; &reg;'
+    result = safe_html_spec_symbols(html)
+    # html.unescape преобразует мнемоники в символы
+    assert '&' not in result or 'amp' not in result, f"Мнемоники не заменены: {result}"
+    assert '€' in result, f"Euro не заменён: {result}"
+    assert '©' in result, f"Copyright не заменён: {result}"
+    assert '®' in result, f"Registered не заменён: {result}"
+    print("✓ Тест 5 (именованные мнемоники): пройден")
+
+
+def test_numeric_entities_decimal():
+    """Тест 6: Замена десятичных числовых мнемоник (&#ЧИСЛО;)"""
+    html = '&#8470; &#169; &#8364;'  # № © €
+    result = safe_html_spec_symbols(html)
+    assert '№' in result, f"Decimal entity &#8470; не заменена: {result}"
+    assert '©' in result, f"Decimal entity &#169; не заменена: {result}"
+    assert '€' in result, f"Decimal entity &#8364; не заменена: {result}"
+    print("✓ Тест 6 (десятичные мнемоники): пройден")
+
+
+def test_numeric_entities_hex():
+    """Тест 7: Замена шестнадцатеричных числовых мнемоник (&#xHEX;)"""
+    html = '&#x20AC; &#xA9; &#x2116;'  # € © №
+    result = safe_html_spec_symbols(html)
+    assert '€' in result, f"Hex entity &#x20AC; не заменена: {result}"
+    assert '©' in result, f"Hex entity &#xA9; не заменена: {result}"
+    assert '№' in result, f"Hex entity &#x2116; не заменена: {result}"
+    print("✓ Тест 7 (шестнадцатеричные мнемоники): пройден")
+
+
+def test_whitespace_cleanup():
+    """Тест 8: Очистка лишних пробелов"""
+    html = 'Текст     с    множественными     пробелами\nи\tтабуляцией'
+    result = safe_html_spec_symbols(html)
+    assert '  ' not in result, f"Лишние пробелы не удалены: {repr(result)}"
+    assert 'Текст с множественными пробелами и табуляцией' == result, \
+        f"Ожидается 'Текст с множественными пробелами и табуляцией', получено: {repr(result)}"
+    print("✓ Тест 8 (очистка пробелов): пройден")
+
+
+def test_strip_edges():
+    """Тест 9: Удаление пробелов в начале и конце"""
+    html = '   Текст    '
+    result = safe_html_spec_symbols(html)
+    assert result == 'Текст', f"Пробелы не удалены: {repr(result)}"
+    print("✓ Тест 9 (удаление пробелов в начале/конце): пройден")
+
+
+def test_complex_html():
+    """Тест 10: Комплексный тест с комбинацией всего"""
+    html = '''
+        <div class="content">
+            <p>Текст &nbsp; с   <b>мнемониками</b>: &euro; №&#8470; &#x20AC;</p>
+            <script>malicious_code()</script>
+            <style>.hide { display: none; }</style>
+            <span>Ещё текст &copy; 2024</span>
+        </div>
+    '''
+    result = safe_html_spec_symbols(html)
+    
+    # Проверяем, что исключены опасные теги
+    assert 'malicious_code' not in result, f"Script не удалён: {result}"
+    assert 'display: none' not in result, f"Style не удалён: {result}"
+    
+    # Проверяем, что обычный текст остался
+    assert 'Текст' in result and 'Ещё текст' in result, f"Обычный текст потеряился: {result}"
+    
+    # Проверяем, что HTML-теги удалены
+    assert '<' not in result and '>' not in result, f"HTML-теги не удалены: {result}"
+    
+    # Проверяем, что мнемоники заменены
+    assert '€' in result, f"Мнемоники не заменены: {result}"
+    assert '©' in result, f"Copyright не заменён: {result}"
+    
+    print(f"✓ Тест 10 (комплексный): пройден")
+    print(f"  Результат: {result[:80]}...")
+
+
+def test_empty_string():
+    """Тест 11: Пустая строка"""
+    result = safe_html_spec_symbols('')
+    assert result == '', f"Ожидается пустая строка, получено: {repr(result)}"
+    print("✓ Тест 11 (пустая строка): пройден")
+
+
+def test_only_html_tags():
+    """Тест 12: Строка только с HTML-тегами"""
+    html = '<div><p></p></div>'
+    result = safe_html_spec_symbols(html)
+    assert result == '', f"Ожидается пустая строка, получено: {repr(result)}"
+    print("✓ Тест 12 (только теги): пройден")
+
+
+def test_russian_text():
+    """Тест 13: Русский текст с мнемониками"""
+    html = 'Цена: <b>1000&nbsp;₽</b> &laquo;Российский&raquo; &mdash; лучший выбор'
+    result = safe_html_spec_symbols(html)
+    assert 'Цена' in result, f"Русский текст потеряется: {result}"
+    assert '«' in result and '»' in result, f"Кавычки не заменены: {result}"
+    assert '—' in result, f"Длинное тире не заменено: {result}"
+    print(f"✓ Тест 13 (русский текст): пройден")
+
+
+if __name__ == '__main__':
+    print("=" * 60)
+    print("Запуск тестов функции safe_html_spec_symbols()")
+    print("=" * 60)
+    
+    tests = [
+        test_remove_script_tags,
+        test_remove_style_tags,
+        test_remove_code_tags,
+        test_remove_object_tags,
+        test_remove_form_tags,
+        test_remove_svg_canvas,
+        test_remove_html_tags,
+        test_named_entities,
+        test_numeric_entities_decimal,
+        test_numeric_entities_hex,
+        test_whitespace_cleanup,
+        test_strip_edges,
+        test_complex_html,
+        test_empty_string,
+        test_only_html_tags,
+        test_russian_text,
+    ]
+    
+    failed = 0
+    for test in tests:
+        try:
+            test()
+        except AssertionError as e:
+            print(f"✗ {test.__name__} ОШИБКА: {e}")
+            failed += 1
+        except Exception as e:
+            print(f"✗ {test.__name__} ИСКЛЮЧЕНИЕ: {e}")
+            failed += 1
+    
+    print("=" * 60)
+    if failed == 0:
+        print(f"✅ Все {len(tests)} тестов пройдены успешно!")
+    else:
+        print(f"❌ Провалено {failed} из {len(tests)} тестов")
+        sys.exit(1)
+    print("=" * 60)
+