// $Header: /home/cvs/jakarta-jmeter/src/htmlparser/org/htmlparser/tests/visitorsTests/NodeVisitorTest.java,v 1.2 2004/02/11 02:16:59 woolfel Exp $
/*
 * ====================================================================
 * Copyright 2002-2004 The Apache Software Foundation.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * 
 */

// The developers of JMeter and Apache are greatful to the developers
// of HTMLParser for giving Apache Software Foundation a non-exclusive
// license. The performance benefits of HTMLParser are clear and the
// users of JMeter will benefit from the hard work the HTMLParser
// team. For detailed information about HTMLParser, the project is
// hosted on sourceforge at http://htmlparser.sourceforge.net/.
//
// HTMLParser was originally created by Somik Raha in 2000. Since then
// a healthy community of users has formed and helped refine the
// design so that it is able to tackle the difficult task of parsing
// dirty HTML. Derrick Oswald is the current lead developer and was kind
// enough to assist JMeter.

package org.htmlparser.tests.visitorsTests;

import java.util.HashMap;
import java.util.Map;

import org.htmlparser.StringNode;
import org.htmlparser.tags.Tag;
import org.htmlparser.tests.ParserTestCase;
import org.htmlparser.visitors.NodeVisitor;

public class NodeVisitorTest extends ParserTestCase
{

    public NodeVisitorTest(String name)
    {
        super(name);
    }

    public void testVisitTag() throws Exception
    {
        ParameterVisitor visitor = new ParameterVisitor();
        createParser(
            ""
                + "value1"
                + "value2"
                + "");
        parser.visitAllNodesWith(visitor);
        assertEquals("value of key1", "value1", visitor.getValue("key1"));
        assertEquals("value of key2", "value2", visitor.getValue("key2"));
    }

    class ParameterVisitor extends NodeVisitor
    {
        Map paramsMap = new HashMap();
        String lastKeyVisited;

        public String getValue(String key)
        {
            return (String) paramsMap.get(key);
        }

        public void visitStringNode(StringNode stringNode)
        {
            paramsMap.put(lastKeyVisited, stringNode.getText());
        }

        public void visitTag(Tag tag)
        {
            if (tag.getTagName().equals("PARAM"))
            {
                lastKeyVisited = tag.getAttribute("NAME");
            }
        }
    }
}