-module(erlxslt_test).
 
-export([run/0]).
 
run() ->
    test_simple(),
    %%test_param(),
    test_multi(10000),
    test_extfun_arity(),
    test_extfun_retval(),
    ok.
 
stylesheet() ->
    "<xsl:stylesheet version='1.0'
                     xmlns:xsl='http://www.w3.org/1999/XSL/Transform'
                     xmlns:x='foo:bar'>
<xsl:output media-type='text/plain' method='text' omit-xml-declaration='yes'/>
 
<xsl:template match='simple'>
  <xsl:value-of select='string()'/>
</xsl:template>
 
<xsl:variable name='p1'/>
<xsl:template match='param'>
  <xsl:value-of select='$p1'/>
</xsl:template>
 
<xsl:template match='extfun1'>
0 <xsl:value-of select='x:arity0()'/>
1 <xsl:value-of select='x:arity1(23)'/>
2 <xsl:value-of select='x:arity2(23,42)'/>
3 <xsl:value-of select='x:arity3(23,42,7)'/>
</xsl:template>
 
<xsl:template match='extfun2'>
n <xsl:value-of select='x:number()'/>
s <xsl:value-of select='x:string()'/>
t <xsl:for-each select='x:treeish()/bar'><xsl:value-of select='.'/></xsl:for-each>
</xsl:template>
 
</xsl:stylesheet>".
 
test_simple() ->
    {ok, X} = erlxslt:start_link(),
    erlxslt:set_xslt(X, "style.xsl", stylesheet()),
    erlxslt:set_xml(X, "doc.xml", "<simple>Hello World</simple>"),
    {ok, "text/plain", "Hello World"} = erlxslt:process(X),
    erlxslt:stop(X).
 
test_param() ->
    {ok, X} = erlxslt:start_link(),
    erlxslt:set_xslt(X, "style.xsl", stylesheet()),
    erlxslt:set_xml(X, "doc.xml", "<param/>"),
    Param1 = "this is param1",
    erlxslt:set_params(X, [{"p1", Param1}]),
    {ok, "text/plain", Param1} = erlxslt:process(X),
    erlxslt:stop(X).
    
 
test_extfun_arity() ->
    {ok, X} = erlxslt:start_link(),
    erlxslt:register_function(X, "foo:bar", "arity0",
			      fun() -> ok end),
    erlxslt:register_function(X, "foo:bar", "arity1",
			      fun(23) -> ok end),
    erlxslt:register_function(X, "foo:bar", "arity2",
			      fun(23, 42) -> ok end),
    erlxslt:register_function(X, "foo:bar", "arity3",
			      fun(23, 42, 7) -> ok end),
    erlxslt:set_xslt(X, "style.xsl", stylesheet()),
    erlxslt:set_xml(X, "doc.xml", "<extfun1/>"),
    {ok, _, S} = erlxslt:process(X),
    "\n0 ok\n1 ok\n2 ok\n3 ok" = S,
    erlxslt:stop(X).
 
test_extfun_retval() ->
    {ok, X} = erlxslt:start_link(),
    erlxslt:register_function(X, "foo:bar", "number",
			      fun() -> 23 end),
    erlxslt:register_function(X, "foo:bar", "string",
			      fun() -> "foo" end),
    erlxslt:register_function(X, "foo:bar", "treeish",
			      fun() ->
				      {tree, "<foo><bar>23</bar><bar>42</bar></foo>"}
			      end),
    erlxslt:set_xslt(X, "style.xsl", stylesheet()),
    erlxslt:set_xml(X, "doc.xml", "<extfun2/>"),
    {ok, _, S} = erlxslt:process(X),
    "\nn 23\ns foo\nt 2342" = S,
    erlxslt:stop(X).
 
 
test_multi(N) ->
    {ok, X} = erlxslt:start_link(),
    erlxslt:set_xslt(X, "style.xsl", stylesheet()),
    erlxslt:register_function(X, "foo:bar", "arity0",
			      fun() -> ok end),
    erlxslt:register_function(X, "foo:bar", "arity1",
			      fun(23) -> ok end),
    erlxslt:register_function(X, "foo:bar", "arity2",
			      fun(23, 42) -> ok end),
    erlxslt:register_function(X, "foo:bar", "arity3",
			      fun(23, 42, 7) -> ok end),
    erlxslt:register_function(X, "foo:bar", "number",
			      fun() -> 23 end),
    erlxslt:register_function(X, "foo:bar", "string",
			      fun() -> "foo" end),
    erlxslt:register_function(X, "foo:bar", "treeish",
			      fun() ->
				      {tree, "<foo><bar>23</bar><bar>42</bar></foo>"}
			      end),
    test_multi(X, N),
    erlxslt:stop(X).
 
test_multi(_, 0) ->
    ok;
 
%% This is a leakcheck and should use all functionality
test_multi(X, N) ->
    erlxslt:set_xml(X, "doc.xml", "<r>
<simple>Hello World</simple>
<param/>
<extfun1/>
<extfun2/>
</r>"),
    %%erlxslt:set_params(X, [{"foo","bar"},{"baz","foobar"}]),
    {ok, "text/plain", _} = erlxslt:process(X),
    test_multi(X, N - 1).