source: josm/test/org/openstreetmap/josm/data/osm/visitor/AllNodesVisitorTest.java@ 146

package org.openstreetmap.josm.data.osm.visitor;

import java.util.Collection;
import java.util.Iterator;
import java.util.LinkedList;

import org.openstreetmap.josm.data.osm.Node;
import org.openstreetmap.josm.data.osm.OsmPrimitive;
import org.openstreetmap.josm.data.osm.Segment;
import org.openstreetmap.josm.data.osm.Way;
import org.openstreetmap.josm.testframework.MotherObject;

public class AllNodesVisitorTest extends MotherObject {

        private AllNodesVisitor v;

        @Override protected void setUp() throws Exception {
                super.setUp();
                v = new AllNodesVisitor();
        }

        public void testVisitNode() {
                Node node = createNode();
                node.visit(v);
                assertEquals(1, v.nodes.size());
                assertSame(node, v.nodes.iterator().next());
        }

        public void testVisitSegment() {
                Segment s = createSegment();
                s.visit(v);
                assertEquals(2, v.nodes.size());
                Iterator<Node> it = v.nodes.iterator();
                assertSame(s.from, it.next());
                assertSame(s.to, it.next());
        }

        public void testVisitWay() {
                Way w = createWay(createSegment());
                w.visit(v);
                int numberOfNodes = 2*w.segments.size();
                assertEquals(numberOfNodes, v.nodes.size());
                Iterator<Node> itNodes = v.nodes.iterator();
                Iterator<Segment> it = w.segments.iterator();
                for (int i = 0; i < numberOfNodes; ++i)
                        assertSame(i%2==0?it.next().from:it.next().to, itNodes.next());
        }

        public void testGetAllNodes() {
                Collection<OsmPrimitive> all = new LinkedList<OsmPrimitive>();
                all.add(createNode());
                all.add(createSegment());
                Collection<Node> nodes = AllNodesVisitor.getAllNodes(all);
                
                assertEquals(3, nodes.size());
        }

}