1.import java.net.URL;
2.
3.import junit.framework.TestCase;
4.
5.import org.apache.log4j.Logger;
6.import org.htmlparser.Node;
7.import org.htmlparser.NodeFilter;
8.import org.htmlparser.Parser;
9.import org.htmlparser.Tag;
10.import org.htmlparser.beans.LinkBean;
11.import org.htmlparser.filters.NodeClassFilter;
12.import org.htmlparser.filters.OrFilter;
13.import org.htmlparser.filters.TagNameFilter;
14.import org.htmlparser.tags.HeadTag;
15.import org.htmlparser.tags.ImageTag;
16.import org.htmlparser.tags.InputTag;
17.import org.htmlparser.tags.LinkTag;
18.import org.htmlparser.tags.OptionTag;
19.import org.htmlparser.tags.SelectTag;
20.import org.htmlparser.tags.TableColumn;
21.import org.htmlparser.tags.TableRow;
22.import org.htmlparser.tags.TableTag;
23.import org.htmlparser.tags.TitleTag;
24.import org.htmlparser.util.NodeIterator;
25.import org.htmlparser.util.NodeList;
26.import org.htmlparser.util.ParserException;
27.import org.htmlparser.visitors.HtmlPage;
28.import org.htmlparser.visitors.NodeVisitor;
29.import org.htmlparser.visitors.ObjectFindingVisitor;
30.
31.
public
class
T extends TestCase {
32.
33.
private
static
final Logger logger = Logger.getLogger(T.
class
);
34.
35.
public
T(String name) {
36. super(name);
37. }
38.
39.
/*
40. * 测试ObjectFindVisitor的用法
41. */
42.
public
void
testImageVisitor() {
43.
try
{
44. ImageTag imgLink;
45. ObjectFindingVisitor visitor =
new
ObjectFindingVisitor(ImageTag.
class
);
46. Parser parser =
new
Parser();
47. parser.setURL(
"http://www.google.com"
);
48. parser.setEncoding(parser.getEncoding());
49. parser.visitAllNodesWith(visitor);
50. Node[] nodes = visitor.getTags();
51.
for
(
int
i = 0; i < nodes.length; i++) {
52. imgLink = (ImageTag) nodes[i];
53. logger.fatal(
"testImageVisitor() ImageURL = "
+ imgLink.getImageURL());
54. logger.fatal(
"testImageVisitor() ImageLocation = "
+ imgLink.extractImageLocn());
55. logger.fatal(
"testImageVisitor() SRC = "
+ imgLink.getAttribute(
"SRC"
));
56. }
57. }
catch
(Exception e) {
58. e.printStackTrace();
59. }
60. }
61.
62.
/*
63. * 测试TagNameFilter用法
64. */
65.
public
void
testNodeFilter() {
66.
try
{
67. NodeFilter filter =
new
TagNameFilter(
"IMG"
);
68. Parser parser =
new
Parser();
69. parser.setURL(
"http://www.google.com"
);
70. parser.setEncoding(parser.getEncoding());
71. NodeList list = parser.extractAllNodesThatMatch(filter);
72.
for
(
int
i = 0; i < list.size(); i++) {
73. logger.fatal(
"testNodeFilter() "
+ list.elementAt(i).toHtml());
74. }
75. }
catch
(Exception e) {
76. e.printStackTrace();
77. }
78.
79. }
80.
81.
/*
82. * 测试NodeClassFilter用法
83. */
84.
public
void
testLinkTag() {
85.
try
{
86.
87. NodeFilter filter =
new
NodeClassFilter(LinkTag.
class
);
88. Parser parser =
new
Parser();
89. parser.setURL(
"http://www.google.com"
);
90. parser.setEncoding(parser.getEncoding());
91. NodeList list = parser.extractAllNodesThatMatch(filter);
92.
for
(
int
i = 0; i < list.size(); i++) {
93. LinkTag node = (LinkTag) list.elementAt(i);
94. logger.fatal(
"testLinkTag() Link is :"
+ node.extractLink());
95. }
96. }
catch
(Exception e) {
97. e.printStackTrace();
98. }
99.
100. }
101.
102.
/*
103. * 测试<link href=" text=’text/css’ rel=’stylesheet’ />用法
104. */
105.
public
void
testLinkCSS() {
106.
try
{
107.
108. Parser parser =
new
Parser();
109. parser.setInputHTML(
"<head><title>Link Test</title>"
110. +
"<link href=’/test01/css.css‘ text=‘text/css‘ rel=‘stylesheet‘ />"
111. +
"<link href=‘/test02/css.css‘ text=‘text/css‘ rel=‘stylesheet‘ />"
+
"</head>"
112. +
"<body>"
);
113. parser.setEncoding(parser.getEncoding());
114.
115.
for
(NodeIterator e = parser.elements(); e.hasMoreNodes();) {
116. Node node = e.nextNode();
117. logger.fatal(
"testLinkCSS()"
+ node.getText() + node.getClass());
118.
119. }
120. }
catch
(Exception e) {
121. e.printStackTrace();
122. }
123. }
124.
125.
/*
126. * 测试OrFilter的用法
127. */
128.
public
void
testOrFilter() {
129. NodeFilter inputFilter =
new
NodeClassFilter(InputTag.
class
);
130. NodeFilter selectFilter =
new
NodeClassFilter(SelectTag.
class
);
131.
132. NodeList nodeList = null;
133.
134.
try
{
135. Parser parser =
new
Parser();
136. parser
137. .setInputHTML(
"<head><title>OrFilter Test</title>"
138. +
"<link href=‘/test01/css.css‘ text=‘text/css‘ rel=‘stylesheet‘ />"
139. +
"<link href=‘/test02/css.css‘ text=‘text/css‘ rel=‘stylesheet‘ />"
140. +
"</head>"
141. +
"<body>"
142. +
"<input type=‘text‘ value=‘text1′ name=‘text1′/>"
143. +
"<input type=‘text‘ value=‘text2′ name=‘text2′/>"
144. +
"<select><option id=‘1′>1</option><option id=‘2′>2</option><option id=‘3′></option></select>"
145. +
"<a href=‘http://www.yeeach.com‘>yeeach.com</a>"
+
"</body>"
);
146.
147. parser.setEncoding(parser.getEncoding());
148. OrFilter lastFilter =
new
OrFilter();
149. lastFilter.setPredicates(
new
NodeFilter[] { selectFilter, inputFilter });
150. nodeList = parser.parse(lastFilter);
151.
for
(
int
i = 0; i <= nodeList.size(); i++) {
152.
if
(nodeList.elementAt(i) instanceof InputTag) {
153. InputTag tag = (InputTag) nodeList.elementAt(i);
154. logger.fatal(
"OrFilter tag name is :"
+ tag.getTagName() +
" ,tag value is:"
155. + tag.getAttribute(
"value"
));
156. }
157.
if
(nodeList.elementAt(i) instanceof SelectTag) {
158. SelectTag tag = (SelectTag) nodeList.elementAt(i);
159. NodeList list = tag.getChildren();
160.
161.
for
(
int
j = 0; j < list.size(); j++) {
162. OptionTag option = (OptionTag) list.elementAt(j);
163. logger.fatal(
"OrFilter Option"
+ option.getOptionText());
164. }
165.
166. }
167. }
168.
169. }
catch
(ParserException e) {
170. e.printStackTrace();
171. }
172. }
173.
174.
/*
175. * 测试对<table><tr><td></td></tr></table>的解析
176. */
177.
public
void
testTable() {
178. Parser myParser;
179. NodeList nodeList = null;
180. myParser = Parser.createParser(
"<body> "
+
"<table id=‘table1′ >"
181. +
"<tr><td>1-11</td><td>1-12</td><td>1-13</td>"
182. +
"<tr><td>1-21</td><td>1-22</td><td>1-23</td>"
183. +
"<tr><td>1-31</td><td>1-32</td><td>1-33</td></table>"
+
"<table id=‘table2′ >"
184. +
"<tr><td>2-11</td><td>2-12</td><td>2-13</td>"
185. +
"<tr><td>2-21</td><td>2-22</td><td>2-23</td>"
186. +
"<tr><td>2-31</td><td>2-32</td><td>2-33</td></table>"
+
"</body>"
,
"GBK"
);
187. NodeFilter tableFilter =
new
NodeClassFilter(TableTag.
class
);
188. OrFilter lastFilter =
new
OrFilter();
189. lastFilter.setPredicates(
new
NodeFilter[] { tableFilter });
190.
try
{
191. nodeList = myParser.parse(lastFilter);
192.
for
(
int
i = 0; i <= nodeList.size(); i++) {
193.
if
(nodeList.elementAt(i) instanceof TableTag) {
194. TableTag tag = (TableTag) nodeList.elementAt(i);
195. TableRow[] rows = tag.getRows();
196.
197.
for
(
int
j = 0; j < rows.length; j++) {
198. TableRow tr = (TableRow) rows[j];
199. TableColumn[] td = tr.getColumns();
200.
for
(
int
k = 0; k < td.length; k++) {
201. logger.fatal(
"<td>"
+ td[k].toPlainTextString());
202. }
203.
204. }
205.
206. }
207. }
208.
209. }
catch
(ParserException e) {
210. e.printStackTrace();
211. }
212. }
213.
214.
/*
215. * 测试NodeVisitor的用法,遍历所有节点
216. */
217.
public
void
testVisitorAll() {
218.
try
{
219. Parser parser =
new
Parser();
220. parser.setURL(
"http://www.google.com"
);
221. parser.setEncoding(parser.getEncoding());
222. NodeVisitor visitor =
new
NodeVisitor() {
223.
public
void
visitTag(Tag tag) {
224. logger.fatal(
"testVisitorAll() Tag name is :"
+ tag.getTagName() +
" \n Class is :"
225. + tag.getClass());
226. }
227.
228. };
229.
230. parser.visitAllNodesWith(visitor);
231. }
catch
(ParserException e) {
232. e.printStackTrace();
233. }
234. }
235.
236.
/*
237. * 测试对指定Tag的NodeVisitor的用法
238. */
239.
public
void
testTagVisitor() {
240.
try
{
241.
242. Parser parser =
new
Parser(
"<head><title>dddd</title>"
243. +
"<link href=‘/test01/css.css‘ text=‘text/css‘ rel=‘stylesheet‘ />"
244. +
"<link href=‘/test02/css.css‘ text=‘text/css‘ rel=‘stylesheet‘ />"
+
"</head>"
245. +
"<body>"
+
"<a href=‘http://www.yeeach.com‘>yeeach.com</a>"
+
"</body>"
);
246. NodeVisitor visitor =
new
NodeVisitor() {
247.
public
void
visitTag(Tag tag) {
248.
if
(tag instanceof HeadTag) {
249. logger.fatal(
"visitTag() HeadTag : Tag name is :"
+ tag.getTagName()
250. +
" \n Class is :"
+ tag.getClass() +
"\n Text is :"
+ tag.getText());
251. }
else
if
(tag instanceof TitleTag) {
252. logger.fatal(
"visitTag() TitleTag : Tag name is :"
+ tag.getTagName()
253. +
" \n Class is :"
+ tag.getClass() +
"\n Text is :"
+ tag.getText());
254.
255. }
else
if
(tag instanceof LinkTag) {
256. logger.fatal(
"visitTag() LinkTag : Tag name is :"
+ tag.getTagName()
257. +
" \n Class is :"
+ tag.getClass() +
"\n Text is :"
+ tag.getText()
258. +
" \n getAttribute is :"
+ tag.getAttribute(
"href"
));
259. }
else
{
260. logger.fatal(
"visitTag() : Tag name is :"
+ tag.getTagName() +
" \n Class is :"
261. + tag.getClass() +
"\n Text is :"
+ tag.getText());
262. }
263.
264. }
265.
266. };
267.
268. parser.visitAllNodesWith(visitor);
269. }
catch
(Exception e) {
270. e.printStackTrace();
271. }
272. }
273.
274.
/*
275. * 测试HtmlPage的用法
276. */
277.
public
void
testHtmlPage() {
278. String inputHTML =
"<html>"
+
"<head>"
279. +
"<title>Welcome to the HTMLParser website</title>"
+
"</head>"
+
"<body>"
280. +
"Welcome to HTMLParser"
+
"<table id=‘table1′ >"
281. +
"<tr><td>1-11</td><td>1-12</td><td>1-13</td>"
282. +
"<tr><td>1-21</td><td>1-22</td><td>1-23</td>"
283. +
"<tr><td>1-31</td><td>1-32</td><td>1-33</td></table>"
+
"<table id=‘table2′ >"
284. +
"<tr><td>2-11</td><td>2-12</td><td>2-13</td>"
285. +
"<tr><td>2-21</td><td>2-22</td><td>2-23</td>"
286. +
"<tr><td>2-31</td><td>2-32</td><td>2-33</td></table>"
+
"</body>"
+
"</html>"
;
287. Parser parser =
new
Parser();
288.
try
{
289.
290. parser.setInputHTML(inputHTML);
291. parser.setEncoding(parser.getURL());
292. HtmlPage page =
new
HtmlPage(parser);
293. parser.visitAllNodesWith(page);
294. logger.fatal(
"testHtmlPage -title is :"
+ page.getTitle());
295. NodeList list = page.getBody();
296.
297.
for
(NodeIterator iterator = list.elements(); iterator.hasMoreNodes();) {
298. Node node = iterator.nextNode();
299. logger.fatal(
"testHtmlPage -node is :"
+ node.toHtml());
300. }
301.
302. }
catch
(ParserException e) {
303.
// TODO Auto-generated catch block
304. e.printStackTrace();
305. }
306. }
307.
308.
/*
309. * 测试LinkBean的用法
310. */
311.
public
void
testLinkBean() {
312. Parser parser =
new
Parser();
313.
314. LinkBean linkBean =
new
LinkBean();
315. linkBean.setURL(
"http://www.google.com"
);
316. URL[] urls = linkBean.getLinks();
317.
318.
for
(
int
i = 0; i < urls.length; i++) {
319. URL url = urls[i];
320. logger.fatal(
"testLinkBean() -url is :"
+ url);
321. }
322.
323. }
324.
325.}