X-Git-Url: http://nitlanguage.org

diff --git a/src/test_highlight.nit b/src/test_highlight.nit
index 96d0835..399691b 100644
--- a/src/test_highlight.nit
+++ b/src/test_highlight.nit
@@ -15,12 +15,19 @@
 # Program used to test the Nit highlighter
 module test_highlight
 
-import highlight
+import htmlight
 import test_phase
 
+class TestHighlightVisitor
+	super HtmlightVisitor
+	redef fun hrefto(e) do
+		return "#" + e.c_name
+	end
+end
+
 redef fun do_work(mainmodule, mmodules, mb)
 do
-	var v = new HighlightVisitor
+	var v = new TestHighlightVisitor
 	print """<head>
 	<meta charset="utf-8">
 	<style type="text/css">
@@ -33,9 +40,9 @@ do
 	for mm in mmodules do for cd in mm.mclassdefs do for pd in cd.mpropdefs do
 		var n = mb.mpropdef2node(pd)
 		if not n isa APropdef then continue
-		var hl = new HighlightVisitor
-		hl.enter_visit(n)
-		print "<h1>{pd.full_name}</h1>"
+		var hl = new TestHighlightVisitor
+		hl.highlight_node(n)
+		print "<h1 id=\"{pd.c_name}\">{pd.full_name}</h1>"
 		printn "<pre><code>"
 		hl.html.write_to(stdout)
 		print "</code></pre>"
@@ -61,8 +68,8 @@ class THLVisitor
 		if not seen.has(cn) then
 			seen.add cn
 
-			var hl = new HighlightVisitor
-			hl.enter_visit(n)
+			var hl = new TestHighlightVisitor
+			hl.highlight_node(n)
 			print "<h2>AST node: {cn} at {n.location}</h2>"
 			printn "<pre><code>"
 			hl.html.write_to(stdout)
