rspamd/test/lua/unit/html.lua
2018-08-21 15:04:35 +01:00

129 wiersze
2.8 KiB
Lua

context("HTML processing", function()
local rspamd_util = require("rspamd_util")
local logger = require("rspamd_logger")
test("Extract text from HTML", function()
local cases = {
{[[
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="utf-8">
<title>title</title>
<link rel="stylesheet" href="style.css">
<script src="script.js"></script>
</head>
<body>
<!-- page content -->
Hello, world! <b>test</b>
<p>data<>
</P>
<b>stuff</p>?
</body>
</html>
]], "Hello, world! test\r\ndata\r\nstuff\r\n?"},
{[[
<?xml version="1.0" encoding="iso-8859-1"?>
<!DOCTYPE html
PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN"
"http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
<html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
<head>
<title>
Wikibooks
</title>
</head>
<body>
<p>
Hello, world!
</p>
</body>
</html>]], 'Hello, world!\r\n'},
{[[
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="utf-8">
<title>title</title>
<link rel="stylesheet" href="style.css">
<script src="script.js"></script>
<style><!--
- -a -a -a -- --- -
--></head>
<body>
<!-- page content -->
Hello, world!
</body>
</html>
]], 'Hello, world!'},
{[[
<html lang="en">
<head>
<meta charset="utf-8">
<title>title</title>
<link rel="stylesheet" href="style.css">
<script src="script.js"></script>
</head>
<body>
<!-- page content -->
Hello, world!<br>test</br><br>content</hr>more content<br>
<div>
content inside div
</div>
</body>
</html>
]], 'Hello, world!\r\ntest\r\ncontent\r\nmore content\r\ncontent inside div\r\n'},
{[[
<html lang="en">
<head>
<meta charset="utf-8">
<title>title</title>
<link rel="stylesheet" href="style.css">
<script src="script.js"></script>
</head>
<body>
<!-- tabular content -->
<table>
content
</table>
<table>
<tr>
<th>heada</th>
<th>headb</th>
</tr>
<tr>
<td>data1</td>
<td>data2</td>
</tr>
</table>
</body>
</html>
]], 'content\r\nheada headb\r\ndata1 data2\r\n'},
{[[
<html lang="en">
<head>
<meta charset="utf-8">
<title>title</title>
<link rel="stylesheet" href="style.css">
<script src="script.js"></script>
</head>
<body>
<!-- escape content -->
a&nbsp;b a &gt; b a &lt; b a &amp; b &apos;a &quot;a&quot;
</body>
</html>
]], 'a b a > b a < b a & b \'a "a"'},
}
for _,c in ipairs(cases) do
local t = rspamd_util.parse_html(c[1])
assert_not_nil(t)
assert_equal(c[2], tostring(t), string.format("'%s' doesn't match with '%s'",
c[2], t))
end
end)
end)